Update app.py

app.py

@@ -40,7 +40,21 @@ weather_info_tool = WeatherInfoTool()
 
 # # Initialize the Hub stats tool
 hub_stats_tool = HubStatsTool()
+
+
+from datasets import load_dataset
+from langchain_core.documents import Document
+
+# Load the dataset from Hugging Face
+dataset = load_dataset("agents-course/unit3-invitees")["train"]
+
+# Create Document objects from the "info" column
+docs = [Document(page_content=row["description"]) for row in dataset]
+
+# Initialize the tool
 guest_info_tool = GuestInfoRetrieverTool(docs)
+
+# guest_info_tool = GuestInfoRetrieverTool(docs)
 # # Load the guest dataset and initialize the guest info tool
 latest_news_tool = LatestNewsTool()
 # Create Alfred with all the tools