from __future__ import annotations

import json
import sys
from pathlib import Path

import gradio as gr

ROOT = Path(__file__).resolve().parent
sys.path.insert(0, str(ROOT / "src"))

from amongus_env.eval_suite import run_eval_suite  # noqa: E402
from amongus_env.golden_episode import run_golden_episode  # noqa: E402


def run_baseline_eval() -> str:
    return json.dumps(run_eval_suite(), indent=2)


def run_golden_trace() -> str:
    return json.dumps(run_golden_episode(), indent=2)


with gr.Blocks(title="Among Us OpenEnv Eval") as demo:
    gr.Markdown(
        "# Among Us OpenEnv Eval\n"
        "Deterministic social-deduction eval for the OpenEnv MVP."
    )
    with gr.Row():
        eval_button = gr.Button("Run Baseline Eval")
        trace_button = gr.Button("Run Golden Trace")
    output = gr.Code(language="json", label="Output")

    eval_button.click(run_baseline_eval, outputs=output)
    trace_button.click(run_golden_trace, outputs=output)


if __name__ == "__main__":
    demo.launch()