Update app.py

app.py

@@ -72,12 +72,9 @@ def process_input():
             # Append user message to history
             st.session_state.history.append({"role": "user", "content": user_prompt})
 
-            # Use the agent to get the response
+            # Use the agent to get the response WITHOUT 'return_intermediate_steps'
             with st.spinner("Processing..."):
-                response = agent_executor(user_prompt, return_intermediate_steps=True)
-
-            # Log the response keys for debugging
-            logging.info(f"Response keys: {response.keys()}")
+                response = agent_executor(user_prompt)  # Removed 'return_intermediate_steps' here
 
             # Extract the final answer and the data from intermediate steps
             final_answer = response['output']
@@ -95,7 +92,6 @@ def process_input():
 
             # Convert the result to a DataFrame for better formatting
             if sql_result:
-                # Adjust the column names based on your query
                 df_result = pd.DataFrame(sql_result)
                 sql_result_formatted = df_result.to_markdown(index=False)
             else:
@@ -133,6 +129,9 @@ def process_input():
         # Reset user input
         st.session_state['user_input'] = ''
 
+ 
+
+
 # Step 5: Display conversation history
 for message in st.session_state.history:
     if message['role'] == 'user':