Update app.py

app.py

@@ -5,6 +5,11 @@ import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
+import requests
+from bs4 import BeautifulSoup
+from transformers import pipeline
+
+
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -33,6 +38,45 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_productName_and_prices_from_amazon_link(Amazon_link: str) -> str:
+    """A tool that gets the product name and price from provided amazon link.
+    Args:
+        Amazon_link: Amazon product link (e.g., 'https://www.amazon.co.uk/s-Oliver-Mens-Weste-Outdoor-Gilet/dp/B0B5HCKFX4?ref_=ast_sto_dp&psc=1').
+    """
+    try:
+
+        
+        # Amazon product URL
+        url = Amazon_link
+        
+        # Set headers to mimic a real browser
+        headers = {
+            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36"
+        }
+        
+        # Send a request to the webpage
+        response = requests.get(url, headers=headers)
+        
+        # Parse the page
+        soup = BeautifulSoup(response.content, "html.parser")
+        
+        # Extract product title
+        title = soup.find("span", {"id": "productTitle"}).text.strip()
+        
+        # Extract product price (may vary based on availability)
+        price_element = soup.find("span", class_="a-offscreen")
+        price = price_element.text.strip() if price_element else "Price not available"
+        
+        # Print results
+        print(f"Product Name: {title}")
+        print(f"Price: {price}")
+        
+        return f"The Product Name is: {title} and the Price is:{price}"
+    except Exception as e:
+        return f"Error fetching Product name and price for this link{url}"
+
+
 
 final_answer = FinalAnswerTool()
 
@@ -42,7 +86,7 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -50,12 +94,26 @@ custom_role_conversions=None,
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+
+# Load a text-generation model
+text_generator = pipeline("text-generation", model="meta-llama/Llama-3-8B-Instruct")
+
+
+@tool
+def generate_text(prompt: str) -> str:
+    """A tool that generates text based on a given prompt."""
+    output = text_generator(prompt, max_length=100)
+    return output[0]["generated_text"]
+
+
+    
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer,my_custom_tool,get_current_time_in_timezone,get_productName_and_prices_from_amazon_link,image_generation_tool,generate_text], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,