Update app.py

app.py

@@ -1,33 +1,19 @@
-import gradio as gr
-import random
-from smolagents import GradioUI, CodeAgent, HfApiModel
-
-# Import our custom tools from their modules
-from tools import DuckDuckGoSearchTool, WeatherInfoTool, HubStatsTool
-from retriever import load_guest_dataset
-
-# Initialize the Hugging Face model
-model = HfApiModel()
-
-# Initialize the web search tool
-search_tool = DuckDuckGoSearchTool()
-
-# Initialize the weather tool
-weather_info_tool = WeatherInfoTool()
-
-# 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
-)
-
-if __name__ == "__main__":
-    GradioUI(alfred).launch()
+import datasets
+from langchain_core.documents import Document
+
+# Load the dataset
+guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
+
+# Convert dataset entries into Document objects
+docs = [
+    Document(
+        page_content="\n".join([
+            f"Name: {guest['name']}",
+            f"Relation: {guest['relation']}",
+            f"Description: {guest['description']}",
+            f"Email: {guest['email']}"
+        ]),
+        metadata={"name": guest["name"]}
+    )
+    for guest in guest_dataset
+]