Update main.py

main.py

@@ -1,31 +1,30 @@
 from fastapi import FastAPI, UploadFile, Form
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-import shutil
-import os
+from fastapi.responses import JSONResponse, RedirectResponse
+import shutil, os
 
-# ✅ Create FastAPI app
 app = FastAPI()
 
-# ✅ Enable CORS so frontend can talk to backend
+# CORS
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
+    allow_origins=["*"], allow_credentials=True,
+    allow_methods=["*"], allow_headers=["*"]
 )
 
-# ✅ Prediction endpoint
+# ✅ Basic root redirection (optional)
 @app.get("/")
+def home():
+    return RedirectResponse(url="/docs")  # Or to a frontend URL
+
+# ✅ File prediction endpoint
+@app.post("/predict")
 async def predict(question: str = Form(...), file: UploadFile = Form(...)):
     try:
-        # Save uploaded file temporarily
         temp_path = f"temp_{file.filename}"
         with open(temp_path, "wb") as f:
             shutil.copyfileobj(file.file, f)
 
-        # ✅ Decide handler based on file type
         if file.content_type.startswith("image/"):
             from appImage import answer_question_from_image
             from PIL import Image
@@ -34,8 +33,7 @@ async def predict(question: str = Form(...), file: UploadFile = Form(...)):
         else:
             from app import answer_question_from_doc
             class NamedFile:
-                def __init__(self, name):
-                    self.name = name
+                def __init__(self, name): self.name = name
             answer = answer_question_from_doc(NamedFile(temp_path), question)
 
         os.remove(temp_path)