Update app.py

app.py

@@ -186,11 +186,15 @@ def echo(message, history):
     return error_message, error_message
 
 def echo_agent(message, history):
-  message="There are 2 df's. If you find a KeyError check for the same in the other df." + "<br>" + message
   try:
-    with io.StringIO() as buffer:
-      with contextlib.redirect_stdout(buffer):
-        result= agent.chat(message)
+    response = agent.chat(query, output_type= 'text')
+    explanation = agent.explain()
+
+    result = "Answer: \n"
+       + '\n' + response + '\n'
+       + '\n' + "Explanation: \n"
+       + '\n' + explanation
+    
     return result
   except Exception as e:
     error_message = f"An error occurred: {e}"+str(e.with_traceback) + str(e.args)