Update app.py

app.py

@@ -136,24 +136,23 @@ def CTXGen(τ, g_num, length_range, progress=gr.Progress()):
                     progress(count / gen_num, desc="Generating conotoxins...")
     return 'output.csv', f"Generated {count} conotoxins."
 
-# 定义 Gradio 界面
-iface = gr.Interface(
-    fn=CTXGen,
-    inputs=[
-        gr.Slider(minimum=1, maximum=2, step=0.1, label="τ"),
-        gr.Dropdown(choices=[1, 10, 100], label="Number of generations"),
-        RangeSlider(minimum=8, maximum=50, step=1, value=(12, 17), label="Length range")
-    ],
-    outputs=[
-        gr.File(label="Download generated conotoxins"),
-        gr.Textbox(label="Progress")
-    ],
-    live=True
-)
-
-# 添加停止按钮
-stop_button = gr.Button("Stop Generation")
-stop_button.click(stop_generation, outputs=gr.Textbox(label="Status"))
+# 使用 gr.Blocks 构建界面
+with gr.Blocks() as demo:
+    gr.Markdown("# Conotoxin Generation")
+    with gr.Row():
+        τ = gr.Slider(minimum=1, maximum=2, step=0.1, label="τ")
+        g_num = gr.Dropdown(choices=[1, 10, 100], label="Number of generations")
+        length_range = RangeSlider(minimum=8, maximum=50, step=1, value=(12, 17), label="Length range")
+    with gr.Row():
+        start_button = gr.Button("Start Generation")
+        stop_button = gr.Button("Stop Generation")
+    with gr.Row():
+        output_file = gr.File(label="Download generated conotoxins")
+        status_text = gr.Textbox(label="Status")
+    
+    # 绑定事件
+    start_button.click(CTXGen, inputs=[τ, g_num, length_range], outputs=[output_file, status_text])
+    stop_button.click(stop_generation, outputs=status_text)
 
 # 启动 Gradio 应用
-iface.launch()
+demo.launch()