Update Gradio code for 6.0 compatibility

app.py

@@ -119,7 +119,7 @@ def clear_inputs():
 # BEAUTIFUL UI LAYOUT
 # =========================
 
-# 1. Custom Theme definition (FIXED: Removed button_shadow)
+# 1. Custom Theme definition
 custom_theme = gr.themes.Soft(
     primary_hue="indigo",
     secondary_hue="blue",
@@ -146,7 +146,8 @@ custom_css = """
 .execute-btn:hover { background: linear-gradient(90deg, #4338ca, #2563eb) !important; transform: scale(1.02); box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1); }
 """
 
-with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Data Analyst") as demo:
+# 🚀 FIX 1: Removed theme and css from Blocks()
+with gr.Blocks(title="AI Data Analyst") as demo:
 
     # --- HEADER WITH REPORT LINK ---
     gr.HTML("""
@@ -216,11 +217,11 @@ with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Data Analyst") as d
                 # Using a Tabbed layout makes the right side cleaner
                 with gr.Tabs():
                     with gr.TabItem("📋 Data Table"):
+                        # 🚀 FIX 2: Removed the unsupported 'height' parameter
                         result_table = gr.Dataframe(
                             label="Query Result Table",
                             interactive=False,
-                            wrap=True,
-                            height=350 # Fixes height so UI doesn't jump aggressively
+                            wrap=True
                         )
                     with gr.TabItem("🔍 Execution Analysis"):
                         explanation = gr.Textbox(label="Agent Details", lines=10, interactive=False)
@@ -247,4 +248,5 @@ with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Data Analyst") as d
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # 🚀 FIX 3: Moved theme and css into launch()
+    demo.launch(theme=custom_theme, css=custom_css)