update app.py

app.py

@@ -3,13 +3,14 @@ import datetime
 import requests
 import pytz
 import yaml
+import re
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 @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.
+    """A tool that fetches the capital and altitude of a US state using DuckDuckGo search and regex.
     Args:
         state: A US state name.
     """
@@ -18,14 +19,17 @@ def fetch_state_capital_and_altitude(state: str) -> str:
         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
+        # Attempt to extract capital using a regex pattern
+        capital_match = re.search(r"The capital of [\w\s]+ is ([A-Z][a-zA-Z\s]+)", capital_result)
+        capital = capital_match.group(1).strip() if capital_match else "Unknown"
 
         # Query altitude
         altitude_query = f"What is the altitude of {capital}?"
         altitude_result = duckduckgo_tool(altitude_query)
-        altitude = altitude_result[:100]  # crude again
+
+        # Attempt to extract altitude (look for digits followed by meters or feet)
+        altitude_match = re.search(r"(\d{2,5})\s?(meters|feet|m|ft)", altitude_result, re.IGNORECASE)
+        altitude = f"{altitude_match.group(1)} {altitude_match.group(2)}" if altitude_match else "Unknown"
 
         return f"The capital of {state} is {capital}. Its altitude is {altitude}."
     except Exception as e:
@@ -68,7 +72,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, fetch_state_capital_and_altitude], ## add your tools here (don't remove final answer)
+    tools=[final_answer, fetch_state_capital_and_altitude, duckduckgo_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,