Update app.py

app.py

@@ -16,30 +16,31 @@ class BasicAgent:
         self.app_id = os.getenv("WOLFARM_APP_ID")
         
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+    print(f"Agent received question (first 50 chars): {question[:50]}...")
+
         if not self.app_id:
-            print("WolfarmAlpha ID not found.")
-            return "Error : WolfarmAlpha ID not set."
+            print("❌ WolframAlpha App ID not found.")
+            return "Error: WolframAlpha App ID not set."
 
-        try : 
+        try:
             url = "http://api.wolframalpha.com/v1/result"
             params = {
-                "i" : question,
-                "appid" : self.app_id
+                "i": question,
+                "appid": self.app_id
             }
-            response = requests.get(url, params = params, timeout=10)
+            response = requests.get(url, params=params, timeout=10)
+
+            print(f"🔁 Request URL: {response.url}")
 
             if response.status_code == 200:
-                print("WolfarmAlpha Response Affirmative.")
                 return response.text
+            elif response.status_code == 501:
+                return "Error: WolframAlpha API endpoint doesn't support this query type (501)."
+            else:
+                return f"Couldn't retrieve a response from WolframAlpha. Status {response.status_code}"
 
-            else : 
-                print(f"WolfarmAlpha Status Update {response.status_code}")
-                return f"Couldn't retreive a response from WolfarmAlpha. Status{response.status_code} "
-                
         except Exception as e:
-            print(f"Execption calling WolfarmAlpha: {e}")
-            return f"Error Retrieveing WolfarmAlpha API: {e}"
+            return f"Exception occurred: {str(e)}"
         
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):