Update endpoints.py

endpoints.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, Query,
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.encoders import jsonable_encoder
 from typing import Optional, List
-from models import ChatRequest, VoiceOutputRequest, RiskLevel
+from pydantic import BaseModel
 from auth import get_current_user
 from utils import clean_text_response
 from analysis import analyze_patient_report
@@ -15,6 +15,26 @@ from bson import ObjectId
 import asyncio
 from bson.errors import InvalidId
 
+# Define the ChatRequest model with an optional patient_id
+class ChatRequest(BaseModel):
+    message: str
+    history: Optional[List[dict]] = None
+    format: Optional[str] = "clean"
+    temperature: Optional[float] = 0.7
+    max_new_tokens: Optional[int] = 512
+    patient_id: Optional[str] = None  # Added optional patient_id field
+
+class VoiceOutputRequest(BaseModel):
+    text: str
+    language: str = "en-US"
+    slow: bool = False
+    return_format: str = "mp3"
+
+class RiskLevel(BaseModel):
+    level: str
+    score: float
+    factors: Optional[List[str]] = None
+
 def create_router(agent, logger, patients_collection, analysis_collection, users_collection):
     router = APIRouter()
 
@@ -93,7 +113,7 @@ def create_router(agent, logger, patients_collection, analysis_collection, users
                 # Store chat session in database
                 chat_entry = {
                     "user_id": current_user["email"],
-                    "patient_id": request.patient_id if request.patient_id else None,
+                    "patient_id": request.patient_id,  # Now safely optional, defaults to None
                     "message": request.message,
                     "response": cleaned_text,
                     "chat_type": "chat",