Update app.py

app.py

@@ -43,6 +43,50 @@ from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_ty
 
 from smolagents import Tool
 
+from urllib import parse
+
+WOLFRAM_RESPONSE_KEYS = [
+    "Result",
+    "Solution",
+    "RealSolution",
+]
+
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+@tool
+def wolfram_alpha(query: str)-> str:
+    """
+    A wrapper arounf Wolfram Alpha, nn intelligent tool that answers questions about Math, Geography,
+    Demographics, Sports, Music, Science, Technology, Culture, Society
+    and Everyday Life. Input should be a search query."    
+    Args:
+        query: The search query.
+    Returns:
+        A string containing the answer for the query.
+    """
+    api_key = os.environ["WOLFRAM_ALPHA_APPID"]
+    formatted_query = parse.quote_plus(query)
+    url = f"http://api.wolframalpha.com/v2/query?appid={api_key}&input={formatted_query}&output=json&format=plaintext"
+    for key in WOLFRAM_RESPONSE_KEYS:
+        url += f"&includepodid={key}"
+    print(f"\nURL:\n{url}")
+
+    try:
+        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.')}"
+
+        response = ""
+        for result in data.get("queryresult").get("pods")[0].get("subpods"):
+            response += f"{result.get('plaintext')}; "
+        return response
+        
+    except requests.exceptions.RequestException as e:
+        return f"Error fetching Wolfram response: {str(e)}"
 
 class GoogleSearchTool(Tool):
     name = "web_search"
@@ -126,10 +170,7 @@ class GoogleSearchTool(Tool):
 
         return "## Search Results\n" + "\n\n".join(web_snippets)
 
-# web_search = GoogleSearchTool()
 
-# print(web_search(query="Donald Trump news"))
-# quit()
 AUTHORIZED_IMPORTS = [
     "requests",
     "zipfile",
@@ -185,6 +226,7 @@ ti_tool = TextInspectorTool(model, text_limit)
 browser = SimpleTextBrowser(**BROWSER_CONFIG)
 
 WEB_TOOLS = [
+    wolfram_alpha,
     GoogleSearchTool(),
     VisitTool(browser),
     PageUpTool(browser),
@@ -200,7 +242,7 @@ def create_agent():
     """Creates a fresh agent instance for each session"""
     return CodeAgent(
         model=model,
-        tools=[visualizer] + WEB_TOOLS,
+        tools=[visualizer, image_generation_tool] + WEB_TOOLS,
         max_steps=10,
         verbosity_level=1,
         additional_authorized_imports=AUTHORIZED_IMPORTS,