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.Row.update(visible=show_api_fields)  # ✅ 用 Row 的更新
    )

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_fields]
    )

    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)