from fastapi import FastAPI, Query
from fastapi.middleware.cors import CORSMiddleware
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("/")
def root():
    return {"status": "ok"}

@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)
    return fig.to_dict()