Update app.py

app.py

@@ -59,9 +59,10 @@ def get_stock_price(stock: str) -> str:
     except Exception as e:
         return f"Error fetching price for stock '{stock}': {str(e)}"
 
+from typing import Dict
 
 @tool
-def web_search(query: str) -> str:
+def web_search(query: str) -> Dict[str, str]:
     """
     Fetches the result from a web search using DuckDuckGo.
 
@@ -69,19 +70,19 @@ def web_search(query: str) -> str:
         query (str): A search query or question.
 
     Returns:
-        str: The top search result as a string.
+        Dict[str, str]: A dictionary with either the result or an error message.
     """
     try:
         search_tool = DuckDuckGoSearchTool()
         search_results = search_tool.run(query)
 
         if isinstance(search_results, list) and len(search_results) > 0:
-            return f"Top result for '{query}': {search_results[0]}"
+            return {"query": query, "result": search_results[0]}
 
-        return f"No results found for '{query}'."
+        return {"query": query, "error": "No results found."}
 
     except Exception as e:
-        return f"Error during search: {str(e)}"
+        return {"query": query, "error": f"Error during search: {str(e)}"}
 
     
 final_answer = FinalAnswerTool()