Update main.py

main.py

@@ -1,23 +1,33 @@
-from fastapi import FastAPI, UploadFile, Form
+from fastapi import FastAPI, UploadFile, Form, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse, RedirectResponse
+from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 import shutil, os
 
 app = FastAPI()
 
-# CORS
+# ✅ CORS to allow frontend to access backend
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"], allow_credentials=True,
-    allow_methods=["*"], allow_headers=["*"]
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
 )
 
-# ✅ Basic root redirection (optional)
-@app.get("/")
-def home():
-    return RedirectResponse(url="/docs")  # Or to a frontend URL
+# ✅ Mount static folder (optional JS/CSS support)
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-# ✅ File prediction endpoint
+# ✅ Set templates folder
+templates = Jinja2Templates(directory="templates")
+
+# ✅ Serve home.html
+@app.get("/", response_class=HTMLResponse)
+async def serve_home(request: Request):
+    return templates.TemplateResponse("home.html", {"request": request})
+
+# ✅ Backend prediction API
 @app.post("/predict")
 async def predict(question: str = Form(...), file: UploadFile = Form(...)):
     try: