Update app.py

app.py

@@ -49,11 +49,11 @@ from smolagents.gradio_ui import pull_messages_from_step, handle_agent_output_ty
 
 # login(os.getenv("HF_TOKEN"))
 
-WOLFRAM_RESPONSE_KEYS = [
-    "Result",
-    "Solution",
-    "RealSolution",
-]
+# WOLFRAM_RESPONSE_KEYS = [
+#     "Result",
+#     "Solution",
+#     "RealSolution",
+# ]
 
 image_generation_tool = load_tool("danielkorat/text-to-image", trust_remote_code=True)
 
@@ -71,8 +71,6 @@ def wolfram_alpha(query: str)-> str:
     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}"
 
     try:
         response = requests.get(url)
@@ -87,8 +85,19 @@ def wolfram_alpha(query: str)-> str:
         if (pods := query_result.get("pods")) is None:  # Check if the response is missing an answer
             return "Wolfram did not provide an answer (no result pods)."
 
-        return "; ".join(res.get("plaintext") for res in pods[0].get("subpods"))
-        
+        res = ""
+        for pod in pods:
+            res += f"{pod['title']}: "
+            for subpod in pod.get("subpods"):
+                sub_title = subpod.get('title')
+                if sub_title is not None and sub_title != "":
+                    sub_title += ": "
+                res += f"{sub_title}{subpod.get('plaintext', 'N/A')}; "
+            res += "\n"
+
+        print(f"{queryres=}")
+        return res
+                
     except requests.exceptions.RequestException as e:
         print(requests.exceptions.RequestException, e)