Update app.py

app.py

@@ -85,12 +85,12 @@ class BasicAgent:
             additional_authorized_imports=["requests", "bs4", "json"]
         )
 
-    def run(self, input_text):
-        # Explicitly call run() method of agent instead of invoking the agent directly
-        if hasattr(self.agent, 'run'):
-            return self.agent.run(input_text)  # Call the `run` method of CodeAgent
-        else:
-            raise AttributeError("The agent does not have a 'run' method.")
+    def __call__(self, question: str) -> str:
+        try:
+            response = self.agent.run(question)
+            return response
+        except Exception as e:
+            return f"An error occurred: {e}"
         
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """