Update app.py

app.py

@@ -129,12 +129,13 @@ def format_bpe_display(bpe):
 #         return current_vis[x], format_bpe_display(current_bpe[x])
 #     else:
 #         return None, "索引超出范围"
-
-def update_slider_index(x):
-    if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe):
-        return current_vis[x], format_bpe_display(current_bpe[x])
+# 状态更新函数，利用传递的状态（vis, bpe）
+def update_slider_index(x, vis, bpe):
+    if 0 <= x < len(vis):
+        return vis[x], format_bpe_display(bpe[x]), vis, bpe
     else:
-        return None, "索引超出范围"
+        return None, "索引超出范围", vis, bpe
+
 
 # Gradio界面
 with gr.Blocks(title="BPE Visualization Demo") as demo:
@@ -188,26 +189,38 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #     print("current_vis",len(current_vis))
     #     print("current_bpe",len(current_bpe))
     #     return image, vis, bpe_text, slider_max_val
+
     @spaces.GPU
     def on_run_clicked(model_type, image, text):
-        global current_vis, current_bpe, current_index
-        current_index = 0  # Reset index when new image is processed
         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)
-        current_vis = vis
-        current_bpe = bpe
-        slider_max_val = len(current_bpe) - 1
-        bpe_text = format_bpe_display(current_bpe[current_index])
-        return image, current_vis[current_index], bpe_text, slider_max_val
+        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
+
+
+
     
+    # run_btn.click(
+    #     on_run_clicked,
+    #     inputs=[model_type, image_input, text_input],
+    #     outputs=[orig_img, heatmap, bpe_display, index_slider],
+    # ).then(
+    #     lambda max_val: (gr.update(visible=True), gr.update(visible=True, maximum=max_val, value=0), gr.update(visible=True), gr.update(visible=True)),
+    #     inputs=index_slider,
+    #     outputs=[prev_btn, index_slider, next_btn, bpe_display],
+    # )
+    # Gradio 按钮点击后的处理
     run_btn.click(
         on_run_clicked,
         inputs=[model_type, image_input, text_input],
-        outputs=[orig_img, heatmap, bpe_display, index_slider],
+        outputs=[orig_img, heatmap, bpe_display, index_slider, 'state', 'state']
     ).then(
-        lambda max_val: (gr.update(visible=True), gr.update(visible=True, maximum=max_val, value=0), gr.update(visible=True), gr.update(visible=True)),
+        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],
+        outputs=[prev_btn, index_slider, next_btn, bpe_display, 'state', 'state']
     )
     
     prev_btn.click(
@@ -219,12 +232,7 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
         lambda: (*update_index(1), current_index),
         outputs=[heatmap, bpe_display, index_slider]
     )
-    
-    # index_slider.change(
-    #     lambda x: (current_vis[x], format_bpe_display(current_bpe[x])) if 0<=x<len(current_vis else (None,"Invaild")
-    #     inputs=index_slider,
-    #     outputs=[heatmap, bpe_display]
-    # )
+
 
     # index_slider.change(
     #         update_slider_index,
@@ -232,9 +240,10 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     #         outputs=[heatmap, bpe_display]
     #     )
     index_slider.change(
-    update_slider_index,
-    inputs=[index_slider],
-    outputs=[heatmap, bpe_display])
+        update_slider_index,
+        inputs=[index_slider, 'state', 'state'],
+        outputs=[heatmap, bpe_display, 'state', 'state']
+    )
 
 
 if __name__ == "__main__":