Update app.py

app.py

@@ -4,28 +4,71 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from bs4 import BeautifulSoup
 
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def meme_generator(theme: str, character: str) -> str:
-    """A tool that generates a meme image using the given theme and character.
+from bs4 import BeautifulSoup
+
+@tool
+def get_amazon_product_price(product: str) -> str:
+    """A tool that fetches the current price of an Amazon product given its product name.
     
     Args:
-        theme: The theme of the meme (e.g., "motivational", "funny", "sarcastic").
-        character: The character to feature in the meme (e.g., "cat", "superhero").
+        product: The name of the product to search on Amazon.
     
     Returns:
-        A string representing the URL or description of the generated meme image.
+        A string indicating the current price of the product or an error message.
     """
-    prompt = f"Create a meme with a {theme} theme featuring a {character}. The meme should be humorous and visually appealing."
-    
     try:
-        image_result = image_generation_tool(prompt=prompt)
-        return image_result
+        # Use the DuckDuckGo search tool to find an Amazon page for the product
+        search_query = f"site:amazon.com {product}"
+        search_results = search_tool(query=search_query)
+        
+        # Assume search_results is a list of URLs; if not, wrap it into a list.
+        if not isinstance(search_results, list):
+            search_results = [search_results]
+            
+        amazon_url = None
+        for result in search_results:
+            if "amazon.com" in result:
+                amazon_url = result
+                break
+        
+        if not amazon_url:
+            return "Could not find an Amazon product page for the given product."
+        
+        headers = {
+            "User-Agent": (
+                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
+                "AppleWebKit/537.36 (KHTML, like Gecko) "
+                "Chrome/90.0.4430.93 Safari/537.36"
+            )
+        }
+        response = requests.get(amazon_url, headers=headers, timeout=10)
+        if response.status_code != 200:
+            return f"Error fetching the page: Received status code {response.status_code}"
+        
+        soup = BeautifulSoup(response.text, "html.parser")
+        
+        price_tag = soup.find(id="priceblock_ourprice")
+        if price_tag is None:
+            price_tag = soup.find(id="priceblock_dealprice")
+        if price_tag is None:
+            price_container = soup.find("span", class_="a-price")
+            if price_container:
+                price_tag = price_container.find("span", class_="a-offscreen")
+        
+        if price_tag:
+            price = price_tag.get_text().strip()
+            return f"The current price for '{product}' is: {price}"
+        else:
+            return "Price not found on the product page."
     except Exception as e:
-        return f"Error generating meme: {str(e)}"
+        return f"Error fetching product price: {str(e)}"
+
 
 
 @tool
@@ -65,7 +108,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, meme_generator], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_amazon_product_price], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,