Update app.py

app.py

@@ -35,7 +35,7 @@ def stop_generation():
     is_stopped = True
     return "Generation stopped."
 
-def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
+def CTXGen(τ, g_num, length_range):
     global is_stopped
     is_stopped = False  # 重置停止标志
     start, end = length_range
@@ -66,7 +66,7 @@ def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
         start_time = time.time()
         while count < gen_num:
             if is_stopped:  # 检查是否停止
-                return pd.DataFrame(), f"Generation stopped. Generated {count} conotoxins."
+                yield pd.DataFrame(), f"Generation stopped. Generated {count} conotoxins."
 
             if time.time() - start_time > 1200:
                 break
@@ -84,7 +84,7 @@ def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
 
             for i in range(length):
                 if is_stopped:  # 检查是否停止
-                    return pd.DataFrame(), f"Generation stopped. Generated {count} conotoxins."
+                    yield pd.DataFrame(), f"Generation stopped. Generated {count} conotoxins."
 
                 _, idx_seq, idx_msa, attn_idx = get_paded_token_idx_gen(vocab_mlm, seq, new_seq)
                 idx_seq = torch.tensor(idx_seq).unsqueeze(0).to(device)
@@ -140,9 +140,8 @@ def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
                     })
                     out.to_csv("output.csv", index=False, encoding='utf-8-sig')
                     count += 1
-                    progress(count / gen_num, desc="Generating conotoxins...")
                     yield out, f"Generated {count} conotoxins."
-    return out, f"Generated {count} conotoxins."
+    yield out, f"Generated {count} conotoxins."
 
 # 使用 gr.Blocks 构建界面
 with gr.Blocks() as demo:
@@ -161,7 +160,7 @@ with gr.Blocks() as demo:
         output_file = gr.File(label="Download generated conotoxins")
     
     # 绑定事件
-    start_button.click(CTXGen, inputs=[τ, g_num, length_range], outputs=[output_df, status_text], live=True)
+    start_button.click(CTXGen, inputs=[τ, g_num, length_range], outputs=[output_df, status_text])
     stop_button.click(stop_generation, outputs=status_text)
 
 # 启动 Gradio 应用