Update app.py

app.py

@@ -1,62 +1,34 @@
-import os
+# app.py - Clean version
 import gradio as gr
-from smolagents import CodeAgent, InferenceClientModel
-from retriever import guest_info_tool
-from tools import all_tools
-
-print("🎩 Starting Alfred...")
-
-# ============================================================================
-# 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")
-
-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(all_agent_tools)} tools")
-
-def ask_alfred(query):
-    try:
-        return alfred.run(query)
-    except Exception as e:
-        return f"Error: {str(e)}"
-
-demo = gr.Interface(
-    fn=ask_alfred,
-    inputs=gr.Textbox(label="Ask Alfred"),
-    outputs=gr.Textbox(label="Response"),
-    title="🎩 Alfred's Gala Assistant"
+from smolagents import GradioUI, CodeAgent, InferenceClientModel
+
+# Import our custom tools
+from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
+from retriever import load_guest_dataset
+
+# Initialize
+model = InferenceClientModel()  # Updated
+search_tool = DuckDuckGoSearchTool()
+weather_info_tool = WeatherInfoTool()
+hub_stats_tool = HubStatsTool()
+guest_tool = load_guest_dataset()
+
+# Enhance tool descriptions
+guest_tool.description = "USE THIS FOR GALA GUESTS! Database with names, relations, descriptions, emails of attendees. Examples: 'Lady Ada Lovelace', 'guest list', 'who is coming'. NEVER use web search for guests."
+
+# Create agent
+alfred = CodeAgent(
+    tools=[guest_tool, weather_info_tool, hub_stats_tool, search_tool], 
+    model=model,
+    add_base_tools=True,
+    planning_interval=3
 )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    # Quick test
+    print("Testing...")
+    response = alfred.run("Tell me about Lady Ada Lovelace at the gala")
+    print(f"Response: {response[:300]}...")
+    
+    # Launch UI
+    GradioUI(alfred).launch()