Update main.py

main.py

@@ -25,4 +25,23 @@ async def root():
     return FileResponse("index.html")
 
 # TEXT summarization
-@app.post("/summariz
+@app.post("/summarize/text")
+async def summarize_text(data: dict):
+    text = data.get("text", "")
+    if not text.strip():
+        return {"summary": "No text provided."}
+
+    summary = generate_summary(text)
+    return {"summary": summary}
+
+# PDF summarization
+@app.post("/summarize/pdf")
+async def summarize_pdf(file: UploadFile = File(...)):
+    pdf_bytes = await file.read()
+    text = extract_text_from_pdf(pdf_bytes)
+
+    if not text.strip():
+        return {"summary": "PDF is empty or could not be processed."}
+
+    summary = generate_summary(text)
+    return {"summary": summary}