update

app.py

@@ -1,17 +1,30 @@
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import Response
+from pydantic import BaseModel
 import ollama
-
 app = FastAPI()
 
 @app.get('/')
 def home():
     return "Hello world"
 
+# @app.post("/chat")
+# async def chat(query: str):
+#     if not query:
+#         return {"error": "Message cannot be empty."}
+
+#     response = ollama.chat(model="llama3", messages=[{"role": "user", "content": query}])
+#     return {"response": response['message']['content']}
+
+
+
+class ChatRequest(BaseModel):
+    query: str
+
 @app.post("/chat")
-async def chat(query: str):
-    if not query:
-        return {"error": "Message cannot be empty."}
+async def chat(request: ChatRequest):
+    if not request.query.strip():
+        raise HTTPException(status_code=400, detail="Message cannot be empty.")
 
-    response = ollama.chat(model="llama3", messages=[{"role": "user", "content": query}])
+    response = ollama.chat(model="llama3", messages=[{"role": "user", "content": request.query}])
     return {"response": response['message']['content']}