Update app/main.py

app/main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 from typing import List
 import os
@@ -22,16 +23,30 @@ app = FastAPI(
 @app.on_event("startup")
 async def startup_event():
     """Display clickable link on startup"""
+    import os
+    
+    # Detect if running on HuggingFace Spaces
+    space_id = os.getenv("SPACE_ID")
+    
     print("\n" + "="*70)
     print("🚀 Google Docs Knowledge Chatbot is running!")
     print("="*70)
-    print("\n📱 Access the application here:")
-    print("\n   👉 \033[94m\033[4mhttp://localhost:8000/static/index.html\033[0m\n")
+    
+    if space_id:
+        # Running on HuggingFace Spaces
+        print("\n📱 Application deployed on HuggingFace Spaces")
+        print(f"   Space ID: {space_id}")
+    else:
+        # Running locally
+        print("\n📱 Access the application here:")
+        print("\n   👉 \033[94m\033[4mhttp://localhost:8000\033[0m\n")
+    
     print("="*70)
     print("\n💡 Quick Tips:")
     print("   • Click 'Index All Documents' to get started")
     print("   • Make sure your Google Drive folder is shared")
-    print("   • Press CTRL+C to stop the server")
+    if not space_id:
+        print("   • Press CTRL+C to stop the server")
     print("\n" + "="*70 + "\n")
 
 # Add CORS middleware
@@ -55,9 +70,18 @@ llm_service = LLMService(settings.groq_api_key)
 # Create data directory
 os.makedirs(settings.vector_store_path, exist_ok=True)
 
+# Mount static files BEFORE defining routes
+app.mount("/static", StaticFiles(directory="frontend"), name="static")
+
 
 @app.get("/")
 async def root():
+    """Serve the frontend HTML"""
+    return FileResponse("frontend/index.html")
+
+
+@app.get("/api/status")
+async def api_status():
     """Health check endpoint"""
     return {
         "status": "running",
@@ -574,8 +598,4 @@ async def clear_index():
     except HTTPException:
         raise
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error clearing index: {str(e)}")
-
-
-# Serve frontend
-app.mount("/static", StaticFiles(directory="frontend"), name="static")
+        raise HTTPException(status_code=500, detail=f"Error clearing index: {str(e)}")