update UI

app.py

@@ -1,242 +0,0 @@
-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()