from fastapi import FastAPI, Query
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import plotly.io as pio
from src.streamlit_app import load_data, create_bar_chart

api = FastAPI(title="Leaderboard API")

api.add_middleware(
    CORSMiddleware,
    allow_origins=["https://leggedrobotics.github.io"],
    allow_methods=["*"],
    allow_headers=["*"],
)

@api.get("/chart")
def get_chart(view: str = Query("Total Score", description="Chart type")):
    """Return chart JSON for the given view type."""
    
    df = load_data()
    fig = create_bar_chart(df, view)
    fig_json = pio.to_json(fig)
    return JSONResponse(content=fig_json)