Update app.py

app.py

@@ -25,6 +25,11 @@ current_vis = []
 current_bpe = []
 current_index = 0
 
+# 设置初始状态
+initial_state = {
+    "vis": [],
+    "bpe": []
+}
 
 def load_model(check_type):
     # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -130,11 +135,17 @@ def format_bpe_display(bpe):
 #     else:
 #         return None, "索引超出范围"
 # 状态更新函数，利用传递的状态（vis, bpe）
-def update_slider_index(x, vis, bpe):
+# 使用状态信息来处理滑动条改变
+def update_slider_index(x, state):
+    vis = state['vis']
+    bpe = state['bpe']
     if 0 <= x < len(vis):
-        return vis[x], format_bpe_display(bpe[x]), vis, bpe
+        return vis[x], format_bpe_display(bpe[x]), state
     else:
-        return None, "索引超出范围", vis, bpe
+        return None, "索引超出范围", state
+
+
+
 
 
 # Gradio界面
@@ -191,14 +202,17 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     return image, vis, bpe_text, slider_max_val
 
     @spaces.GPU
-    def on_run_clicked(model_type, image, text):
+    def on_run_clicked(model_type, image, text, state):
         model, tokenizer, transform, device = load_model(model_type)
         current_index = 0  # Reset index when new image is processed
         image, vis, bpe = process_image(model, tokenizer, transform, device, model_type, image, text)
         slider_max_val = len(bpe) - 1
         bpe_text = format_bpe_display(bpe[current_index])
-        # 将处理结果传递给后续步骤
-        return image, vis[current_index], bpe_text, slider_max_val, vis, bpe
+        # 更新状态并返回
+        state['vis'] = vis
+        state['bpe'] = bpe
+        return image, vis[current_index], bpe_text, slider_max_val, state
+
 
 
 
@@ -213,14 +227,12 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     outputs=[prev_btn, index_slider, next_btn, bpe_display],
     # )
     # Gradio 按钮点击后的处理
+# Gradio 按钮点击后的处理
     run_btn.click(
         on_run_clicked,
-        inputs=[model_type, image_input, text_input],
-        outputs=[orig_img, heatmap, bpe_display, index_slider, 'state', 'state']
-    ).then(
-        lambda outputs: (gr.update(visible=True), gr.update(visible=True, maximum=outputs[3], value=0), gr.update(visible=True), gr.update(visible=True), outputs[4], outputs[5]),
-        inputs=index_slider,
-        outputs=[prev_btn, index_slider, next_btn, bpe_display, 'state', 'state']
+        inputs=[model_type, image_input, text_input, 'state'],
+        outputs=[orig_img, heatmap, bpe_display, index_slider, 'state'],
+        _js="{state: { vis: [], bpe: []}}"
     )
     
     prev_btn.click(
@@ -241,8 +253,8 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     )
     index_slider.change(
         update_slider_index,
-        inputs=[index_slider, 'state', 'state'],
-        outputs=[heatmap, bpe_display, 'state', 'state']
+        inputs=[index_slider, 'state'],
+        outputs=[heatmap, bpe_display, 'state']
     )