Update app.py

app.py

@@ -1,3 +1,10 @@
+# Import necessary libraries
+import random
+from smolagents import CodeAgent, InferenceClientModel
+
+# Import our custom tools from their modules
+from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
+
 import gradio as gr
 
 def greet(name):
@@ -11,16 +18,46 @@ demo.launch()
 # Step 3: Integrate the Tool with Alfred
 # Finally, let’s bring everything together by creating our agent and equipping it with our custom tool:
 
-from smolagents import CodeAgent, InferenceClientModel
+# from smolagents import CodeAgent, InferenceClientModel
+
+# # Initialize the Hugging Face model
+# model = InferenceClientModel()
+
+# # Create Alfred, our gala agent, with the guest info tool
+# alfred = CodeAgent(tools=[guest_info_tool, search_tool, weather_info_tool, hub_stats_tool, latest_news_tool], model=model)
+
+# # Example query Alfred might receive during the gala
+# response = alfred.run("What's the latest news about quantum computing?.")
+
+# print("🎩 Alfred's Response:")
+# print(response)
+
 
 # Initialize the Hugging Face model
 model = InferenceClientModel()
 
-# Create Alfred, our gala agent, with the guest info tool
-alfred = CodeAgent(tools=[guest_info_tool, search_tool, weather_info_tool, hub_stats_tool, latest_news_tool], model=model)
+# Initialize the web search tool
+search_tool = DuckDuckGoSearchTool()
+
+# Initialize the weather tool
+weather_info_tool = WeatherInfoTool()
 
-# Example query Alfred might receive during the gala
-response = alfred.run("What's the latest news about quantum computing?.")
+# Initialize the Hub stats tool
+hub_stats_tool = HubStatsTool()
+
+# Load the guest dataset and initialize the guest info tool
+guest_info_tool = load_guest_dataset()
+
+# Create Alfred with all the tools
+alfred = CodeAgent(
+    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
+    model=model,
+    add_base_tools=True,  # Add any additional base tools
+    planning_interval=3   # Enable planning every 3 steps
+)
+query = "I need to speak with Dr. Nikola Tesla about recent advancements in wireless energy. Can you help me prepare for this conversation?"
+response = alfred.run(query)
 
 print("🎩 Alfred's Response:")
-print(response)
+print(response)
+