Update app.py

app.py

@@ -5,8 +5,18 @@ import pandas as pd
 import yaml
 from tools.final_answer import FinalAnswerTool
 
-@tool
-def amazon_product_scraper(search_url: str) -> tuple[pd.DataFrame, str]:
+@tool(
+    name="amazon_product_scraper",
+    description="Scrapes Amazon search results for product titles, prices, delivery fees, and links, then returns a sorted list of products based on price.",
+    arguments=[
+        {"name": "search_url", "type": "str", "description": "The URL of the Amazon search results page."}
+    ],
+    outputs=[
+        {"name": "products", "type": "list", "description": "A list of dictionaries with product details sorted by price."},
+        {"name": "recommendation", "type": "str", "description": "The best deal based on price and delivery fees."}
+    ]
+)
+def amazon_product_scraper(search_url: str) -> list:
     """
     Scrapes Amazon search results for product titles, prices, delivery fees, and links.
     
@@ -14,9 +24,9 @@ def amazon_product_scraper(search_url: str) -> tuple[pd.DataFrame, str]:
         search_url (str): The URL of the Amazon search results page. This should be a valid Amazon search URL containing product listings.
     
     Returns:
-        tuple[pd.DataFrame, str]: A tuple containing:
-            - A Pandas DataFrame with columns: Title, Price, Delivery, Link, sorted by price.
-            - A recommendation string for the best deal.
+        list: A list containing two elements:
+            - A list of dictionaries, each with keys "Title", "Price", "Delivery", "Link", sorted by price.
+            - A string containing a recommendation for the best deal.
     """
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36"
@@ -24,7 +34,7 @@ def amazon_product_scraper(search_url: str) -> tuple[pd.DataFrame, str]:
     
     response = requests.get(search_url, headers=headers)
     if response.status_code != 200:
-        return None, "Failed to retrieve Amazon results. Amazon may be blocking requests."
+        return [], "Failed to retrieve Amazon results. Amazon may be blocking requests."
     
     soup = BeautifulSoup(response.text, 'html.parser')
     
@@ -52,16 +62,13 @@ def amazon_product_scraper(search_url: str) -> tuple[pd.DataFrame, str]:
     product_data = [p for p in product_data if p["Price"] is not None]
     product_data.sort(key=lambda x: x["Price"])
     
-    # Convert to DataFrame for better visualization
-    df = pd.DataFrame(product_data)
-    
     # Recommendation logic
-    best_deal = df.iloc[0] if not df.empty else None
+    best_deal = product_data[0] if product_data else None
     recommendation = ""
     if best_deal is not None:
         recommendation = f"Best deal: {best_deal['Title']} at ${best_deal['Price']} with {best_deal['Delivery']} (Link: {best_deal['Link']})"
     
-    return df, recommendation
+    return product_data, recommendation
 
 # Define the Agent
 final_answer = FinalAnswerTool()