Update main.py

main.py

@@ -17,12 +17,12 @@ load_dotenv()
 # Initialize FastAPI app
 app = FastAPI()
 
-# HuggingFace credentials from environment variables
 # HuggingFace credentials from environment variables
 EMAIL = "[email protected]"
 PASSWORD = "Allahisgreatest17"
 ASSISTANT_ID = "682e0c1f5f0c3d952a27498e"
 
+
 # Cookie directory path
 COOKIE_PATH_DIR = "./cookies/"
 
@@ -36,17 +36,22 @@ class PromptRequest(BaseModel):
 # Ensure cookie directory exists
 def ensure_cookie_directory():
     if not os.path.exists(COOKIE_PATH_DIR):
-        os.makedirs(COOKIE_PATH_DIR)
-        logger.info("Created cookie directory: %s", COOKIE_PATH_DIR)
+        try:
+            os.makedirs(COOKIE_PATH_DIR)
+            logger.info("Created cookie directory: %s", COOKIE_PATH_DIR)
+        except Exception as e:
+            logger.error("Failed to create cookie directory: %s", str(e))
+            raise
 
 # Lifespan handler for startup and shutdown
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    # Startup: Initialize chatbot
     global chatbot
     try:
         # Validate credentials
         if not EMAIL or not PASSWORD or not ASSISTANT_ID:
+            logger.error("Missing environment variables: EMAIL=%s, PASSWORD=%s, ASSISTANT_ID=%s", 
+                         EMAIL, "****" if PASSWORD else None, ASSISTANT_ID)
             raise ValueError("HuggingFace credentials or assistant ID not provided in environment variables")
 
         # Ensure cookie directory exists
@@ -54,17 +59,24 @@ async def lifespan(app: FastAPI):
 
         # Log in to HuggingFace
         sign = Login(EMAIL, PASSWORD)
-        cookies = sign.login(cookie_path_dir=COOKIE_PATH_DIR, save_cookies=True)
-        logger.info("Successfully logged in to HuggingFace")
+        try:
+            cookies = sign.login(cookie_path_dir=COOKIE_PATH_DIR, save_cookies=True)
+            logger.info("Successfully logged in to HuggingFace with cookie_path_dir")
+        except TypeError as e:
+            logger.warning("cookie_path_dir not supported, trying default cookie storage: %s", str(e))
+            cookies = sign.login(save_cookies=True)  # Fallback to default cookie storage
+        except Exception as e:
+            logger.error("Login failed: %s", str(e))
+            raise
 
         # Initialize chatbot
         chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
-        # Start a new conversation
         chatbot.new_conversation(assistant=ASSISTANT_ID, switch_to=True)
         logger.info("Chatbot initialized with assistant ID: %s", ASSISTANT_ID)
+
     except Exception as e:
         logger.error("Failed to initialize chatbot: %s", str(e))
-        raise HTTPException(status_code=500, detail="Failed to initialize chatbot")
+        raise HTTPException(status_code=500, detail=f"Failed to initialize chatbot: {str(e)}")
 
     yield  # Application runs here
 
@@ -79,6 +91,7 @@ app.lifespan = lifespan
 async def ask(prompt_request: PromptRequest):
     try:
         if not chatbot:
+            logger.error("Chatbot not initialized")
             raise HTTPException(status_code=500, detail="Chatbot not initialized")
 
         # Send prompt to chatbot