Update app.py

app.py

@@ -82,17 +82,25 @@ def calculate_position(org_size, add_size, position):
     elif position == "하단 우측":
         return (org_size[0] - add_size[0], org_size[1] - add_size[1])
 
-def merge(org_image, add_image, scale, position):
+
+def merge(org_image, add_image, scale, position, display_size):
+    # 사용자가 선택한 디스플레이 크기에 따라 결과 이미지 크기 조절
+    display_width, display_height = map(int, display_size.split('x'))
+    
+    # 이미지 병합 로직
     scale_percentage = scale / 100.0
     new_size = (int(add_image.width * scale_percentage), int(add_image.height * scale_percentage))
     add_image = add_image.resize(new_size, Image.Resampling.LANCZOS)
     
     position = calculate_position(org_image.size, add_image.size, position)
-    org_image.paste(add_image, position, add_image)
+    merged_image = Image.new("RGBA", org_image.size)
+    merged_image.paste(org_image, (0, 0))
+    merged_image.paste(add_image, position, add_image)
     
-    return org_image
-
-
+    # 결과 이미지 디스플레이 크기 조절
+    final_image = merged_image.resize((display_width, display_height), Image.Resampling.LANCZOS)
+    
+    return final_image
 
 with gr.Blocks() as demo:
     with gr.Tab("Background Removal"):
@@ -111,12 +119,18 @@ with gr.Blocks() as demo:
 
     with gr.Tab("Merge"):
         with gr.Column():
-            org_image = gr.Image(label="Background", type='pil', image_mode='RGBA', height="40vh")
-            add_image = gr.Image(label="Foreground", type='pil', image_mode='RGBA', height="40vh")
+            org_image = gr.Image(label="Background", type='pil', image_mode='RGBA', height=400) # 예시로 높이 조절
+            add_image = gr.Image(label="Foreground", type='pil', image_mode='RGBA', height=400) # 예시로 높이 조절
             scale = gr.Slider(minimum=10, maximum=200, step=1, value=100, label="Scale of Foreground Image (%)")
             position = gr.Radio(choices=["중앙 가운데", "상단 좌측", "상단 가운데", "상단 우측", "중앙 좌측", "중앙 우측", "하단 좌측", "하단 가운데", "하단 우측"], value="중앙 가운데", label="Position of Foreground Image")
+            display_size = gr.Textbox(value="800x600", label="Display Size (Width x Height)")
             btn_merge = gr.Button("Merge Images")
-            result_merge = gr.Image(height="40vh")
-            btn_merge.click(fn=merge, inputs=[org_image, add_image, scale, position], outputs=result_merge)
+            result_merge = gr.Image()
+            
+            btn_merge.click(
+                fn=merge,
+                inputs=[org_image, add_image, scale, position, display_size],
+                outputs=result_merge,
+            )
 
 demo.launch()