update UI

app.py

@@ -0,0 +1,242 @@
+import gradio as gr
+from PIL import Image
+import os
+import random
+
+# ========== 配置图片路径 ==========
+ENHANCE_INPUT_DIR = "sr_input"
+ENHANCE_OUTPUT_DIR = "sr_output"
+ENHANCE_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 30)]
+ENHANCE_CHOICES = ["请选择样例图片或者上传"] + ENHANCE_IMG_NAMES
+
+SEG_INPUT_DIR = "seg_input"
+SEG_OUTPUT_DIR = "seg_output"
+SEG_IMG_NAMES = [f"img_{i}.png" for i in range(1, 30)]
+SEG_CHOICES = ["请选择样例图片或者上传"] + SEG_IMG_NAMES
+
+DIAG_INPUT_DIR = "diag_input"
+DIAG_IMG_NAMES = [f"img_{i}.jpg" for i in range(1, 32)]
+DIAG_CHOICES = ["请选择样例图片或者上传"] + DIAG_IMG_NAMES
+
+KVASIR_CLASSES = [
+    "正常粘膜 (normal-z-line)",
+    "食管静脉曲张 (esophagitis)",
+    "正常盲肠 (normal-cecum)",
+    "息肉 (polyps)",
+    "溃疡 (ulcerative-colitis)",
+    "糜烂 (dyed-lifted-polyps)",
+    "出血 (dyed-resection-margins)",
+    "正常幽门 (normal-pylorus)"
+]
+
+SUGGESTION_DICT = {
+    "正常粘膜 (normal-z-line)": "建议：无需特殊治疗，常规随访。",
+    "食管静脉曲张 (esophagitis)": "建议：根据分级考虑内镜下治疗或药物治疗。",
+    "正常盲肠 (normal-cecum)": "建议：无需特殊治疗，常规随访。",
+    "息肉 (polyps)": "建议：考虑内镜下息肉切除，并随访。",
+    "溃疡 (ulcerative-colitis)": "建议：药物治疗，必要时内镜下活检。",
+    "糜烂 (dyed-lifted-polyps)": "建议：药物保护胃黏膜，可考虑内镜下进一步处理。",
+    "出血 (dyed-resection-margins)": "建议：内镜下止血处理，密切观察。",
+    "正常幽门 (normal-pylorus)": "建议：无需特殊治疗，常规随访。"
+}
+
+def open_img(img_path):
+    if not os.path.exists(img_path):
+        img = Image.new("RGB", (300, 300), (200, 200, 200))
+    else:
+        img = Image.open(img_path)
+    w, h = img.size
+    if max(w, h) != 300:
+        if w >= h:
+            new_w = 300
+            new_h = int(h * 300 / w)
+        else:
+            new_h = 300
+            new_w = int(w * 300 / h)
+        img = img.resize((new_w, new_h), Image.LANCZOS)
+    return img
+
+def show_enhance_input(img_name):
+    if img_name in (None, "", "请选择样例图片或者上传"):
+        return None
+    path = os.path.join(ENHANCE_INPUT_DIR, img_name)
+    return open_img(path)
+
+def enhance_demo(img_name):
+    if img_name in (None, "", "请选择样例图片或者上传"):
+        return None
+    path = os.path.join(ENHANCE_OUTPUT_DIR, img_name)
+    return open_img(path)
+
+def show_seg_input(img_name):
+    if img_name in (None, "", "请选择样例图片或者上传"):
+        return None
+    path = os.path.join(SEG_INPUT_DIR, img_name)
+    return open_img(path)
+
+def segment_demo(img_name):
+    if img_name in (None, "", "请选择样例图片或者上传"):
+        return None
+    path = os.path.join(SEG_OUTPUT_DIR, img_name)
+    return open_img(path)
+
+def show_diag_input(img_name):
+    if img_name in (None, "", "请选择样例图片或者上传"):
+        return None
+    path = os.path.join(DIAG_INPUT_DIR, 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_", ""))
+    idx2cat = [
+        (range(1, 5), "糜烂 (dyed-lifted-polyps)"),
+        (range(5, 9), "出血 (dyed-resection-margins)"),
+        (range(9, 13), "食管静脉曲张 (esophagitis)"),
+        (range(13, 17), "正常盲肠 (normal-cecum)"),
+        (range(17, 21), "正常幽门 (normal-pylorus)"),
+        (range(21, 25), "正常粘膜 (normal-z-line)"),
+        (range(25, 29), "息肉 (polyps)"),
+        (range(29, 33), "溃疡 (ulcerative-colitis)")
+    ]
+    main_cat = None
+    for r, cat in idx2cat:
+        if idx in r:
+            main_cat = cat
+            break
+    if main_cat is None:
+        main_cat = random.choice(KVASIR_CLASSES)
+    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_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_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 ---
+css = """
+.orange-btn {
+    background: #FF7F1F !important;
+    color: white !important;
+    font-weight: bold;
+    border: 2px solid #ff6100 !important;
+    border-radius: 12px !important;
+    font-size: 16px !important;
+    padding: 10px 18px !important;
+    margin-left: 10px !important;
+    box-shadow: none !important;
+}
+.gray-btn {
+    background: #EBEBEE !important;
+    color: black !important;
+    font-weight: bold;
+    border: 2px solid #bfbfbf !important;
+    border-radius: 12px !important;
+    font-size: 16px !important;
+    padding: 10px 18px !important;
+    margin-right: 10px !important;
+    box-shadow: none !important;
+}
+.button-row {
+    margin-top: 4px;
+}
+.big-group {
+    border: 2px solid #dadce0 !important;
+    border-radius: 15px !important;
+    padding: 28px 12px 24px 12px !important;
+    margin-top: 16px !important;
+    margin-bottom: 16px !important;
+    background: #FAFAFA !important;
+    box-shadow: 0 2px 6px 0 rgba(60,64,67,.11),0 1.5px 5px 0 rgba(60,64,67,.11);
+}
+.big-title {
+    font-size: 22px;
+    font-weight: bold;
+    margin-bottom: 14px;
+    margin-left: 6px;
+    color: #ff7f1f;
+}
+"""
+
+def reset_enhance():
+    return "请选择样例图片或者上传", None, None
+
+def reset_seg():
+    return "请选择样例图片或者上传", None, None
+
+def reset_diag():
+    return "请选择样例图片或者上传", None, [], "", ""
+
+with gr.Blocks(title="AI内窥镜影像辅助分析系统") as demo:
+    gr.HTML(f"<style>{css}</style>")
+    gr.Markdown("<span style='font-size: 32px; font-weight: bold;'>AI内窥镜影像辅助分析系统</span>")
+
+    # 第一行：增强&分割（大框包裹）
+    with gr.Row():
+        # 图像增强
+        with gr.Column():
+            with gr.Group(elem_classes="big-group"):
+                gr.HTML(f"<div class='big-title'>图像增强模块</div>")
+                enhance_select = gr.Dropdown(choices=ENHANCE_CHOICES, value="请选择样例图片或者上传", label="选择样例图片", filterable=True)
+                with gr.Row():
+                    enhance_input_img = gr.Image(label="原始图片", interactive=False, width=300, height=300)
+                    enhance_output_img = gr.Image(label="增强结果图片", interactive=False, width=300, height=300)
+                with gr.Row(elem_classes="button-row"):
+                    enhance_reset_btn = gr.Button("Reset", 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])
+
+        # 图像分割
+        with gr.Column():
+            with gr.Group(elem_classes="big-group"):
+                gr.HTML(f"<div class='big-title'>图像分割模块</div>")
+                seg_select = gr.Dropdown(choices=SEG_CHOICES, value="请选择样例图片或者上传", label="选择样例图片", filterable=True)
+                with gr.Row():
+                    seg_input_img = gr.Image(label="输入图片", interactive=False, width=300, height=300)
+                    seg_output_img = gr.Image(label="分割结果图片", interactive=False, width=300, height=300)
+                with gr.Row(elem_classes="button-row"):
+                    seg_reset_btn = gr.Button("Reset", 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])
+
+    # 第二行：疾病诊断（大框包裹）
+    with gr.Row():
+        with gr.Group(elem_classes="big-group"):
+            gr.HTML(f"<div class='big-title'>疾病分类模块</div>")
+            with gr.Row():
+                # 1. 输入图片+按钮
+                with gr.Column():
+                    diag_select = gr.Dropdown(choices=DIAG_CHOICES, value="请选择样例图片或者上传", label="选择样例图片", filterable=True)
+                    diag_input_img = gr.Image(label="输入图片", interactive=False, width=300, height=300)
+                    with gr.Row(elem_classes="button-row"):
+                        diag_reset_btn = gr.Button("Reset", 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)
+                # 2. 结果表格
+                with gr.Column():
+                    diag_table = gr.Dataframe(headers=["类别", "概率"])
+                # 3. 预测与建议
+                with gr.Column():
+                    diag_result = gr.Textbox(label="诊断结果")
+                    diag_suggestion = gr.Textbox(label="建议的治疗��案")
+            diag_btn.click(fn=diagnose_demo, inputs=diag_select, outputs=[diag_table, diag_result, diag_suggestion])
+            diag_reset_btn.click(fn=reset_diag, inputs=None, outputs=[diag_select, diag_input_img, diag_table, diag_result, diag_suggestion])
+
+demo.launch()