Restore proper Python formatting for app.py

app.py

@@ -35,32 +35,42 @@ SAMPLE_QUESTIONS = [q[0] for q in SAMPLES]
 # SQL EXPLAINER
 # =========================
 def explain_sql(sql):
+
     explanation = "This SQL query retrieves information from the database."
+
     sql_lower = sql.lower()
 
     if "join" in sql_lower:
         explanation += "\n• It combines data from multiple tables using JOIN."
+
     if "where" in sql_lower:
         explanation += "\n• It filters rows using a WHERE condition."
+
     if "group by" in sql_lower:
         explanation += "\n• It groups results using GROUP BY."
+
     if "order by" in sql_lower:
         explanation += "\n• It sorts the results using ORDER BY."
+
     if "limit" in sql_lower:
         explanation += "\n• It limits the number of returned rows."
 
     return explanation
 
+
 # =========================
 # CORE FUNCTIONS
 # =========================
 def run_query(question, db_id):
+
     if not question.strip():
-        return "", None, "⚠️ Please enter a question."
+        return "", None, " Please enter a question."
 
     start_time = time.time()
+
     result = engine.ask(question, db_id)
     final_sql = result["sql"]
+
     end_time = time.time()
     latency = round(end_time - start_time, 3)
 
@@ -71,22 +81,29 @@ def run_query(question, db_id):
     # ZERO ROW HANDLING
     if not result["rows"]:
         df = pd.DataFrame(columns=result.get("columns", []))
-        explanation = f"""✅ Query executed successfully
+
+        explanation = f"""
+✅ Query executed successfully
 
 Rows returned: 0
+
 Note: The query ran perfectly but no matching records exist.
 
 Execution Time: {latency} sec
 
-{explain_sql(final_sql)}"""
+{explain_sql(final_sql)}
+"""
+
         return final_sql, df, explanation
 
     df = pd.DataFrame(result["rows"], columns=result["columns"])
     actual_rows = len(result["rows"])
 
-    explanation = f"""✅ Query executed successfully
+    explanation = f"""
+✅ Query executed successfully
 
 Rows returned: {actual_rows}
+
 Execution Time: {latency} sec
 
 {explain_sql(final_sql)}
@@ -98,6 +115,7 @@ This shows the model understood:
 """
 
     limit_match = re.search(r'LIMIT\s+(\d+)', final_sql, re.IGNORECASE)
+
     if limit_match:
         requested_limit = int(limit_match.group(1))
         if actual_rows < requested_limit:
@@ -105,60 +123,32 @@ This shows the model understood:
 
     return final_sql, df, explanation
 
+
 def load_sample(selected_question):
+
     if not selected_question:
         return gr.update(), gr.update()
+
     db = next((db for q, db in SAMPLES if q == selected_question), "chinook_1")
+
     return gr.update(value=selected_question), gr.update(value=db)
 
+
 def clear_inputs():
     return gr.update(value=None), gr.update(value=""), gr.update(value="chinook_1"), "", None, ""
 
 
 # =========================
-# BEAUTIFUL UI LAYOUT
+# UI LAYOUT
 # =========================
+with gr.Blocks(theme=gr.themes.Soft(), title="Text-to-SQL RLHF") as demo:
 
-# 1. Custom Theme definition
-custom_theme = gr.themes.Soft(
-    primary_hue="indigo",
-    secondary_hue="blue",
-    neutral_hue="slate",
-    font=[gr.themes.GoogleFont("Inter"), "ui-sans-serif", "system-ui", "sans-serif"]
-).set(
-    body_background_fill="*neutral_50",
-    block_background_fill="white",
-    block_border_width="1px",
-    block_border_color="*neutral_200",
-    block_shadow="*shadow_drop_sm"
-)
-
-# 2. Custom CSS for animations, highlights, and layout limits
-custom_css = """
-.gradio-container { max-width: 1300px !important; margin: auto; }
-.header-text { text-align: center; margin-bottom: 2rem; }
-.header-text h1 { color: #1e293b; font-weight: 800; font-size: 2.5rem; margin-bottom: 0.5rem; }
-.header-text p { color: #64748b; font-size: 1.1rem; margin-bottom: 1.5rem; }
-.report-btn { display: inline-block; padding: 10px 24px; background-color: #f8fafc; color: #4338ca; border: 1px solid #cbd5e1; border-radius: 8px; font-weight: bold; text-decoration: none; transition: all 0.2s; box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.05); }
-.report-btn:hover { background-color: #f1f5f9; border-color: #94a3b8; transform: translateY(-1px); }
-.highlight-notice { background-color: #e0e7ff; color: #3730a3; padding: 12px 16px; border-radius: 8px; font-weight: 500; font-size: 0.95rem; border-left: 4px solid #4f46e5; margin-bottom: 1rem; box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.05);}
-.execute-btn { background: linear-gradient(90deg, #4f46e5, #3b82f6) !important; color: white !important; border: none !important; font-weight: bold !important; transition: all 0.2s ease-in-out !important; }
-.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); }
-"""
-
-# 🚀 FIX 1: Removed theme and css from Blocks()
-with gr.Blocks(title="AI Data Analyst") as demo:
-
-    # --- HEADER WITH REPORT LINK ---
-    gr.HTML("""
-        <div class="header-text">
-            <h1>🧠 AI Text-to-SQL Agent</h1>
-            <p>Powered by RLHF & Execution Rewards. Convert natural language to strictly validated SQLite queries.</p>
-            <a href="https://tjhalanigrid.github.io/Text2SQL_Project/" target="_blank" class="report-btn">
-                📄 View the Full Project Report
-            </a>
-        </div>
-    """)
+    gr.Markdown(
+        """
+        # Text-to-SQL using RLHF + Execution Reward
+        Convert Natural Language to SQL, strictly validated and safely executed on local SQLite databases.
+        """
+    )
 
     DBS = sorted([
         "flight_1", "student_assessment", "store_1", "bike_1", "book_2", "chinook_1",
@@ -171,64 +161,53 @@ with gr.Blocks(title="AI Data Analyst") as demo:
     ])
 
     with gr.Row():
-        
-        # --- LEFT COLUMN (Inputs) ---
-        with gr.Column(scale=4):
-            with gr.Group():  # Wraps inputs in a nice white card
-                gr.Markdown("### ⚙️ Query Configuration")
-                
-                # Highlight Notice for Auto-Select
-                gr.HTML("""
-                    <div class="highlight-notice">
-                        ✨ <strong>Pro Tip:</strong> Select a sample question below and the correct database will be <strong>automatically selected</strong> for you!
-                    </div>
-                """)
-
-                sample_dropdown = gr.Dropdown(
-                    choices=SAMPLE_QUESTIONS,
-                    label="💡 Quick Select a Sample Question",
-                    show_label=True
-                )
-                
-                db_id = gr.Dropdown(
-                    choices=DBS,
-                    value="chinook_1",
-                    label="🗄️ Target Database",
-                    interactive=True
-                )
-
-                question = gr.Textbox(
-                    label="💬 Ask a Question",
-                    placeholder="Type your own question or select a sample above...",
-                    lines=4
-                )
-
-                with gr.Row():
-                    clear_btn = gr.Button("🗑️ Clear", variant="secondary", size="lg")
-                    run_btn = gr.Button("🚀 Generate & Execute SQL", elem_classes="execute-btn", size="lg")
-
-        # --- RIGHT COLUMN (Outputs) ---
-        with gr.Column(scale=7):
-            with gr.Group(): # Wraps outputs in a nice white card
-                gr.Markdown("### 📊 Execution Results")
-                
-                final_sql = gr.Code(language="sql", label="Generated SQL Script", interactive=False)
-                
-                # 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
-                        )
-                    with gr.TabItem("🔍 Execution Analysis"):
-                        explanation = gr.Textbox(label="Agent Details", lines=10, interactive=False)
-
-    # =========================
-    # EVENT LISTENERS
-    # =========================
+
+        with gr.Column(scale=1):
+
+            gr.Markdown("### 1. Configuration & Input")
+
+            sample_dropdown = gr.Dropdown(
+                choices=SAMPLE_QUESTIONS,
+                label=" Quick Select a Sample Question",
+                info="Picking a question will automatically select the right database!"
+            )
+
+            gr.Markdown("---")
+
+            db_id = gr.Dropdown(
+                choices=DBS,
+                value="chinook_1",
+                label="Select Database",
+                interactive=True
+            )
+
+            question = gr.Textbox(
+                label="Ask a Question",
+                placeholder="Type your own question or select a sample above...",
+                lines=3
+            )
+
+            with gr.Row():
+                clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+                run_btn = gr.Button(" Generate & Run SQL", variant="primary")
+
+        with gr.Column(scale=2):
+
+            gr.Markdown("### 2. Execution Results")
+
+            final_sql = gr.Code(language="sql", label="Final Executed SQL")
+
+            result_table = gr.Dataframe(
+                label="Query Result Table",
+                interactive=False,
+                wrap=True
+            )
+
+            explanation = gr.Textbox(
+                label="AI Explanation + Execution Details",
+                lines=8
+            )
+
     sample_dropdown.change(
         fn=load_sample,
         inputs=[sample_dropdown],
@@ -248,5 +227,4 @@ with gr.Blocks(title="AI Data Analyst") as demo:
     )
 
 if __name__ == "__main__":
-    # 🚀 FIX 3: Moved theme and css into launch()
-    demo.launch(theme=custom_theme, css=custom_css)
+    demo.launch()