Update app.py

app.py

@@ -59,9 +59,9 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 @tool
 def wolfram_alpha(query: str)-> str:
     """
-    A wrapper arounf Wolfram Alpha, nn intelligent tool that answers questions about Math, Geography,
-    Demographics, American Sports and stadiums, Music, Science, Technology, Culture, Society
-    and Everyday Life. Input should be a search query."    
+    A wrapper around Wolfram Alpha, an intelligent tool that answers questions about Math, Geography,
+    Demographics, American Sports and american sports venues, Music, Science, Technology, Culture, Society
+    and Everyday Life. Input should be a textual search query."    
     Args:
         query: The search query.
     Returns:
@@ -78,19 +78,18 @@ def wolfram_alpha(query: str)-> str:
         response = requests.get(url)
         response.raise_for_status()  # Raise an exception for HTTP errors
 
-        data = response.json()
-
-        if data.get("queryresult").get("error"):  # Check if there's an error in the response
-            return f"Error: {data['queryresult']['error'].get('msg', 'Unable to fetch Wolfram response.')}"
+        if query_result := response.json().get("queryresult") is None or query_result.get("error"): # Check if there's an error in the response
+             return f"Error: {query_result['error'].get('msg', 'Unable to fetch Wolfram response.')}"
+    
+        if pods := query_result.get("pods") is None:
+            return "Wolfram did not provide an answer."
 
-        response = ""
-        for result in data.get("queryresult").get("pods")[0].get("subpods"):
-            response += f"{result.get('plaintext')}; "
-        return response
+        return "; ".join(res.get("plaintext") for res in pods[0].get("subpods"))
         
     except requests.exceptions.RequestException as e:
         return f"Error fetching Wolfram response: {str(e)}"
 
+        
 class GoogleSearchTool(Tool):
     name = "web_search"
     description = """Performs a google web search for your query then returns a string of the top search results."""