Update app.py

app.py

@@ -1,41 +1,64 @@
 import gradio as gr
-import random
-#from smolagents import GradioUI, CodeAgent, HfApiModel
-from smolagents import GradioUI, CodeAgent, OpenAIServerModel
+import pandas as pd
+import datasets
+import os
 
-
-# Import our custom tools from their modules
-#from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
+from smolagents import CodeAgent, OpenAIServerModel
 from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool, NewsSearchTool
 from retriever import load_guest_dataset
 
-# Initialize the Hugging Face model
-#model = HfApiModel()
+# Constants
+SAMPLE_FILE = "sample_guests.csv"
+
+# Generate sample dataset if not present
+def generate_sample_guest_csv():
+    if not os.path.exists(SAMPLE_FILE):
+        guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
+        df = pd.DataFrame(guest_dataset)
+        df.to_csv(SAMPLE_FILE, index=False)
+
+generate_sample_guest_csv()
+
+# Set up model
 model = OpenAIServerModel(model_id="gpt-4o")
 
-# Initialize the web search tool
+# Initialize tools
 search_tool = DuckDuckGoSearchTool()
-
-# Initialize the weather tool
 weather_info_tool = WeatherInfoTool()
-
-# Initialize the Hub stats tool
 hub_stats_tool = HubStatsTool()
+news_tool = NewsSearchTool(api_key=os.getenv("CONTEXTUALWEB_API_KEY"))  # replace env var or set directly
+
+# Wrapper to load guest info tool dynamically (from uploaded or default file)
+def build_agent_with_guest_file(file=None):
+    path = file.name if file else None
+    guest_info_tool = load_guest_dataset(file_path=path)
+    return CodeAgent(
+        tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool, news_tool],
+        model=model,
+        add_base_tools=True,
+        planning_interval=3
+    )
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧑‍💼 Gala Guest Assistant")
+    gr.Markdown("Upload a guest dataset or use the default below.")
+
+    with gr.Row():
+        gr.File(value=SAMPLE_FILE, label="📥 Download sample_guests.csv", interactive=False)
+        guest_file = gr.File(label="📤 Upload your custom .csv or .json guest file", file_types=[".csv", ".json"], type="file")
+
+    query_input = gr.Textbox(label="Ask about a guest (name or relation)")
+    output = gr.Textbox(label="Result", lines=10)
 
-# Initialize the news tool
-news_tool = NewsSearchTool(api_key="CONTEXTUALWEB_API_KEY")
+    run_button = gr.Button("Run")
 
-# Load the guest dataset and initialize the guest info tool
-guest_info_tool = load_guest_dataset()
+    def query_guest_info(query, file):
+        agent = build_agent_with_guest_file(file)
+        return agent.run(query)
 
-# Create Alfred with all the tools
-alfred = CodeAgent(
-    #tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool], 
-    tools=[guest_info_tool, weather_info_tool, hub_stats_tool, search_tool, news_tool], 
-    model=model,
-    add_base_tools=True,  # Add any additional base tools
-    planning_interval=3   # Enable planning every 3 steps
-)
+    run_button.click(fn=query_guest_info, inputs=[query_input, guest_file], outputs=output)
 
+# Launch
 if __name__ == "__main__":
-    GradioUI(alfred).launch() 
+    demo.launch()