Update app.py

app.py

@@ -33,14 +33,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+import re  # Import regular expression library
+
 @tool
 def get_stock_price(stock: str) -> str:
     """
     Fetches the current stock price using DuckDuckGo search.
-
     Args:
         stock (str): A stock ticker symbol or company name (e.g., 'AAPL', 'GOOGL').
-
     Returns:
         str: A message with the stock price or an error message.
     """
@@ -54,16 +54,35 @@ def get_stock_price(stock: str) -> str:
         # Debugging: Print results to see what is returned
         print(f"DEBUG: search results -> {search_results}")
 
-        # Check if results exist and return the first result
+        # Handle different result types
         if isinstance(search_results, list) and len(search_results) > 0:
-            return f"The current price for {stock} is: {search_results[0]}"
-
-        return f"Could not fetch stock price for {stock}. Please try again."
+            # Try to extract the stock price from the first result
+            first_result = search_results[0]
+
+            # Attempt to find a pattern like "123.45 USD" in the text
+            match = re.search(r"(\d+\.\d+)\s*(USD|EUR|CAD|GBP)", first_result, re.IGNORECASE)
+
+            if match:
+                price = match.group(1)
+                currency = match.group(2)
+                return f"The current price for {stock} is: {price} {currency}"
+            else:
+                return f"Could not extract stock price from search results for {stock}."
+        elif isinstance(search_results, str):
+            # Try to directly find a price in the string
+            match = re.search(r"(\d+\.\d+)\s*(USD|EUR|CAD|GBP)", search_results, re.IGNORECASE)
+            if match:
+                price = match.group(1)
+                currency = match.group(2)
+                return f"The current price for {stock} is: {price} {currency}"
+            else:
+               return f"Could not extract stock price from search results for {stock}."
+        else:
+            return f"No search results found for {stock}."
 
     except Exception as e:
         return f"Error fetching price for stock '{stock}': {str(e)}"
 
-
     
 final_answer = FinalAnswerTool()