Update app.py

app.py

@@ -114,20 +114,22 @@ def process_image(model, tokenizer, transform, device, check_type, image, text):
 # 事件处理函数
 def update_index(change):
     global current_vis, current_bpe, current_index
+    
+    # 限制索引范围
     current_index = max(0, min(len(current_vis) - 1, current_index + change))
-    return current_vis[current_index], format_bpe_display(current_bpe[current_index])
+    
+    # 处理按钮可见性
+    prev_visible = current_index > 0
+    next_visible = current_index < len(current_vis) - 1
+    
+    return current_vis[current_index], format_bpe_display(current_bpe[current_index]), prev_visible, next_visible
+
 
 def format_bpe_display(bpe):
     # 使用HTML标签来设置字体大小、颜色，加粗，并居中
     return f"<div style='text-align:center; font-size:20px;'><strong>Current BPE: <span style='color:red;'>{bpe}</span></strong></div>"
 
-def update_slider_index(x):
-    global current_vis, current_bpe, current_index
-    print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}")
-    if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe):
-        return current_vis[x], format_bpe_display(current_bpe[x])
-    else:
-        return None, "索引超出范围"
+
 
 
 
@@ -166,7 +168,6 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
             
             with gr.Row() as controls:
                 prev_btn = gr.Button("⬅ Last", visible=False)
-                index_slider = gr.Slider(0, 1, value=0, step=1, label="BPE index", visible=False)
                 next_btn = gr.Button("⮕ Next", visible=False)
             
             bpe_display = gr.Markdown("Current BPE: ", visible=False)
@@ -175,44 +176,33 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
     @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
+        current_index = 0  # 重新处理新图片时，索引归零
         image, vis, bpe = process_image(*load_model(model_type), model_type, image, text)
-        # Update the slider range and set value to 0
-        slider_max_val = len(current_bpe) - 1
-        bpe_text = format_bpe_display(bpe)
+        
+        if len(current_vis) > 1:
+            prev_visible, next_visible = True, True
+        else:
+            prev_visible, next_visible = False, False
         print("len_current_vis",len(current_vis))
         print("len_current_bpe",len(current_bpe))
         print("current_vis",current_vis)
         print("current_bpe",current_bpe)
-        return image, vis, bpe_text, slider_max_val
-
+        return image, vis, format_bpe_display(bpe), prev_visible, next_visible
     
-    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],
-    )
+
+
     
     prev_btn.click(
-        lambda: (*update_index(-1), current_index),
-        outputs=[heatmap, bpe_display, index_slider]
+        lambda: update_index(-1),
+        outputs=[heatmap, bpe_display, prev_btn, next_btn]
     )
     
     next_btn.click(
-        lambda: (*update_index(1), current_index),
-        outputs=[heatmap, bpe_display, index_slider]
+        lambda: update_index(1),
+        outputs=[heatmap, bpe_display, prev_btn, next_btn]
     )
 
 
-    index_slider.change(
-            update_slider_index,
-            inputs=index_slider,
-            outputs=[heatmap, bpe_display]
-        )