Update main.py

main.py

@@ -1,12 +1,26 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from hugchat import hugchat
 from hugchat.login import Login
 import os
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+# Enable CORS for client compatibility
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Pydantic model for request body
 class QuestionRequest(BaseModel):
     question: str
@@ -28,15 +42,23 @@ def setup_chatbot(email, password, cookie_path, assistant_id):
         hugchat.ChatBot: Configured chatbot instance
     """
     try:
+        logger.info("Setting up chatbot...")
         # Create cookie directory if it doesn't exist
         os.makedirs(cookie_path, exist_ok=True)
         
+        logger.info("Logging in with Hugging Face credentials...")
         sign = Login(email, password)
         cookies = sign.login(cookie_dir_path=cookie_path, save_cookies=True)
+        
+        logger.info("Initializing ChatBot...")
         chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
+        
+        logger.info(f"Creating new conversation with assistant ID: {assistant_id}")
         chatbot.new_conversation(assistant=assistant_id, switch_to=True)
+        logger.info("Chatbot setup complete.")
         return chatbot
     except Exception as e:
+        logger.error(f"Failed to set up chatbot: {str(e)}")
         raise Exception(f"Failed to set up chatbot: {e}")
 
 # Initialize chatbot at startup
@@ -49,7 +71,10 @@ async def startup_event():
     COOKIE_PATH_DIR = "./cookies/"
     ASSISTANT_ID = "682e0c1f5f0c3d952a27498e"  # Replace with your actual assistant ID
     
-    chatbot = setup_chatbot(EMAIL, PASSWD, COOKIE_PATH_DIR, ASSISTANT_ID)
+    try:
+        chatbot = setup_chatbot(EMAIL, PASSWD, COOKIE_PATH_DIR, ASSISTANT_ID)
+    except Exception as e:
+        logger.error(f"Startup error: {str(e)}")
 
 @app.post("/generate")
 async def generate_response(request: QuestionRequest):
@@ -64,22 +89,36 @@ async def generate_response(request: QuestionRequest):
     """
     global chatbot
     if chatbot is None:
-        raise HTTPException(status_code=500, detail="Chatbot not initialized. Please try again later.")
+        logger.error("Chatbot not initialized.")
+        async def generate_error():
+            error_message = "Error: Chatbot not initialized. Please try again later."
+            yield error_message.encode('utf-8')
+        return StreamingResponse(generate_error(), media_type="text/plain")
     
     async def generate():
         try:
+            logger.info(f"Processing question: {request.question}")
             # Use streaming to match client expectations
             for token in chatbot.chat(request.question, stream=True):
                 # Handle hugchat response (token may be dict or string)
                 if isinstance(token, dict):
-                    # Extract token from dictionary (e.g., {"type": "stream", "token": "text"})
                     token_text = token.get("token", "")
                 else:
                     token_text = str(token)
                 
-                # Encode and yield the token
-                yield token_text.encode('utf-8')
+                if token_text:
+                    logger.debug(f"Streaming token: {token_text}")
+                    yield token_text.encode('utf-8')
+                else:
+                    logger.warning("Empty token received.")
+            
+            # Fallback response if no tokens were yielded
+            if not any(True for _ in chatbot.chat(request.question, stream=True)):
+                fallback = "Welcome, valued client! How can Abdullah Ali and our premium team bring your vision to life with a custom website ($200–$600) or AI chatbot ($119)?"
+                logger.info("Using fallback response.")
+                yield fallback.encode('utf-8')
         except Exception as e:
+            logger.error(f"Failed to generate response: {str(e)}")
             error_message = f"Error: Failed to generate response: {str(e)}"
             yield error_message.encode('utf-8')