Update app.py

app.py

@@ -1,21 +1,49 @@
+import os
 import gradio as gr
-from smolagents import CodeAgent
-from smolagents.models import TransformersModel  # ← USE THIS
+from smolagents import CodeAgent, InferenceClientModel
 from retriever import guest_info_tool
 from tools import all_tools
 
-print("🎩 Starting Alfred with TransformersModel...")
+print("🎩 Starting Alfred...")
 
-# USE TransformersModel - IT EXISTS
-model = TransformersModel("google/flan-t5-small")
+# ============================================================================
+# AUTO-GET TOKEN FROM SPACES SECRETS
+# ============================================================================
+# HuggingFace Spaces automatically sets HF_TOKEN from your secrets
+# You don't need to type it in code!
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
-# Create agent
-alfred = CodeAgent(
-    tools=[guest_info_tool] + all_tools,
-    model=model
-)
+if HF_TOKEN:
+    print(f"✅ Found HF_TOKEN (first 10 chars): {HF_TOKEN[:10]}...")
+    
+    # Set for huggingface_hub library
+    os.environ["HUGGINGFACE_HUB_TOKEN"] = HF_TOKEN
+    
+    # Try InferenceClientModel
+    try:
+        model = InferenceClientModel()
+        print("✅ InferenceClientModel created with token")
+    except Exception as e:
+        print(f"❌ InferenceClientModel failed: {e}")
+        # Fallback
+        from smolagents.models import TransformersModel
+        model = TransformersModel("microsoft/phi-2")
+else:
+    print("❌ No HF_TOKEN found in environment!")
+    print("💡 Add it in Space Settings → Repository secrets")
+    
+    # Use local model instead
+    from smolagents.models import TransformersModel
+    model = TransformersModel("microsoft/phi-2")
+    print("🔄 Using local model as fallback")
+
+# ============================================================================
+# CREATE AGENT
+# ============================================================================
+all_agent_tools = [guest_info_tool] + all_tools
+alfred = CodeAgent(tools=all_agent_tools, model=model)
 
-print(f"✅ Alfred ready with {len([guest_info_tool] + all_tools)} tools")
+print(f"🤖 Alfred ready with {len(all_agent_tools)} tools")
 
 def ask_alfred(query):
     try: