Update app.py

app.py

@@ -60,7 +60,7 @@ agent_executor = create_sql_agent(
     llm=llm,
     toolkit=toolkit,
     verbose=True,
-    return_intermediate_steps=True
+    agent_executor_kwargs={"return_intermediate_steps": True}
 )
 
 # Step 4: Define the callback function
@@ -74,7 +74,10 @@ def process_input():
 
             # Use the agent to get the response
             with st.spinner("Processing..."):
-                response = agent_executor(user_prompt)
+                response = agent_executor(user_prompt, return_intermediate_steps=True)
+
+            # Log the response keys for debugging
+            logging.info(f"Response keys: {response.keys()}")
 
             # Extract the final answer and the data from intermediate steps
             final_answer = response['output']
@@ -93,7 +96,7 @@ 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, columns=['Region', 'Total_Sales'])
+                df_result = pd.DataFrame(sql_result)
                 sql_result_formatted = df_result.to_markdown(index=False)
             else:
                 sql_result_formatted = "No results found."