Update app.py

app.py

@@ -10,7 +10,7 @@ from datetime import datetime
 from typing import List, Dict, Optional, Tuple
 from enum import Enum
 
-from fastapi import FastAPI, HTTPException, UploadFile, File, Query
+from fastapi import FastAPI, HTTPException, UploadFile, File, Query, Form
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
@@ -440,6 +440,45 @@ async def synthesize_voice(request: VoiceOutputRequest):
         logger.error(f"Error in voice synthesis: {e}")
         raise HTTPException(status_code=500, detail="Error generating voice output")
 
+@app.post("/upload")
+async def upload_and_analyze_file(file: UploadFile = File(...), format: str = Form("clean")):
+    try:
+        # 1. Read file content
+        contents = await file.read()
+        filename = file.filename.lower()
+
+        # 2. Extract text depending on file type
+        if filename.endswith(".pdf"):
+            import fitz  # PyMuPDF
+            doc = fitz.open(stream=contents, filetype="pdf")
+            text = "\n".join([page.get_text() for page in doc])
+        elif filename.endswith(".txt"):
+            text = contents.decode("utf-8", errors="ignore")
+        elif filename.endswith(".csv"):
+            import pandas as pd
+            df = pd.read_csv(io.BytesIO(contents))
+            text = df.to_string()
+        elif filename.endswith(".xlsx"):
+            import pandas as pd
+            df = pd.read_excel(io.BytesIO(contents))
+            text = df.to_string()
+        else:
+            raise HTTPException(status_code=400, detail="Unsupported file type")
+
+        # 3. Send to TxAgent
+        message = (
+            f"You are a clinical assistant AI.\n\n"
+            f"Analyze the following patient document:\n\n{text[:10000]}"
+        )
+        response = agent.chat(message=message, history=[], temperature=0.7, max_new_tokens=1024)
+
+        return {"status": "success", "filename": file.filename, "response": response}
+
+    except Exception as e:
+        logger.error(f"File upload + analysis failed: {e}")
+        raise HTTPException(status_code=500, detail="Failed to process and analyze the file")
+
+
 @app.post("/voice/chat")
 async def voice_chat_endpoint(
     audio: UploadFile = File(...),