Update qtAnswering/main.py

qtAnswering/main.py

@@ -1,72 +1,67 @@
-from fastapi import FastAPI, UploadFile, Form, Request
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
-from fastapi.staticfiles import StaticFiles
-from fastapi.templating import Jinja2Templates
-import shutil, os
-from tempfile import gettempdir
-
-app = FastAPI()
-
-# ✅ CORS to allow frontend access
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# ✅ Static assets
-app.mount("/resources", StaticFiles(directory="resources"), name="resources")
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
-# ✅ Jinja2 Templates
-templates = Jinja2Templates(directory="templates")
-
-# ✅ Serve Homepage
-@app.get("/", response_class=HTMLResponse)
-async def serve_home(request: Request):
-    return templates.TemplateResponse("home.html", {"request": request})
-
-# ✅ Predict endpoint (handles image + document)
-@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)
-
-        is_image = file.content_type.startswith("image/")
-
-        if is_image:
-            from appImage import answer_question_from_image
-            from PIL import Image
-            image = Image.open(temp_path).convert("RGB")
-            answer, audio_path = answer_question_from_image(image, question)
-
-        else:
-            from app import answer_question_from_doc
-            class NamedFile:
-                def __init__(self, name): self.filename = name
-                def read(self): return open(self.filename, "rb").read()
-            answer, audio_path = answer_question_from_doc(NamedFile(temp_path), question)
-
-        os.remove(temp_path)
-
-        if audio_path and os.path.exists(audio_path):
-            return JSONResponse({
-                "answer": answer,
-                "audio": f"/audio/{os.path.basename(audio_path)}"
-            })
-        else:
-            return JSONResponse({"answer": answer})
-
-    except Exception as e:
-        return JSONResponse({"error": str(e)}, status_code=500)
-
-# ✅ Serve audio
-@app.get("/audio/{filename}")
-async def get_audio(filename: str):
-    filepath = os.path.join(gettempdir(), filename)
-    return FileResponse(filepath, media_type="audio/mpeg")
+from fastapi import FastAPI, UploadFile, Form, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
+from fastapi.templating import Jinja2Templates
+import shutil, os
+from tempfile import gettempdir
+
+# ✅ Create app
+app = FastAPI()
+
+# ✅ CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# ✅ Templates
+templates = Jinja2Templates(directory="templates")
+
+# ✅ Serve Homepage
+@app.get("/", response_class=HTMLResponse)
+async def serve_home(request: Request):
+    return templates.TemplateResponse("home.html", {"request": request})
+
+# ✅ Predict endpoint (handles image + document)
+@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)
+
+        is_image = file.content_type.startswith("image/")
+
+        if is_image:
+            from .appImage import answer_question_from_image
+            from PIL import Image
+            image = Image.open(temp_path).convert("RGB")
+            answer, audio_path = answer_question_from_image(image, question)
+        else:
+            from .app import answer_question_from_doc
+            class NamedFile:
+                def __init__(self, name): self.filename = name
+                def read(self): return open(self.filename, "rb").read()
+            answer, audio_path = answer_question_from_doc(NamedFile(temp_path), question)
+
+        os.remove(temp_path)
+
+        if audio_path and os.path.exists(audio_path):
+            return JSONResponse({
+                "answer": answer,
+                "audio": f"/qtAnswering/audio/{os.path.basename(audio_path)}"
+            })
+        else:
+            return JSONResponse({"answer": answer})
+
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+
+# ✅ Serve audio files
+@app.get("/audio/{filename}")
+async def get_audio(filename: str):
+    filepath = os.path.join(gettempdir(), filename)
+    return FileResponse(filepath, media_type="audio/mpeg")