update UI

app.py

@@ -41,11 +41,7 @@ SUGGESTION_DICT = {
 }
 
 def open_img(img_path):
-    """
-    将图等比缩放成 300x300，并返回 PIL.Image.Image
-    """
     if not os.path.exists(img_path):
-        # 若路径不存在，用灰色方块占位
         img = Image.new("RGB", (300, 300), (200, 200, 200))
     else:
         img = Image.open(img_path)
@@ -85,9 +81,6 @@ def show_diag_input(img_name):
     return open_img(path)
 
 def diagnose_demo(img_name):
-    """
-    随机模拟诊断结果与概率分布
-    """
     if img_name in (None, "", "请选择样例图片或者上传"):
         return [], "", ""
     idx = int(os.path.splitext(img_name)[0].replace("img_", ""))
@@ -107,34 +100,27 @@ def diagnose_demo(img_name):
             main_cat = cat
             break
     if main_cat is None:
-        import random
         main_cat = random.choice(KVASIR_CLASSES)
-
-    import random
     n_cat = len(KVASIR_CLASSES)
     probs = [0.0 for _ in range(n_cat)]
     main_idx = KVASIR_CLASSES.index(main_cat)
     main_prob = random.uniform(0.85, 0.99)
     rest = 1 - main_prob
     other_probs_raw = [random.uniform(0.01, 1) for _ in range(n_cat-1)]
-    other_probs_sum = sum(other_probs_raw)
-    other_probs = [x / other_probs_sum * rest for x in other_probs_raw]
+    other_probs_norm = [x / sum(other_probs_raw) * rest for x in other_probs_raw]
     pidx = 0
     for i in range(n_cat):
         if i == main_idx:
             probs[i] = main_prob
         else:
-            probs[i] = other_probs[pidx]
+            probs[i] = other_probs_norm[pidx]
             pidx += 1
-
     result_table = [[KVASIR_CLASSES[i], f"{probs[i]:.4f}"] for i in range(n_cat)]
     result_text = f"诊断类别：{main_cat}（概率最大）"
     suggestion = SUGGESTION_DICT.get(main_cat, "建议：请咨询医生。")
     return result_table, result_text, suggestion
 
-# 自定义CSS，特别是 img-label-bar 用于在图片下方添加彩条
 css = """
-/* 主标题居中加大 */
 #main-title {
     text-align: center;
     font-size: 44px;
@@ -143,7 +129,6 @@ css = """
     margin-top: 18px;
     letter-spacing: 2px;
 }
-
 /* 模块外层深蓝虚线圆角 */
 .blue-dash-border {
     border: 2.5px dashed #164fa0 !important;
@@ -152,8 +137,6 @@ css = """
     margin-bottom: 32px !important;
     box-sizing: border-box;
 }
-
-/* 模块内卡片背景、阴影、圆角等 */
 .big-group {
     background: #F5F6F7 !important;
     border: 2px solid #e0e2e5 !important;
@@ -162,8 +145,6 @@ css = """
     margin: 16px !important;
     box-shadow: 0 3px 12px 2px rgba(60,64,67,.14);
 }
-
-/* 模块标题居中 */
 .big-title {
     font-size: 25px;
     font-weight: bold;
@@ -172,8 +153,6 @@ css = """
     letter-spacing: 1px;
     text-align: center;
 }
-
-/* 按钮样式（橙色无边框，灰色带黑色边框） */
 .orange-btn {
     background: #FF7F1F !important;
     color: white !important;
@@ -201,20 +180,15 @@ css = """
 .button-row {
     margin-top: 8px;
 }
-
-/* 图片圆角阴影 */
 .gr-image, .gr-image-preview {
     border-radius: 10px !important;
     background: #fff !important;
     box-shadow: 0 2px 12px 0 rgba(60,64,67,.08);
 }
-/* 禁止卡片内嵌套group的边框和阴影 */
 .gr-box .gr-block.gr-group {
     box-shadow: none;
     border: none;
 }
-
-/* 下方彩条，用于显示文字说明 */
 .img-label-bar {
     background: rgb(76, 180, 184);
     color: #fff;
@@ -222,7 +196,7 @@ css = """
     padding: 6px 0;
     border-bottom-left-radius: 10px;
     border-bottom-right-radius: 10px;
-    margin-top: -5px; /* 让bar紧贴图片下边缘 */
+    margin-top: -5px;
     font-size: 14px;
     font-weight: normal;
 }
@@ -251,20 +225,15 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
                     enhance_select = gr.Dropdown(choices=ENHANCE_CHOICES, value="请选择样例图片或者上传", label="选择样例图片", filterable=True)
                     with gr.Row():
                         with gr.Column():
-                            # 1) 不在 gr.Image 里加 label
                             enhance_input_img = gr.Image(show_label=False, interactive=False, width=300, height=300)
-                            # 2) 自定义bar来显示 "原始图片"
                             gr.HTML("<div class='img-label-bar'>原始图片</div>")
                         with gr.Column():
                             enhance_output_img = gr.Image(show_label=False, interactive=False, width=300, height=300)
                             gr.HTML("<div class='img-label-bar'>增强结果图片</div>")
-                    # 按钮区域避免横向拉伸
+                    # 按钮直接并排
                     with gr.Row(elem_classes="button-row"):
-                        with gr.Column(scale=0):
-                            enhance_reset_btn = gr.Button("Reset", elem_classes="gray-btn")
-                        with gr.Column(scale=0):
-                            enhance_btn = gr.Button("点击执行图片增强", elem_classes="orange-btn")
-                    # 交互
+                        enhance_reset_btn = gr.Button("清空", elem_classes="gray-btn")
+                        enhance_btn = gr.Button("点击执行图片增强", elem_classes="orange-btn")
                     enhance_select.change(fn=show_enhance_input, inputs=enhance_select, outputs=enhance_input_img)
                     enhance_btn.click(fn=enhance_demo, inputs=enhance_select, outputs=enhance_output_img)
                     enhance_reset_btn.click(fn=reset_enhance, inputs=None, outputs=[enhance_select, enhance_input_img, enhance_output_img])
@@ -283,10 +252,8 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
                             seg_output_img = gr.Image(show_label=False, interactive=False, width=300, height=300)
                             gr.HTML("<div class='img-label-bar'>分割结果图片</div>")
                     with gr.Row(elem_classes="button-row"):
-                        with gr.Column(scale=0):
-                            seg_reset_btn = gr.Button("Reset", elem_classes="gray-btn")
-                        with gr.Column(scale=0):
-                            seg_btn = gr.Button("点击执行分割", elem_classes="orange-btn")
+                        seg_reset_btn = gr.Button("清空", elem_classes="gray-btn")
+                        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, inputs=seg_select, outputs=seg_output_img)
                     seg_reset_btn.click(fn=reset_seg, inputs=None, outputs=[seg_select, seg_input_img, seg_output_img])
@@ -304,10 +271,8 @@ with gr.Blocks(title="消化道疾病智能分析系统") as demo:
                             diag_input_img = gr.Image(show_label=False, interactive=False, width=300, height=300)
                             gr.HTML("<div class='img-label-bar'>输入图片</div>")
                         with gr.Row(elem_classes="button-row"):
-                            with gr.Column(scale=0):
-                                diag_reset_btn = gr.Button("Reset", elem_classes="gray-btn")
-                            with gr.Column(scale=0):
-                                diag_btn = gr.Button("点击执行诊断", elem_classes="orange-btn")
+                            diag_reset_btn = gr.Button("清空", elem_classes="gray-btn")
+                            diag_btn = gr.Button("点击执行诊断", elem_classes="orange-btn")
                         diag_select.change(fn=show_diag_input, inputs=diag_select, outputs=diag_input_img)
 
                     # 中列：结果表格