Update app.py

app.py

@@ -305,8 +305,9 @@ def process_with_alignment(location, reference_image, input_image):
 
 # Create the Gradio interface
 def create_interface():
-    scale = 0.5
-    disp_w, disp_h = int(1365*scale), int(1024*scale)
+    # 设置统一的显示尺寸
+    disp_w, disp_h = 683, 512  # 统一设置宽高比
+    
     with gr.Blocks(title="海岸侵蚀分析系统") as demo:
         gr.Markdown("""# 海岸侵蚀分析系统
 
@@ -316,9 +317,10 @@ def create_interface():
                 with gr.Row():
                     loc1 = gr.Radio(list(MODEL_PATHS.keys()), label="选择模型", value=list(MODEL_PATHS.keys())[0])
                 with gr.Row():
-                    inp = gr.Image(label="输入图像", type="numpy", image_mode="RGB")
-                    seg = gr.Image(label="分割图像", type="numpy", width=disp_w, height=disp_h)
-                    ovl = gr.Image(label="叠加图像", type="numpy", width=disp_w, height=disp_h)
+                    # 确保所有图像组件使用相同的尺寸
+                    inp = gr.Image(label="输入图像", type="numpy", image_mode="RGB", height=disp_h, width=disp_w)
+                    seg = gr.Image(label="分割图像", type="numpy", height=disp_h, width=disp_w)
+                    ovl = gr.Image(label="叠加图像", type="numpy", height=disp_h, width=disp_w)
                 with gr.Row():
                     btn1 = gr.Button("执行分割")
                 url1 = gr.Text(label="分割图URL")
@@ -330,16 +332,17 @@ def create_interface():
                 with gr.Row():
                     loc2 = gr.Radio(list(MODEL_PATHS.keys()), label="选择模型", value=list(MODEL_PATHS.keys())[0])
                 with gr.Row():
-                    ref_img = gr.Image(label="参考图像", type="numpy", image_mode="RGB")
-                    tgt_img = gr.Image(label="待分析图像", type="numpy", image_mode="RGB")
+                    # 确保所有图像组件使用相同的尺寸
+                    ref_img = gr.Image(label="参考图像", type="numpy", image_mode="RGB", height=disp_h, width=disp_w)
+                    tgt_img = gr.Image(label="待分析图像", type="numpy", image_mode="RGB", height=disp_h, width=disp_w)
                 with gr.Row():
                     btn2 = gr.Button("执行空间对齐分割")
                 with gr.Row():
-                    orig = gr.Image(label="原始图像", type="numpy", width=disp_w, height=disp_h)
-                    aligned = gr.Image(label="对齐后图像", type="numpy", width=disp_w, height=disp_h)
+                    orig = gr.Image(label="原始图像", type="numpy", height=disp_h, width=disp_w)
+                    aligned = gr.Image(label="对齐后图像", type="numpy", height=disp_h, width=disp_w)
                 with gr.Row():
-                    seg2 = gr.Image(label="分割图像", type="numpy", width=disp_w, height=disp_h)
-                    ovl2 = gr.Image(label="叠加图像", type="numpy", width=disp_w, height=disp_h)
+                    seg2 = gr.Image(label="分割图像", type="numpy", height=disp_h, width=disp_w)
+                    ovl2 = gr.Image(label="叠加图像", type="numpy", height=disp_h, width=disp_w)
                 url2 = gr.Text(label="分割图URL")
                 status2 = gr.HTML(label="空间对齐状态")
                 res2 = gr.HTML(label="分析结果")