update function 1

app.py

@@ -5,18 +5,18 @@ import random
 
 # ========== 配置图片路径 ==========
 # 图像增强
-ENHANCE_INPUT_DIR = "f1_input"
-ENHANCE_OUTPUT_DIR = "f1_output"
-ENHANCE_IMG_NAMES = [f"{i}.png" for i in range(1, 613)]  # 1.png ~ 612.png
+ENHANCE_INPUT_DIR = "sr_input"
+ENHANCE_OUTPUT_DIR = "sr_output"
+ENHANCE_IMG_NAMES = [f"{i}.png" for i in range(1, 30)]
 
 # 分割
 SEG_INPUT_DIR = "seg_input"
 SEG_OUTPUT_DIR = "seg_output"
-SEG_IMG_NAMES = [f"{i}.png" for i in range(1, 6)]        # 1.png ~ 5.png
+SEG_IMG_NAMES = [f"{i}.png" for i in range(1, 30)]
 
 # 诊断
 DIAG_INPUT_DIR = "diag_input"
-DIAG_IMG_NAMES = [f"{i}.png" for i in range(1, 6)]       # 1.png ~ 5.png
+DIAG_IMG_NAMES = [f"{i}.png" for i in range(1, 30)]
 
 # Kvasir 类别
 KVASIR_CLASSES = [
@@ -38,7 +38,6 @@ def open_img(img_path):
         return Image.new("RGB", (256, 256), (200, 200, 200))
     return Image.open(img_path)
 
-# 图像增强模块
 def show_enhance_input(img_name):
     path = os.path.join(ENHANCE_INPUT_DIR, img_name)
     return open_img(path)
@@ -47,7 +46,6 @@ def enhance_demo(img_name):
     path = os.path.join(ENHANCE_OUTPUT_DIR, img_name)
     return open_img(path)
 
-# 图像分割模块
 def show_seg_input(img_name):
     path = os.path.join(SEG_INPUT_DIR, img_name)
     return open_img(path)
@@ -56,7 +54,6 @@ def segment_demo(img_name):
     path = os.path.join(SEG_OUTPUT_DIR, img_name)
     return open_img(path)
 
-# 疾病诊断模块
 def show_diag_input(img_name):
     path = os.path.join(DIAG_INPUT_DIR, img_name)
     return open_img(path)
@@ -70,14 +67,15 @@ def diagnose_demo(img_name):
 with gr.Blocks(title="AI内窥镜影像辅助分析系统") as demo:
     gr.Markdown("## AI内窥镜影像辅助分析系统")
 
-    # 上方：图像增强模块
+    # 上方：图像增强模块（输入和输出在一行）
     with gr.Row():
         with gr.Column():
             gr.Markdown("### 图像增强模块")
             enhance_select = gr.Dropdown(choices=ENHANCE_IMG_NAMES, value=ENHANCE_IMG_NAMES[0], label="选择样例图片", filterable=True)
-            enhance_input_img = gr.Image(label="原始图片", interactive=False)
+            with gr.Row():
+                enhance_input_img = gr.Image(label="原始图片", interactive=False)
+                enhance_output_img = gr.Image(label="增强结果图片", interactive=False)
             enhance_btn = gr.Button("增强图片")
-            enhance_output_img = gr.Image(label="增强结果图片", interactive=False)
 
             # 选择图片自动显示原图
             enhance_select.change(fn=show_enhance_input, inputs=enhance_select, outputs=enhance_input_img)