Upload 5 files

agent.py

@@ -66,4 +66,7 @@ class Agent:
                 agents.load_tools(["llm-math", "python_repl"], llm=llm) +
                 common_tools
               )
-      agent_chain = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True, memory=memory)
+      self.agent_chain = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True, memory=memory)
+
+    def run(self, query):
+       return self.agent_chain.run(input=query)

app.py

@@ -14,9 +14,9 @@ def agent_run(q, openai_api_key, mapi_api_key):
     os.environ["MAPI_API_KEY"]=mapi_api_key
     agent_chain = agent.Agent(openai_api_key, mapi_api_key)
     try: 
-        out = agent_chain.run(input=q)
-    except:
-        out = "Something went wrong, please try again"
+        out = agent_chain.run(q)
+    except Exception as err:
+        out = f"Something went wrong, please try again.\nError: {err}"
     return out
 
 with gr.Blocks(css=css_style) as demo: