Update app.py

app.py

@@ -4,6 +4,10 @@ import requests
 import inspect
 import pandas as pd
 
+# HuggingFace authentication
+from huggingface_hub import login
+import warnings
+
 # smolagents imports
 from smolagents import CodeAgent, InferenceClientModel, tool
 import re
@@ -230,21 +234,65 @@ def count_items(text: str, item_type: str = "words") -> str:
     except Exception as e:
         return f"Error counting items: {str(e)}"
 
+def setup_authentication():
+    """Setup HuggingFace authentication for the app."""
+    try:
+        # Try to get HF token from environment variables
+        hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+        
+        if hf_token:
+            login(token=hf_token)
+            print("✅ Authenticated with HuggingFace using environment token")
+            return True
+        else:
+            print("ℹ️ No HF token found in environment")
+            print("💡 If running locally, please set HF_TOKEN environment variable")
+            print("💡 For Spaces deployment, this should work automatically")
+            return False
+    except Exception as e:
+        print(f"⚠️ Authentication issue: {e}")
+        return False
+
 # --- Enhanced Agent Definition ---
 class GAIAAgent:
     def __init__(self):
         print("GAIAAgent initializing with smolagents...")
         
-        # Initialize the model (using HuggingFace Inference Client)
+        # Handle HuggingFace authentication
         try:
-            # Use a powerful model for better reasoning on GAIA benchmark
+            # Try to get HF token from environment (for Spaces)
+            hf_token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
+            if hf_token:
+                login(token=hf_token)
+                print("✅ Authenticated with HuggingFace using environment token")
+            else:
+                # In Spaces, authentication might already be handled
+                print("ℹ️ No HF token found in environment, proceeding without explicit login")
+        except Exception as e:
+            print(f"⚠️ Authentication warning: {e}")
+        
+        # Initialize the model with fallback options
+        try:
+            # Try powerful model first
             model_id = "meta-llama/Llama-3.3-70B-Instruct"
             self.model = InferenceClientModel(model_id=model_id)
             print(f"✅ Model initialized successfully: {model_id}")
         except Exception as e:
-            print(f"❌ Error initializing model: {e}")
-            # Fallback to default model
-            self.model = InferenceClientModel()
+            print(f"⚠️ Error with primary model: {e}")
+            try:
+                # Fallback to a different model
+                fallback_model = "microsoft/DialoGPT-medium"
+                self.model = InferenceClientModel(model_id=fallback_model)
+                print(f"✅ Fallback model initialized: {fallback_model}")
+            except Exception as e2:
+                print(f"⚠️ Error with fallback model: {e2}")
+                try:
+                    # Last resort - use default
+                    self.model = InferenceClientModel()
+                    print("✅ Default model initialized")
+                except Exception as e3:
+                    print(f"❌ Critical error - could not initialize any model: {e3}")
+                    raise e3
         
         # Initialize tools (custom tools + base tools from smolagents)
         self.custom_tools = [
@@ -534,6 +582,10 @@ if __name__ == "__main__":
     print("🤖 ENHANCED GAIA AGENT STARTING UP")
     print("="*60)
     
+    # Setup authentication
+    print("🔐 Setting up HuggingFace authentication...")
+    auth_success = setup_authentication()
+    
     # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")
     space_id_startup = os.getenv("SPACE_ID")
@@ -543,6 +595,10 @@ if __name__ == "__main__":
         print(f"   🌐 Runtime URL: https://{space_host_startup}.hf.space")
     else:
         print("ℹ️  SPACE_HOST environment variable not found (running locally?).")
+        if not auth_success:
+            print("💡 For local testing, you may need to run:")
+            print("   from huggingface_hub import notebook_login")
+            print("   notebook_login()")
 
     if space_id_startup:
         print(f"✅ SPACE_ID found: {space_id_startup}")