Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 from fastapi import FastAPI, HTTPException
 import google.generativeai as genai
+from pydantic import BaseModel
 
 genai.configure(api_key="AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
 
@@ -25,6 +26,8 @@ model = genai.GenerativeModel(
 
 # Start a new chat session
 chat_session = model.start_chat(history=[])
+class ChatRequest(BaseModel):
+    user_input: str
 
 app = FastAPI()
 @app.get("/")
@@ -32,8 +35,8 @@ def root():
     return {"message": "Welcome to Chatbot API"}
 
 @app.post("/chat/")
-def chat_with_bot(user_input: str):
-    if not user_input:
+def chat_with_bot(request: ChatRequest):
+    if not request.user_input:
         raise HTTPException(status_code=400, detail="Input cannot be empty")
-    response = chat_session.send_message(user_input)
+    response = chat_session.send_message(request.user_input)
     return {"response": response.text}