Update app.py

app.py

@@ -8,19 +8,34 @@ import torch
 from transformers import pipeline
 import librosa
 import gradio as gr
-import requests
 
 # Directory to store/load cookies
 cookie_path_dir = "./cookies/"
-os.makedirs(cookie_path_dir, exist_ok=True)
+cookie_file_path = os.path.join(cookie_path_dir, "cookies_snapshot.json")  # Default file name used by hugchat
 
-# Load pre-saved cookies instead of logging in
+# Load pre-saved cookies
 try:
-    print("Attempting to load cookies from:", cookie_path_dir)
-    chatbot = hugchat.ChatBot(cookie_path_dir=cookie_path_dir)
-    print("Cookies loaded successfully.")
+    print("Attempting to load cookies from:", cookie_file_path)
+    if not os.path.exists(cookie_file_path):
+        # If cookies don't exist, attempt to generate them (for local testing; remove in Spaces)
+        EMAIL = os.environ.get("EMAIL", "[email protected]")  # Fallback for local testing
+        PASSWD = os.environ.get("PASSWORD", "e.AKsv$3Q4i4KcX")  # Fallback for local testing
+        os.makedirs(cookie_path_dir, exist_ok=True)
+        sign = Login(EMAIL, PASSWD)
+        cookies = sign.login(cookie_dir_path=cookie_path_dir, save_cookies=True)
+        print("Generated new cookies since none were found.")
+    else:
+        # Load existing cookies
+        with open(cookie_file_path, "r") as f:
+            cookies = json.load(f)  # Load the cookie dictionary
+        print("Cookies loaded from file.")
+    
+    chatbot = hugchat.ChatBot(cookies=cookies)  # Pass cookies directly
+    print("ChatBot initialized successfully.")
 except Exception as e:
-    print(f"Failed to load cookies: {str(e)}")
+    print(f"Failed to initialize ChatBot: {str(e)}")
+    import traceback
+    traceback.print_exc()
     sys.exit(1)
 
 # Model and device configuration for Whisper transcription