Update main.py

main.py

@@ -1,10 +1,15 @@
-from fastapi import FastAPI, Query
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 from hugchat import hugchat
 from hugchat.login import Login
 import os
 
 app = FastAPI()
 
+# Pydantic model for request body
+class QuestionRequest(BaseModel):
+    question: str
+
 # Global variable to store the chatbot instance
 chatbot = None
 
@@ -38,33 +43,32 @@ def setup_chatbot(email, password, cookie_path, assistant_id):
 async def startup_event():
     global chatbot
     # Credentials and configuration
-    EMAIL = os.getenv("EMAIL")
-    PASSWD = os.getenv("PASSWD")
+    EMAIL = os.getenv("HUGCHAT_EMAIL", "[email protected]")
+    PASSWD = os.getenv("HUGCHAT_PASSWD", "Allahisgreatest17")
     COOKIE_PATH_DIR = "./cookies/"
-    ASSISTANT_ID = "682e0c1f5f0c3d952a27498e"  # Replace with your actual assistant ID
+    ASSISTANT_ID = "66017fca58d60bd7d5c5c26c"  # Replace with your actual assistant ID
     
     chatbot = setup_chatbot(EMAIL, PASSWD, COOKIE_PATH_DIR, ASSISTANT_ID)
 
 @app.post("/generate")
-async def generate_response(
-    prompt: str = Query(..., description="The prompt for the AI")
-):
+async def generate_response(request: QuestionRequest):
     """
-    Generates a response from the AI based on the provided prompt.
+    Generates a response from the AI based on the provided question.
     
     Args:
-        prompt (str): The user's input prompt for the AI.
+        request (QuestionRequest): JSON body containing the question.
     
     Returns:
         dict: A dictionary containing the AI's response or an error message.
     """
     global chatbot
     if chatbot is None:
-        return {"error": "Chatbot not initialized. Please try again later."}
+        raise HTTPException(status_code=500, detail="Chatbot not initialized. Please try again later.")
     
     try:
         # Generate response (non-streaming for simplicity in API context)
-        response = chatbot.chat(prompt, stream=False)
+        response = chatbot.chat(request.question, stream=False)
         return {"response": response}
     except Exception as e:
-        return {"error": f"Failed to generate response: {str(e)}"}
+        raise HTTPException(status_code=500, detail=f"Failed to generate response: {str(e)}")
+