Feature: Add SQL query, results, and analysis to visualization responses

app.py

@@ -271,25 +271,47 @@ def process_text_query(query, history):
             # Close the connection
             conn.close()
             
-            # Generate visualization if requested
-            if is_visualization:
-                viz_html = generate_visualization(result_df, query)
-                if viz_html:
-                    # Add the visualization to history
-                    history[-1][1] = viz_html
-                    return viz_html, history
+            # Format the dataframe as a string table for display
+            df_str = result_df.to_string()
             
-            # If no visualization or visualization failed, generate text response
+            # Generate text response
             data_summary = result_df.to_string()
-            response = llm.invoke(interpret_prompt.format(
+            analysis = llm.invoke(interpret_prompt.format(
                 question=query,
                 sql_query=sql_query,
                 data_summary=data_summary
             )).content
             
-            # Add the response to history
-            history[-1][1] = response
-            return response, history
+            # Create a comprehensive response that includes:
+            # 1. SQL Query
+            # 2. Results as a table
+            # 3. Analysis of the results
+            comprehensive_response = f"""
+### SQL Query:
+```sql
+{sql_query}
+```
+
+### Results:
+```
+{df_str}
+```
+
+### Analysis:
+{analysis}
+"""
+            
+            # Generate visualization if requested
+            if is_visualization:
+                viz_html = generate_visualization(result_df, query)
+                if viz_html:
+                    # Add the visualization to history
+                    history[-1][1] = comprehensive_response
+                    return viz_html, history
+            
+            # If no visualization or visualization failed, return text response
+            history[-1][1] = comprehensive_response
+            return comprehensive_response, history
             
         except Exception as e:
             error_msg = f"Error processing query: {str(e)}"
@@ -904,6 +926,7 @@ with gr.Blocks(title="LLM Powered Database Chatbot") as demo:
             # Update the visualization state
             current_viz = viz_html
         
+        # Return the updated state
         return "", new_history, current_viz
     
     # Update the button click handlers