Remove warmup and add strict error handling to catch infinite loading again

app.py

@@ -108,9 +108,11 @@ def run_query(method, sample_q, custom_q, db_id):
     return final_sql, df, explanation
 
 
-def toggle_input_method(method):
+def toggle_input_method(method, current_sample):
     if method == "💡 Pick a Sample":
-        return gr.update(visible=True), gr.update(visible=False), gr.update(interactive=False)
+        # Find the DB matching the current sample (fallback to 'chinook_1')
+        db = next((db for q, db in SAMPLES if q == current_sample), "chinook_1")
+        return gr.update(visible=True), gr.update(visible=False), gr.update(value=db, interactive=False)
     else:
         return gr.update(visible=False), gr.update(visible=True), gr.update(interactive=True)
 
@@ -125,7 +127,7 @@ def load_sample(selected_question):
 def clear_inputs():
     return (
         gr.update(value="💡 Pick a Sample"),
-        gr.update(value=None, visible=True),
+        gr.update(value=SAMPLE_QUESTIONS[0], visible=True), # Reset correctly to first sample
         gr.update(value="", visible=False),
         gr.update(value="chinook_1", interactive=False),
         "", pd.DataFrame(), "" 
@@ -189,6 +191,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Text-to-SQL RLHF") as demo:
 
             sample_dropdown = gr.Dropdown(
                 choices=SAMPLE_QUESTIONS,
+                value=SAMPLE_QUESTIONS[0], # Added default value here
                 label="Select a Sample Question",
                 info="The database will be selected automatically.",
                 visible=True
@@ -225,7 +228,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Text-to-SQL RLHF") as demo:
             explanation = gr.Textbox(label="AI Explanation + Execution Details", lines=8)
 
     # EVENT LISTENERS
-    input_method.change(fn=toggle_input_method, inputs=[input_method], outputs=[sample_dropdown, custom_question, db_id])
+    
+    # Updated listener to pass sample_dropdown
+    input_method.change(
+        fn=toggle_input_method, 
+        inputs=[input_method, sample_dropdown], 
+        outputs=[sample_dropdown, custom_question, db_id]
+    )
+    
     sample_dropdown.change(fn=load_sample, inputs=[sample_dropdown], outputs=[db_id])
     db_id.change(fn=update_schema, inputs=[db_id], outputs=[schema_display])