Fix frontend serving in main.py

backend/main.py

@@ -1,15 +1,17 @@
 """
-LocalShortsLab - FastAPI Application
+Rovin - FastAPI Application
 
 Main entry point for the backend server.
 """
 
 import logging
+from pathlib import Path
 from contextlib import asynccontextmanager
 
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 
 from .config import get_settings
 from .routes import router
@@ -26,23 +28,27 @@ logger = logging.getLogger(__name__)
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Startup and shutdown events."""
-    # Startup
-    logger.info("Starting LocalShortsLab...")
+    logger.info("Starting Rovin...")
     await init_job_manager()
     
     yield
     
+    logger.info("Shutting down Rovin...")
+    await shutdown_job_manager()
+
+
 # Create FastAPI app
 app = FastAPI(
     title="Rovin API",
-    description="Backend for Rovin Video Generator",
+    description="AI-powered Short Video Generator",
     version="1.0.0",
+    lifespan=lifespan,
 )
 
-# CORS middleware for frontend
+# CORS middleware
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # In production, restrict this
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -51,6 +57,46 @@ app.add_middleware(
 # Include API routes
 app.include_router(router, prefix="/api")
 
+# Serve Frontend Static Files (Production)
+# Check for frontend build directory
+static_dir = Path(__file__).parent.parent / "frontend" / "dist"
+
+if static_dir.exists():
+    logger.info(f"Serving frontend from: {static_dir}")
+    
+    # Mount assets directory
+    assets_dir = static_dir / "assets"
+    if assets_dir.exists():
+        app.mount("/assets", StaticFiles(directory=str(assets_dir)), name="assets")
+    
+    # Serve index.html for root
+    @app.get("/")
+    async def serve_index():
+        return FileResponse(static_dir / "index.html")
+    
+    # Catch-all route for SPA (must be last)
+    @app.get("/{full_path:path}")
+    async def serve_spa(full_path: str):
+        # Don't catch API routes
+        if full_path.startswith("api"):
+            raise HTTPException(status_code=404, detail="API endpoint not found")
+        
+        # Check if it's a static file
+        file_path = static_dir / full_path
+        if file_path.exists() and file_path.is_file():
+            return FileResponse(file_path)
+        
+        # Otherwise return index.html for client-side routing
+        return FileResponse(static_dir / "index.html")
+else:
+    logger.warning(f"Frontend not found at {static_dir}. Running API only mode.")
+    
+    @app.get("/")
+    async def root():
+        return {"message": "Rovin API is running", "status": "healthy"}
+
+
+@app.get("/health")
 async def health():
     """Health check endpoint."""
     return {"status": "healthy"}