update

app/main.py

@@ -5,6 +5,7 @@ from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
 from app.dependencies import engine
 from app.sql_models import Base
+from app.routers.songs import get_songs
 
 Base.metadata.create_all(engine)
 
@@ -25,6 +26,24 @@ async def root(request: Request, session_id : str = ""):
             request=request, name="landing.html"
         )
     else:
+        songs = await get_songs(session_id)
+        songs_by_category = {}
+        all_categories = set()
+        for song in songs:
+            if song.main_category not in songs_by_category:
+                songs_by_category[song.main_category] = []
+            songs_by_category[song.main_category].append(song)
+            all_categories.update(song.categories.keys())
         return templates.TemplateResponse(
-            request=request, name="voting.html"
-        )
+            request=request, name="voting.html", context={
+                "songs_by_category": songs_by_category, 
+                "all_categories": {c: i+1 for i, c in enumerate(all_categories)},
+                "session_id": session_id
+                }
+        )
+
+#@app.get("/vote", response_class=HTMLResponse)
+#async def vote(request: Request, session_id : str = ""):
+#    return templates.TemplateResponse(
+#        request=request, name="voting-old.html"
+#    )