aizip-dev commited on
Commit
8b6bcfe
·
verified ·
1 Parent(s): bb6bbaf

Update interruption method

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -102,9 +102,10 @@ def weighted_sample_without_replacement(population, weights, k=2):
102
  def load_context(set_interrupt=False):
103
  if set_interrupt:
104
  generation_interrupt.set()
105
- time.sleep(0.2)
106
 
107
- generation_interrupt.clear()
 
108
  example = get_random_example()
109
 
110
  context_desc = example.get('processed_context_desc', '')
@@ -148,6 +149,10 @@ def generate_model_summaries(example):
148
  "completed": False
149
  }
150
 
 
 
 
 
151
  if generation_interrupt.is_set():
152
  return result
153
 
@@ -180,12 +185,16 @@ def generate_model_summaries(example):
180
  result["model_a"] = m_a_name
181
  result["model_b"] = m_b_name
182
 
 
183
  s_a, s_b = generate_summaries(example, m_a_name, m_b_name)
184
 
185
  if not generation_interrupt.is_set():
186
  result["summary_a"] = s_a
187
  result["summary_b"] = s_b
188
  result["completed"] = bool(s_a and s_b)
 
 
 
189
  except Exception as e:
190
  print(f"Error in generation: {e}")
191
 
@@ -334,7 +343,7 @@ def show_all_after_loading():
334
  gr.update(visible=True), # model_section
335
  gr.update(visible=True), # voting_section
336
  gr.update(visible=True), # submit_button
337
- gr.update(value="🔄 Try a New Question", elem_classes=["query-button"]) # update button text
338
  ]
339
 
340
  with gr.Blocks(theme=gr.themes.Default(
 
102
  def load_context(set_interrupt=False):
103
  if set_interrupt:
104
  generation_interrupt.set()
105
+ time.sleep(0.5) # Give more time for interrupt to take effect
106
 
107
+ # DON'T clear the interrupt here - let the new inference clear it
108
+ # generation_interrupt.clear() # REMOVED THIS LINE
109
  example = get_random_example()
110
 
111
  context_desc = example.get('processed_context_desc', '')
 
149
  "completed": False
150
  }
151
 
152
+ # Clear the interrupt flag when NEW inference starts
153
+ generation_interrupt.clear()
154
+ print("Interrupt flag cleared for new inference")
155
+
156
  if generation_interrupt.is_set():
157
  return result
158
 
 
185
  result["model_a"] = m_a_name
186
  result["model_b"] = m_b_name
187
 
188
+ print(f"Starting generation with models: {m_a_name} vs {m_b_name}")
189
  s_a, s_b = generate_summaries(example, m_a_name, m_b_name)
190
 
191
  if not generation_interrupt.is_set():
192
  result["summary_a"] = s_a
193
  result["summary_b"] = s_b
194
  result["completed"] = bool(s_a and s_b)
195
+ print("Generation completed successfully")
196
+ else:
197
+ print("Generation was interrupted")
198
  except Exception as e:
199
  print(f"Error in generation: {e}")
200
 
 
343
  gr.update(visible=True), # model_section
344
  gr.update(visible=True), # voting_section
345
  gr.update(visible=True), # submit_button
346
+ gr.update(value="🔄 Try a New Question", elem_classes=["query-button"], interactive=True) # update button text and re-enable
347
  ]
348
 
349
  with gr.Blocks(theme=gr.themes.Default(