Gercho312 commited on
Commit
8ef3396
·
verified ·
1 Parent(s): 9a8c112

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -9,14 +9,25 @@ 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_cutom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
  #Keep this format for the description / args / 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:
 
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
12
+ def city_stats(city:str)-> str: #it's import to specify the return type
13
  #Keep this format for the description / args / args description but feel free to modify the tool
14
+
15
+ """A tool that fetches stats about the specified city.
16
  Args:
17
+ city: a string representing the name of the city.
 
18
  """
19
+ api_url = 'https://api.api-ninjas.com/v1/city?name={}'.format(city)
20
+ response = requests.get(api_url, headers={'X-Api-Key': 'YOUR API KEY'})
21
+ if response.status_code == requests.codes.ok:
22
+ # Automatically parse the JSON response
23
+ data = response.json()
24
+ country = data[0]["country"]
25
+ population = data[0]["population"]
26
+ else:
27
+ print("Error:", response.status_code)
28
+
29
+
30
+ return f"{city} is part of {country} and has a population of {population}"
31
 
32
  @tool
33
  def get_current_time_in_timezone(timezone: str) -> str: