Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi import FastAPI, UploadFile, File, HTTPException, Form
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import RedirectResponse
 from transformers import pipeline
@@ -30,12 +30,24 @@ def read_root():
     return RedirectResponse(url="/static/index.html")
 
 @app.post("/summarize")
-async def summarize_text(file: UploadFile = File(...)):
-    logger.info(f"Received document for summarization: {file.filename}")
+async def summarize_text(
+    file: UploadFile = File(None),  # Optional file upload
+    text: str = Form(None)  # Optional manual text input
+):
+    if file:
+        logger.info(f"Received document for summarization: {file.filename}")
+        try:
+            # Extract text from the document
+            text = await extract_text_from_file(file)
+        except Exception as e:
+            logger.error(f"Error extracting text from file: {e}")
+            raise HTTPException(status_code=400, detail=str(e))
+    elif text:
+        logger.info("Received manual text for summarization")
+    else:
+        raise HTTPException(status_code=400, detail="No file or text provided")
+
     try:
-        # Extract text from the document
-        text = await extract_text_from_file(file)
-        
         # Use the text pipeline to summarize the text
         summary = text_pipeline(f"summarize: {text}", max_length=100)
         logger.info(f"Generated summary: {summary[0]['generated_text']}")
@@ -61,13 +73,25 @@ async def caption_image(file: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/answer")
-async def answer_question(file: UploadFile = File(...), question: str = ""):
-    logger.info(f"Received document for question answering: {file.filename}")
-    logger.info(f"Received question: {question}")
+async def answer_question(
+    file: UploadFile = File(None),  # Optional file upload
+    text: str = Form(None),  # Optional manual text input
+    question: str = Form(...)  # Required question
+):
+    if file:
+        logger.info(f"Received document for question answering: {file.filename}")
+        try:
+            # Extract text from the document
+            text = await extract_text_from_file(file)
+        except Exception as e:
+            logger.error(f"Error extracting text from file: {e}")
+            raise HTTPException(status_code=400, detail=str(e))
+    elif text:
+        logger.info("Received manual text for question answering")
+    else:
+        raise HTTPException(status_code=400, detail="No file or text provided")
+
     try:
-        # Extract text from the document
-        text = await extract_text_from_file(file)
-        
         # Use the text pipeline to answer the question
         answer = text_pipeline(f"question: {question} context: {text}")
         logger.info(f"Generated answer: {answer[0]['generated_text']}")
@@ -77,7 +101,7 @@ async def answer_question(file: UploadFile = File(...), question: str = ""):
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/vqa")
-async def visual_question_answering(file: UploadFile = File(...), question: str = ""):
+async def visual_question_answering(file: UploadFile = File(...), question: str = Form(...)):
     logger.info(f"Received image for visual question answering: {file.filename}")
     logger.info(f"Received question: {question}")
     try: