Update main.py

main.py

@@ -2,65 +2,54 @@
 from fastapi import FastAPI, UploadFile, File, Form, Request
 from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware
 from fastapi.templating import Jinja2Templates
+from fastapi.middleware.cors import CORSMiddleware
 import tempfile, os
-from app import extract_text, generate_summary, text_to_speech, create_pdf
-from appImage import generate_caption
+
+from app import summarize_document
+from appImage import caption_image
 
 app = FastAPI()
 
-app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"])
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 app.mount("/static", StaticFiles(directory="static"), name="static")
 app.mount("/resources", StaticFiles(directory="resources"), name="resources")
 templates = Jinja2Templates(directory="templates")
 
+# Serve homepage
 @app.get("/", response_class=HTMLResponse)
 async def serve_home(request: Request):
     return templates.TemplateResponse("HomeS.html", {"request": request})
 
+# Document summarization endpoint
 @app.post("/summarize/")
-async def summarize_document_endpoint(file: UploadFile = File(...), length: str = Form("medium")):
-    contents = await file.read()
-    with tempfile.NamedTemporaryFile(delete=False) as tmp:
-        tmp.write(contents)
-        tmp_path = tmp.name
-
-    file_ext = file.filename.split('.')[-1].lower()
-    text, error = extract_text(tmp_path, file_ext)
-
-    if error:
-        return JSONResponse({"detail": error}, status_code=400)
-
-    if not text or len(text.split()) < 30:
-        return JSONResponse({"detail": "Document too short to summarize"}, status_code=400)
-
-    summary = generate_summary(text, length)
-    audio_path = text_to_speech(summary)
-    pdf_path = create_pdf(summary, file.filename)
-
-    response = {"summary": summary}
-    if audio_path:
-        response["audioUrl"] = f"/files/{os.path.basename(audio_path)}"
-    if pdf_path:
-        response["pdfUrl"] = f"/files/{os.path.basename(pdf_path)}"
-
-    return JSONResponse(response)
-
+async def summarize(file: UploadFile = File(...), length: str = Form("medium")):
+    try:
+        result = await summarize_document(file, length)
+        return JSONResponse(result)
+    except Exception as e:
+        return JSONResponse({"error": f"Summarization failed: {str(e)}"}, status_code=500)
+
+# Image captioning endpoint
 @app.post("/imagecaption/")
-async def caption_image_endpoint(file: UploadFile = File(...)):
-    contents = await file.read()
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
-        tmp.write(contents)
-        image_path = tmp.name
-
-    caption = generate_caption(image_path)
-    return JSONResponse({"caption": caption})
-
+async def caption(file: UploadFile = File(...)):
+    try:
+        result = await caption_image(file)
+        return JSONResponse(result)
+    except Exception as e:
+        return JSONResponse({"error": f"Image captioning failed: {str(e)}"}, status_code=500)
+
+# Serve audio/pdf generated
 @app.get("/files/{filename}")
 async def serve_file(filename: str):
-    path = os.path.join(tempfile.gettempdir(), filename)
-    if os.path.exists(path):
-        return FileResponse(path)
+    filepath = os.path.join(tempfile.gettempdir(), filename)
+    if os.path.exists(filepath):
+        return FileResponse(filepath)
     return JSONResponse({"error": "File not found"}, status_code=404)