Update app.py

app.py

@@ -48,7 +48,7 @@ def get_stock_price(stock: str) -> str:
         search_tool = DuckDuckGoSearchTool()
         
         # Perform the search query
-        search_results = search_tool.run(f"Current stock price of {stock}")
+        search_results = search_tool.run(f"what is the current price of {stock}?")
         
         # If search_results is a list, extract the first relevant result
         if isinstance(search_results, list) and len(search_results) > 0:
@@ -58,6 +58,32 @@ def get_stock_price(stock: str) -> str:
     
     except Exception as e:
         return f"Error fetching price for stock '{stock}': {str(e)}"
+
+@tool
+def web_search(query: str) -> str:
+    """Fetches the result from a web search.
+    
+    Args:
+        stock: A string representing a query or question.
+    
+    Returns:
+        A string with the answer of the search.
+    """
+    try:
+        # Initialize the search tool
+        search_tool = DuckDuckGoSearchTool()
+        
+        # Perform the search query
+        search_results = search_tool.run(query)
+        
+        # If search_results is a list, extract the first relevant result
+        if isinstance(search_results, list) and len(search_results) > 0:
+            return {search_results[0]}"
+        
+        return f"Could not find {stock}. Please try again."
+    
+    except Exception as e:
+        return f"Error finding '{stock}': {str(e)}"