Update src/api_server.py

src/api_server.py

@@ -1,5 +1,7 @@
 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")
@@ -11,14 +13,11 @@ api.add_middleware(
     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()
+    fig_json = pio.to_json(fig)
+    return JSONResponse(content=fig_json)