Update app/main.py

app/main.py

@@ -4,7 +4,6 @@ import logging
 import traceback
 from pathlib import Path
 from typing import Dict, Any
-import asyncio
 from datetime import datetime
 
 from fastapi import FastAPI, File, UploadFile, HTTPException, Request
@@ -151,12 +150,15 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Use frontend folder as static files
-frontend_dir = Path("frontend")
-if not frontend_dir.exists():
-    logger.warning("⚠️ Frontend directory not found. UI may not load correctly.")
-else:
+# Static frontend directory setup
+BASE_DIR = Path(__file__).parent
+frontend_dir = BASE_DIR / "frontend"
+
+if frontend_dir.exists():
+    logger.info(f"✅ Frontend directory found: {frontend_dir}")
     app.mount("/static", StaticFiles(directory=frontend_dir), name="static")
+else:
+    logger.warning("⚠️ Frontend directory not found. UI will not load correctly.")
 
 @app.on_event("startup")
 async def startup_event():
@@ -169,11 +171,13 @@ async def startup_event():
 
 @app.get("/", response_class=HTMLResponse)
 async def root():
-    """Serve main frontend file"""
-    html_file = Path("frontend/index.html")
+    html_file = frontend_dir / "index.html"
     if html_file.exists():
         return FileResponse(html_file)
-    return HTMLResponse("<h1>⚠️ Frontend not found</h1><p>Please ensure frontend/index.html exists.</p>")
+    return HTMLResponse("""
+        <h1>⚠️ Frontend not found</h1>
+        <p>Please ensure 'frontend/index.html' exists in the project root.</p>
+    """)
 
 @app.get("/health")
 async def health_check():