Update app.py

app.py

@@ -24,6 +24,10 @@ app.add_middleware(
 class TTSRequest(BaseModel):
     text: str
 
+class ChatRequest(BaseModel):
+    message: str
+    session_id: str = "default"
+
 
 UPLOAD_DIR = "uploads"
 
@@ -49,8 +53,9 @@ async def upload_file(file: UploadFile = File(...)):
 
 
 @app.post("/chat")
-async def chat(message: str, session_id: str = "default"):
-
+async def chat(request: ChatRequest):
+    message = request.message
+    session_id = request.session_id
     async def event_generator():
         async for chunk in app_graph.astream(
             {"messages": [HumanMessage(content=message)]},