alexub0103 commited on
Commit
f33c9ea
·
verified ·
1 Parent(s): de29c46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -35
app.py CHANGED
@@ -4,72 +4,54 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
- import requests
8
- from bs4 import BeautifulSoup
9
 
10
  from Gradio_UI import GradioUI
11
 
12
- # Below is an example of a tool that scrapes product data from Meira 18K
13
  @tool
14
- def scrape_meira18k_product_info(url: str) -> str:
15
- """A tool to scrape product names and prices from Meira 18K website
16
-
17
  Args:
18
- url: The URL of the product page to scrape.
19
-
20
- Returns:
21
- str: A formatted string with product names and prices.
22
  """
23
- try:
24
- response = requests.get(url, timeout=10) # Added timeout for better error handling
25
- response.raise_for_status() # Raises HTTPError for bad responses (4xx, 5xx)
26
- except requests.exceptions.RequestException as e:
27
- return f"Error fetching the page: {e}"
28
-
29
- soup = BeautifulSoup(response.text, 'html.parser')
30
-
31
- products = soup.find_all('h2', class_='product-title') # Adjust class as needed
32
- prices = soup.find_all('span', class_='price') # Adjust class as needed
33
-
34
- result = "\n".join([f"{p.get_text().strip()} - {pr.get_text().strip()}" for p, pr in zip(products, prices)])
35
- return result if result else "No product data found."
36
 
37
- # Time tool (unchanged)
38
  @tool
39
  def get_current_time_in_timezone(timezone: str) -> str:
40
  """A tool that fetches the current local time in a specified timezone.
41
-
42
  Args:
43
  timezone: A string representing a valid timezone (e.g., 'America/New_York').
44
  """
45
  try:
 
46
  tz = pytz.timezone(timezone)
 
47
  local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
48
  return f"The current local time in {timezone} is: {local_time}"
49
  except Exception as e:
50
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
51
 
52
- final_answer = FinalAnswerTool()
53
 
54
- # Model initialization (same as your code)
55
  model = HfApiModel(
56
- max_tokens=2096,
57
- temperature=0.5,
58
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct', # it is possible that this model may be overloaded
59
- custom_role_conversions=None,
60
  )
61
 
 
62
  # Import tool from Hub
63
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
64
 
65
- # Load prompt templates (same as your code)
66
  with open("prompts.yaml", 'r') as stream:
67
  prompt_templates = yaml.safe_load(stream)
68
 
69
- # Define the agent with the correct tools
70
  agent = CodeAgent(
71
  model=model,
72
- tools=[final_answer, scrape_meira18k_product_info], # Added the scrape tool here
73
  max_steps=6,
74
  verbosity_level=1,
75
  grammar=None,
@@ -79,5 +61,5 @@ agent = CodeAgent(
79
  prompt_templates=prompt_templates
80
  )
81
 
82
- # Launch the Gradio UI
83
  GradioUI(agent).launch()
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
 
7
 
8
  from Gradio_UI import GradioUI
9
 
10
+ # Below is an example of a tool that does nothing. Amaze us with your creativity!
11
  @tool
12
+ def my_custom_tool(arg1:str, arg2:int)-> str: # it's important to specify the return type
13
+ # Keep this format for the tool description / args description but feel free to modify the tool
14
+ """A tool that does nothing yet
15
  Args:
16
+ arg1: the first argument
17
+ arg2: the second argument
 
 
18
  """
19
+ return "What magic will you build ?"
 
 
 
 
 
 
 
 
 
 
 
 
20
 
 
21
  @tool
22
  def get_current_time_in_timezone(timezone: str) -> str:
23
  """A tool that fetches the current local time in a specified timezone.
 
24
  Args:
25
  timezone: A string representing a valid timezone (e.g., 'America/New_York').
26
  """
27
  try:
28
+ # Create timezone object
29
  tz = pytz.timezone(timezone)
30
+ # Get current time in that timezone
31
  local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
32
  return f"The current local time in {timezone} is: {local_time}"
33
  except Exception as e:
34
  return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
 
 
36
 
37
+ final_answer = FinalAnswerTool()
38
  model = HfApiModel(
39
+ max_tokens=2096,
40
+ temperature=0.5,
41
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
42
+ custom_role_conversions=None,
43
  )
44
 
45
+
46
  # Import tool from Hub
47
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
48
 
 
49
  with open("prompts.yaml", 'r') as stream:
50
  prompt_templates = yaml.safe_load(stream)
51
 
 
52
  agent = CodeAgent(
53
  model=model,
54
+ tools=[final_answer], # add your tools here (don't remove final_answer)
55
  max_steps=6,
56
  verbosity_level=1,
57
  grammar=None,
 
61
  prompt_templates=prompt_templates
62
  )
63
 
64
+
65
  GradioUI(agent).launch()