Update main.py

main.py

@@ -3,17 +3,12 @@ from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from hugchat import hugchat
 from hugchat.login import Login
-from dotenv import load_dotenv
-from contextlib import asynccontextmanager
 import logging
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Load environment variables
-load_dotenv()
-
 # Initialize FastAPI app
 app = FastAPI()
 
@@ -22,10 +17,6 @@ EMAIL = "[email protected]"
 PASSWORD = "Allahisgreatest17"
 ASSISTANT_ID = "682e0c1f5f0c3d952a27498e"
 
-
-# Cookie directory path
-COOKIE_PATH_DIR = "./cookies/"
-
 # Global chatbot instance
 chatbot = None
 
@@ -33,17 +24,9 @@ chatbot = None
 class PromptRequest(BaseModel):
     prompt: str
 
-# Ensure cookie directory exists
-def ensure_cookie_directory():
-    if not os.path.exists(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
+from contextlib import asynccontextmanager
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     global chatbot
@@ -54,17 +37,12 @@ async def lifespan(app: FastAPI):
                          EMAIL, "****" if PASSWORD else None, ASSISTANT_ID)
             raise ValueError("HuggingFace credentials or assistant ID not provided in environment variables")
 
-        # Ensure cookie directory exists
-        ensure_cookie_directory()
-
         # Log in to HuggingFace
         sign = Login(EMAIL, PASSWORD)
         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
+            # Try login without cookie_path_dir to avoid file system issues
+            cookies = sign.login(save_cookies=False)
+            logger.info("Successfully logged in to HuggingFace without saving cookies")
         except Exception as e:
             logger.error("Login failed: %s", str(e))
             raise
@@ -104,3 +82,8 @@ async def ask(prompt_request: PromptRequest):
         logger.error("Error processing request: %s", str(e))
         raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
 
+# Health check endpoint
+@app.get("/health")
+async def health():
+    return {"status": "healthy", "chatbot_initialized": chatbot is not None}
+