"""Benchmark Visualizer Web Service.

A standalone web service for visualizing benchmark results.
Currently supports LoComo benchmark visualization.
"""

import json
from pathlib import Path
from typing import Any

from fastapi import FastAPI, HTTPException
from fastapi.responses import HTMLResponse, FileResponse
from fastapi.staticfiles import StaticFiles

app = FastAPI(title="Benchmark Visualizer")

# Get the benchmarks directory
BENCHMARKS_DIR = Path(__file__).parent.parent


@app.get("/", response_class=HTMLResponse)
async def index():
    """Serve the main benchmark visualizer page."""
    html_path = Path(__file__).parent / "static" / "index.html"
    with open(html_path) as f:
        return f.read()


@app.get("/api/locomo")
async def get_locomo_results(mode: str = "search") -> dict[str, Any]:
    """Get LoComo benchmark results.

    Returns pre-computed benchmark results from the locomo directory.

    Args:
        mode: Either "search" (default) or "think" to select which results to load
    """
    try:
        # Determine filename based on mode
        if mode == "think":
            filename = "benchmark_results_think.json"
        else:
            filename = "benchmark_results.json"

        results_path = BENCHMARKS_DIR / "locomo" / filename

        if not results_path.exists():
            raise HTTPException(
                status_code=404,
                detail=f"Benchmark results not found for mode '{mode}'. Please run the benchmark first with {'--use-think' if mode == 'think' else 'default settings'}."
            )

        with open(results_path) as f:
            results = json.load(f)

        return results
    except json.JSONDecodeError as e:
        raise HTTPException(
            status_code=500,
            detail=f"Failed to parse benchmark results: {str(e)}"
        )
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


# Mount static files
app.mount("/static", StaticFiles(directory=Path(__file__).parent / "static"), name="static")


if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="127.0.0.1", port=8001)