update app.py

app.py

@@ -9,16 +9,38 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(state:str, altitude:int)-> str: #it's import to specify the return type
+def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that fetchs the state capital and its altitude
+    """A tool that does nothing
     Args:
-        state: A string representing the state who's capital we need to identify
-        altitude: The altitude of the state capital we've identified
+        arg1: A string 
+        arg2: another string
     """
-    return f"The altitude of '{state}'s capital is '{altitude}'"
+    return f"we did nothing'"
 
+@tool
+def fetch_state_capital_and_altitude(state: str) -> str:
+    """A tool that fetches the capital and altitude of a US state using DuckDuckGo search.
+    Args:
+        state: A US state name.
+    """
+    try:
+        # Query capital
+        capital_query = f"What is the capital of {state}?"
+        capital_result = duckduckgo_tool(capital_query)
 
+        # Extract a simple capital name from the result
+        # In a real-world case, you'd want better parsing here
+        capital = capital_result[:100]  # crude truncation for demonstration
+
+        # Query altitude
+        altitude_query = f"What is the altitude of {capital}?"
+        altitude_result = duckduckgo_tool(altitude_query)
+        altitude = altitude_result[:100]  # crude again
+
+        return f"The capital of {state} is {capital}. Its altitude is {altitude}."
+    except Exception as e:
+        return f"Error fetching info for {state}: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: