Update interruption method

app.py

@@ -102,9 +102,10 @@ def weighted_sample_without_replacement(population, weights, k=2):
 def load_context(set_interrupt=False):
     if set_interrupt:
         generation_interrupt.set()
-        time.sleep(0.2)
+        time.sleep(0.5)  # Give more time for interrupt to take effect
         
-    generation_interrupt.clear()
+    # DON'T clear the interrupt here - let the new inference clear it
+    # generation_interrupt.clear()  # REMOVED THIS LINE
     example = get_random_example()
     
     context_desc = example.get('processed_context_desc', '')
@@ -148,6 +149,10 @@ def generate_model_summaries(example):
         "completed": False
     }
     
+    # Clear the interrupt flag when NEW inference starts
+    generation_interrupt.clear()
+    print("Interrupt flag cleared for new inference")
+    
     if generation_interrupt.is_set():
         return result
 
@@ -180,12 +185,16 @@ def generate_model_summaries(example):
         result["model_a"] = m_a_name
         result["model_b"] = m_b_name
         
+        print(f"Starting generation with models: {m_a_name} vs {m_b_name}")
         s_a, s_b = generate_summaries(example, m_a_name, m_b_name)
         
         if not generation_interrupt.is_set():
             result["summary_a"] = s_a
             result["summary_b"] = s_b
             result["completed"] = bool(s_a and s_b)
+            print("Generation completed successfully")
+        else:
+            print("Generation was interrupted")
     except Exception as e:
         print(f"Error in generation: {e}")
         
@@ -334,7 +343,7 @@ def show_all_after_loading():
         gr.update(visible=True),  # model_section
         gr.update(visible=True),  # voting_section
         gr.update(visible=True),  # submit_button
-        gr.update(value="🔄 Try a New Question", elem_classes=["query-button"])  # update button text
+        gr.update(value="🔄 Try a New Question", elem_classes=["query-button"], interactive=True)  # update button text and re-enable
     ]
 
 with gr.Blocks(theme=gr.themes.Default(