Add a new POST endpoint for chat in main.py to handle user queries and return responses from the graph model.

main.py

@@ -123,6 +123,14 @@ async def chat(request: Query):
     # response = response[4:]
     return {"response": response}
 
+@app.post("/chat")
+async def chat(request: Query):
+    response = graph.invoke({"question": request.question})
+    response = response["answer"]
+    response = re.sub(r'<think>.*?</think>', '', response, flags=re.DOTALL)
+    # response = response[4:]
+    return {"response": response}
+
 @app.get("/")
 async def root():
     return {"message": "Hello World"}