Update app.py

app.py

@@ -195,10 +195,19 @@ with gr.Blocks(title="BPE Visualization Demo") as demo:
         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(
-        lambda x: (current_vis[x], format_bpe_display(current_bpe[x])),
-        inputs=index_slider,
-        outputs=[heatmap, bpe_display]
+    lambda x: (
+        print(f"x: {x}, current_vis length: {len(current_vis)}, current_bpe length: {len(current_bpe)}"),  # 打印调试信息
+        (current_vis[x], format_bpe_display(current_bpe[x])) if 0 <= x < len(current_vis) and 0 <= x < len(current_bpe) else (None, "索引超出范围")
+    ),
+    inputs=index_slider,
+    outputs=[heatmap, bpe_display]
     )
 
 if __name__ == "__main__":