Update main.py

main.py

@@ -4,7 +4,7 @@ from fastapi.responses import FileResponse
 from pydantic import BaseModel
 from typing import List, Optional
 import os
-import chat, voice, database
+from . import chat, voice, database
 import traceback
 
 app = FastAPI()
@@ -72,6 +72,14 @@ async def chat_session_endpoint(session_id: str, request: ChatRequest):
         database.add_message(session_id, "user", request.message)
         database.add_message(session_id, "assistant", response_text)
         
+        # If this was the first message (history was empty before this turn), generate a summary
+        if not db_history:
+            try:
+                summary = chat.generate_summary(request.message, response_text)
+                database.update_session_name(session_id, summary)
+            except Exception as e:
+                print(f"Failed to generate summary: {e}")
+        
         return {"response": response_text}
     except Exception as e:
         traceback.print_exc()
@@ -98,3 +106,30 @@ async def speak_endpoint(request: VoiceRequest):
         return FileResponse(audio_path, media_type="audio/mpeg", filename="response.mp3")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
+# Mount static files if they exist (for production/served build)
+frontend_dist = os.path.join(os.path.dirname(__file__), "../frontend/dist")
+assets_path = os.path.join(frontend_dist, "assets")
+
+if os.path.exists(assets_path):
+    app.mount("/assets", StaticFiles(directory=assets_path), name="assets")
+
+@app.get("/{full_path:path}")
+async def serve_frontend(full_path: str):
+    # Only serve if dist exists
+    if os.path.exists(frontend_dist):
+        # Serve index.html for any other path (SPA routing)
+        # Check if file exists in dist, else serve index.html
+        target_file = os.path.join(frontend_dist, full_path)
+        if full_path and os.path.exists(target_file):
+            return FileResponse(target_file)
+        return FileResponse(os.path.join(frontend_dist, "index.html"))
+    
+    # If dist doesn't exist, just return a message or 404 for frontend routes
+    # This allows backend to run even if frontend isn't built
+    if full_path == "":
+        return {"message": "Backend is running. Frontend build not found. Use 'npm run dev' for frontend development."}
+    raise HTTPException(status_code=404, detail="Frontend build not found")