Update main.py

main.py

@@ -4,6 +4,7 @@ import asyncio
 import nest_asyncio
 import requests
 from hugchat import hugchat
+from hugchat.login import Login
 
 # Import Telegram bot components
 from telegram import Update, Bot
@@ -14,7 +15,7 @@ from telegram.ext import Application, CommandHandler, MessageHandler, filters, C
 # -------------------------
 # Expected environment variables:
 # TELEGRAM_BOT_TOKEN - Telegram bot token
-# WEBHOOK_DOMAIN - example: your-space.hf.space
+# WEBHOOK_DOMAIN -  your-space.hf.space
 # HF_EMAIL and HF_PASSWORD 
 
 TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
@@ -25,6 +26,15 @@ WEBHOOK_DOMAIN = os.getenv("WEBHOOK_DOMAIN")
 if not WEBHOOK_DOMAIN:
     raise ValueError("Missing WEBHOOK_DOMAIN environment variable.")
 
+HF_EMAIL = os.getenv("HF_EMAIL")
+if not HF_EMAIL:
+    raise ValueError("Missing HF_EMAIL environment variable.")
+
+HF_PASSWORD = os.getenv("HF_PASSWORD")
+if not HF_PASSWORD:
+    raise ValueError("Missing HF_PASSWORD environment variable.")
+
+
 # -------------------------
 # Configure logging
 # -------------------------
@@ -35,23 +45,26 @@ logger = logging.getLogger(__name__)
 # Set up Hugging Chat integration
 # -------------------------
 # Logs into Hugging Face or uses a fallback to get cookies.
-if os.getenv("HF_EMAIL") and os.getenv("HF_PASSWORD"):
-    from hugchat.login import Login
-    logger.info("Logging into Hugging Face using provided credentials.")
-    sign = Login(os.getenv("HF_EMAIL"), os.getenv("HF_PASSWORD"))
-    cookies = sign.login()
-else:
-    logger.info("No HF_EMAIL/HF_PASSWORD provided. Using fallback method to get cookies.")
-    cookies = requests.get("https://huggingface.co/chat/").cookies
-
+try:
+    if os.getenv("HF_EMAIL") and os.getenv("HF_PASSWORD"):
+        from hugchat.login import Login
+        logger.info("Logging into Hugging Face using provided credentials.")
+        sign = Login(HF_EMAIL, HF_PASSWORD)
+        cookies = sign.login()
+    else:
+        logger.info("No HF_EMAIL/HF_PASSWORD provided. Using fallback method to get cookies.")
+        cookies = requests.get("https://huggingface.co/chat/").cookies
+except Exception as e:
+        logger.error(f"Error in Logging-into Hugging Chat: {e}")
+ 
 
 # Create a ChatBot instance from the hugchat library.
 chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
 
+
 # -------------------------
 # Helper Functions
 # -------------------------
-
 def detect_language(user_input: str) -> str:
     """
     Detect language using langid.
@@ -138,6 +151,7 @@ async def handle_message(update: Update, context: CallbackContext):
 # We use the bot token as the URL path for uniqueness.
 WEBHOOK_URL = f"https://{WEBHOOK_DOMAIN}/{TELEGRAM_BOT_TOKEN}"
 
+    
 # -------------------------
 # Main function to run the Telegram Bot using Webhook mode
 # -------------------------
@@ -160,6 +174,7 @@ async def main():
         webhook_url=WEBHOOK_URL     # Full webhook URL that Telegram will call.
     )
 
+
 # -------------------------
 # Run the main function (handling event loop issues)
 # -------------------------