Update main.py

main.py

@@ -30,12 +30,38 @@ async def serve_home(request: Request):
 
 # Document summarization endpoint
 @app.post("/summarize/")
-async def summarize(file: UploadFile = File(...), length: str = Form("medium")):
+async def summarize_api(file: UploadFile = File(...), length: str = Form("medium")):
     try:
-        result = await summarize_document(file, length)
-        return JSONResponse(result)
+        contents = await file.read()
+        with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
+            tmp_file.write(contents)
+            tmp_path = tmp_file.name
+
+        file_ext = tmp_path.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)
+
     except Exception as e:
-        return JSONResponse({"error": f"Summarization failed: {str(e)}"}, status_code=500)
+        print(f"Error during summarization: {str(e)}")  # << SUPER IMPORTANT FOR DEBUG
+        return JSONResponse({"detail": f"Internal server error: {str(e)}"}, status_code=500)
+
 
 # Image captioning endpoint
 @app.post("/imagecaption/")