evalscope_v0.17.0/gradio_ui.py

89 lines
2.9 KiB
Python

import gradio as gr
def run_eval(inputs, native, other, outputs, api_url, api_token):
result = (
f"\n[Eval Started]\n"
f"Inputs: {inputs}\n"
f"API URL: {api_url}\n"
f"API Token: {api_token}\n"
f"Native Modules: {native}\n"
f"Other Backends: {other}\n"
f"Outputs: {outputs}\n"
f"[Eval Finished]"
)
return result
def enforce_input_exclusive_and_toggle_fields(selected):
group1 = {"API Models", "Local Models"}
group2 = {"Benchmarks", "Custom Datasets"}
def keep_only_one(group):
filtered = [item for item in selected if item in group]
return filtered[-1:] # 只保留最后一个
final_selection = set(selected)
final_selection -= group1
final_selection |= set(keep_only_one(group1))
final_selection -= group2
final_selection |= set(keep_only_one(group2))
show_api_fields = "API Models" in final_selection
return (
gr.update(value=list(final_selection)),
gr.update(visible=show_api_fields),
gr.update(visible=show_api_fields)
)
with gr.Blocks(title="EvalScope 全功能界面") as demo:
with gr.Group():
with gr.Row():
input_choices = gr.CheckboxGroup(
label="选择输入源",
choices=["API Models", "Local Models", "Benchmarks", "Custom Datasets"],
interactive=True
)
with gr.Row(visible=False) as api_fields:
api_url_input = gr.Textbox(label="API 地址", placeholder="https://api.example.com/v1/chat")
api_token_input = gr.Textbox(label="Token 密钥", type="password", placeholder="sk-xxx")
with gr.Row():
with gr.Column():
native_choices = gr.CheckboxGroup(
label="启用本地模块",
choices=["Model Adapter", "Data Adapter", "Evaluator", "Perf Monitor"]
)
with gr.Column():
other_choices = gr.CheckboxGroup(
label="启用外部后端",
choices=["OpenCompass", "VLMEvalKit", "RAGAS", "MTEB/CMTEB"]
)
with gr.Row():
output_choices = gr.CheckboxGroup(
label="输出形式",
choices=["Evaluation Report", "Gradio", "WandB", "Swanlab"]
)
run_button = gr.Button("Run Evaluation")
output_text = gr.Textbox(label="执行结果", lines=10)
# 绑定输入源选择的变化,控制互斥+是否显示API字段
input_choices.change(
fn=enforce_input_exclusive_and_toggle_fields,
inputs=input_choices,
outputs=[input_choices, api_url_input, api_token_input]
)
run_button.click(
run_eval,
inputs=[input_choices, native_choices, other_choices, output_choices, api_url_input, api_token_input],
outputs=output_text
)
if __name__ == '__main__':
demo.launch(server_name="0.0.0.0", server_port=7900)