Update main.py

main.py

@@ -1,58 +1,38 @@
-from fastapi import FastAPI, UploadFile, Form, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
+# main.py
+from fastapi import FastAPI, Request, UploadFile, Form
+from fastapi.responses import FileResponse, HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-import shutil, os
+import shutil
+import os
+import app
+import appImage
 
 app = FastAPI()
 
-
-
-# ✅ CORS to allow frontend to access backend
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-app.mount("/resources", StaticFiles(directory="resources"), name="resources")
-
-# ✅ Mount static folder (optional JS/CSS support)
+# Mount static and templates
 app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/resources", StaticFiles(directory="resources"), name="resources")
 
-# ✅ Set templates folder
 templates = Jinja2Templates(directory="templates")
 
-# ✅ Serve home.html
+# Serve your main home page
 @app.get("/", response_class=HTMLResponse)
 async def serve_home(request: Request):
     return templates.TemplateResponse("home.html", {"request": request})
 
-# ✅ Backend prediction API
+# Route to handle prediction requests (image or doc)
 @app.post("/predict")
-async def predict(question: str = Form(...), file: UploadFile = Form(...)):
-    try:
-        temp_path = f"temp_{file.filename}"
-        with open(temp_path, "wb") as f:
-            shutil.copyfileobj(file.file, f)
-
-        if file.content_type.startswith("image/"):
-            from appImage import answer_question_from_image
-            from PIL import Image
-            image = Image.open(temp_path)
-            answer = answer_question_from_image(image, question)
-        else:
-            from app import answer_question_from_doc
-            class NamedFile:
-                def __init__(self, name): self.name = name
-            answer = answer_question_from_doc(NamedFile(temp_path), question)
-
-        os.remove(temp_path)
-        return JSONResponse(content={"answer": answer})
-
-    except Exception as e:
-        return JSONResponse(content={"error": str(e)}, status_code=500)
-
-
+async def predict(file: UploadFile = Form(...), question: str = Form(...)):
+    ext = file.filename.split(".")[-1].lower()
+    with open(file.filename, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    if ext in ["pdf", "docx", "pptx", "xlsx"]:
+        answer, audio = app.answer_question_from_doc(file=buffer, question=question)
+    else:
+        image = appImage.Image.open(file.filename)
+        answer, audio = appImage.answer_question_from_image(image, question)
+
+    os.remove(file.filename)
+    return JSONResponse({"answer": answer, "audio": audio})