Added: CSS template switch in Dashboard.

main.py

@@ -1,5 +1,5 @@
 import os
-from fastapi import FastAPI, Request, Depends
+from fastapi import FastAPI, Request, Depends, HTTPException
 from starlette.middleware.sessions import SessionMiddleware
 from starlette.staticfiles import StaticFiles
 from starlette.responses import FileResponse, RedirectResponse
@@ -10,6 +10,7 @@ from sqlalchemy.orm import Session
 import models
 from database import engine
 import auth # Import the new auth module
+import schemas
 from config import settings # Import settings to get the secret key
 
 # --- Absolute Path Configuration ---
@@ -56,8 +57,13 @@ async def read_dashboard(request: Request, db: Session = Depends(auth.get_db)):
     
     user = db.query(models.User).filter(models.User.id == user_id).first()
     overlay_url = f"{settings.APP_BASE_URL}/overlay/{user.id}"
+    
+    # Ensure user has settings, create if they don't for some reason
+    if not user.settings:
+        user.settings = models.Setting()
+        db.commit()
 
-    return templates.TemplateResponse("dashboard.html", {"request": request, "user": user, "overlay_url": overlay_url})
+    return templates.TemplateResponse("dashboard.html", {"request": request, "user": user, "overlay_url": overlay_url, "current_theme": user.settings.overlay_theme})
 
 @app.get("/logout")
 async def logout(request: Request):
@@ -66,7 +72,29 @@ async def logout(request: Request):
     return RedirectResponse(url="/")
 
 @app.get("/overlay/{user_id}")
-async def read_overlay(request: Request, user_id: int):
+async def read_overlay(request: Request, user_id: int, db: Session = Depends(auth.get_db)):
     # This endpoint serves the overlay page.
-    # For now, it just renders the template with placeholder text.
-    return templates.TemplateResponse("overlay.html", {"request": request})
+    user = db.query(models.User).filter(models.User.id == user_id).first()
+    if not user:
+        raise HTTPException(status_code=404, detail="User not found")
+
+    theme = "dark-purple" # Default theme
+    if user.settings and user.settings.overlay_theme:
+        theme = user.settings.overlay_theme
+
+    return templates.TemplateResponse(f"overlay-{theme}.html", {"request": request})
+
+@app.post("/api/settings")
+async def update_settings(settings_data: schemas.SettingsUpdate, request: Request, db: Session = Depends(auth.get_db)):
+    user_id = request.session.get('user_id')
+    if not user_id:
+        raise HTTPException(status_code=401, detail="Not authenticated")
+
+    user = db.query(models.User).filter(models.User.id == user_id).first()
+    if not user.settings:
+        user.settings = models.Setting()
+    
+    user.settings.overlay_theme = settings_data.overlay_theme
+    db.commit()
+    
+    return {"message": "Settings updated successfully"}