Update app.py

app.py

@@ -125,7 +125,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         with gr.Column(scale=2):
             gr.Markdown("### 📊 Query Results")
-            results_out = gr.DataFrame(label="", interactive=False, row_count=10, max_rows=100)
+            results_out = gr.DataFrame(label="", interactive=False, row_count=10)
             status_info = gr.Markdown("", visible=False, elem_id="status_info")
 
     # =========================
@@ -153,11 +153,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     def handle_example_click(example_query):
         sql_query, error = parse_query(example_query)
         if error:
-            return "", f"❌ {error}", None, f"❌ {error}"
+            return "", f"❌ {error}", None
         result_df, exec_error = execute_sql_query(sql_query)
         if exec_error:
-            return sql_query, f"❌ {exec_error}", None, f"❌ {exec_error}"
-        return sql_query, "", result_df, ""
+            return sql_query, f"❌ {exec_error}", None
+        return sql_query, "", result_df
 
     # =========================
     # Button Click Event Handlers
@@ -176,27 +176,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
     # Assign click events to example buttons
-    for btn in example_buttons:
-        btn.click(
-            fn=handle_example_click,
-            inputs=None,  # We'll pass the query via lambda
-            outputs=[sql_query_out, error_out, results_out, error_out],
-            _js=None,
-            # Use a lambda to pass the button's value as the example_query
-            # Note: Gradio passes the button's value as the first input if inputs are not specified
-            # To ensure the correct query is passed, we use a lambda with no inputs
-            # and retrieve the button's value within the function
-            # However, Gradio does not support passing the button's value directly here
-            # So, we need to create a partial function or use lambda with default argument
-            # Here, we use a lambda with a default argument
-        )
-
-    # To properly assign each button's click event with its specific query, redefine the loop:
     for btn, query in zip(example_buttons, example_queries):
         btn.click(
             fn=lambda q=query: handle_example_click(q),
             inputs=None,
-            outputs=[sql_query_out, error_out, results_out, error_out]
+            outputs=[sql_query_out, error_out, results_out]
         )
 
 # Launch the Gradio App