update legend

app.py

@@ -176,7 +176,7 @@ def reset_enhance():
     return "请选择样例图片或者上传", None, None
 
 def reset_seg_full():
-    return "请选择样例图片或者上传", None, None, None, False
+    return "请选择样例图片或者上传", None, None, gr.HTML.update(value=None, visible=False)
 
 def reset_diag():
     return "请选择样例图片或者上传", None, [], "", ""
@@ -186,11 +186,11 @@ def segment_demo_with_legend(img_name):
     if result is not None:
         legend_html = """<div style="margin-top:8px; font-size:15px;">
         <span style="display:inline-block;width:18px;height:18px;background:#F33;border-radius:3px;vertical-align:-4px;border:1.5px solid #B00;margin-right:9px"></span>
-        红色部分为息肉
+        红色是息肉
         </div>"""
-        return result, legend_html, True
+        return result, gr.HTML.update(value=legend_html, visible=True)
     else:
-        return None, None, False
+        return None, gr.HTML.update(value=None, visible=False)
 
 with gr.Blocks(title="AI内窥镜影像辅助分析系统") as demo:
     gr.HTML(f"<style>{css}</style>")
@@ -227,10 +227,10 @@ with gr.Blocks(title="AI内窥镜影像辅助分析系统") as demo:
                     seg_btn = gr.Button("点击执行分割", elem_classes="orange-btn")
                 seg_select.change(fn=show_seg_input, inputs=seg_select, outputs=seg_input_img)
                 seg_btn.click(fn=segment_demo_with_legend, inputs=seg_select,
-                              outputs=[seg_output_img, seg_legend, seg_legend.visible])
+                              outputs=[seg_output_img, seg_legend])
                 seg_reset_btn.click(fn=reset_seg_full, 
                                     inputs=None, 
-                                    outputs=[seg_select, seg_input_img, seg_output_img, seg_legend, seg_legend.visible])
+                                    outputs=[seg_select, seg_input_img, seg_output_img, seg_legend])
 
     # 第二行：疾病诊断（大框包裹）
     with gr.Row():