Update app.py

app.py

@@ -158,10 +158,76 @@ def generate_background(prompt: str, aspect_ratio: str) -> Image.Image:
         raise gr.Error(f"Background generation failed: {str(e)}")
 
 
-def combine_with_background(foreground: Image.Image, background: Image.Image) -> Image.Image:
+def create_position_grid():
+    """3x3 위치 선택 그리드를 생성하는 HTML"""
+    return """
+    <div style="display: grid; grid-template-columns: repeat(3, 1fr); gap: 10px; width: 150px; margin: auto;">
+        <button class="position-btn" data-pos="top-left">↖</button>
+        <button class="position-btn" data-pos="top-center">↑</button>
+        <button class="position-btn" data-pos="top-right">↗</button>
+        <button class="position-btn" data-pos="middle-left">←</button>
+        <button class="position-btn" data-pos="middle-center">•</button>
+        <button class="position-btn" data-pos="middle-right">→</button>
+        <button class="position-btn" data-pos="bottom-left">↙</button>
+        <button class="position-btn" data-pos="bottom-center" data-default="true">↓</button>
+        <button class="position-btn" data-pos="bottom-right">↘</button>
+    </div>
+    <script>
+        const buttons = document.querySelectorAll('.position-btn');
+        buttons.forEach(btn => {
+            btn.style.width = '40px';
+            btn.style.height = '40px';
+            btn.style.border = '1px solid #ccc';
+            btn.style.borderRadius = '4px';
+            btn.style.cursor = 'pointer';
+            if (btn.dataset.default === 'true') {
+                btn.style.backgroundColor = '#2196F3';
+                btn.style.color = 'white';
+            }
+        });
+    </script>
+    """
+
+def calculate_object_position(position: str, bg_size: tuple[int, int], obj_size: tuple[int, int]) -> tuple[int, int]:
+    """오브젝트의 위치 계산"""
+    bg_width, bg_height = bg_size
+    obj_width, obj_height = obj_size
+    
+    positions = {
+        "top-left": (0, 0),
+        "top-center": ((bg_width - obj_width) // 2, 0),
+        "top-right": (bg_width - obj_width, 0),
+        "middle-left": (0, (bg_height - obj_height) // 2),
+        "middle-center": ((bg_width - obj_width) // 2, (bg_height - obj_height) // 2),
+        "middle-right": (bg_width - obj_width, (bg_height - obj_height) // 2),
+        "bottom-left": (0, bg_height - obj_height),
+        "bottom-center": ((bg_width - obj_width) // 2, bg_height - obj_height),
+        "bottom-right": (bg_width - obj_width, bg_height - obj_height)
+    }
+    
+    return positions.get(position, positions["bottom-center"])
+
+def resize_object(image: Image.Image, scale_percent: float) -> Image.Image:
+    """오브젝트 크기 조정"""
+    width = int(image.width * scale_percent / 100)
+    height = int(image.height * scale_percent / 100)
+    return image.resize((width, height), Image.Resampling.LANCZOS)
+
+def combine_with_background(foreground: Image.Image, background: Image.Image, 
+                          position: str = "bottom-center", scale_percent: float = 100) -> Image.Image:
     """전경과 배경 합성 함수"""
-    background = background.resize(foreground.size)
-    return Image.alpha_composite(background.convert('RGBA'), foreground)
+    # 배경 이미지 준비
+    result = background.convert('RGBA')
+    
+    # 오브젝트 크기 조정
+    scaled_foreground = resize_object(foreground, scale_percent)
+    
+    # 오브젝트 위치 계산
+    x, y = calculate_object_position(position, result.size, scaled_foreground.size)
+    
+    # 합성
+    result.paste(scaled_foreground, (x, y), scaled_foreground)
+    return result
 
 @spaces.GPU
 def _gpu_process(img: Image.Image, prompt: str | BoundingBox | None) -> tuple[Image.Image, BoundingBox | None, list[str]]:
@@ -270,7 +336,7 @@ def update_box_button(img, box_input):
         return gr.update(interactive=False, variant="secondary")
 
 
-# 맨 앞부분에 CSS 정의 추가
+# CSS 정의
 css = """
 footer {display: none}
 .main-title {
@@ -321,9 +387,19 @@ button.primary {
 button.primary:hover {
     background: #1976D2;
 }
+.position-btn {
+    transition: all 0.3s ease;
+}
+.position-btn:hover {
+    background-color: #e3f2fd;
+}
+.position-btn.selected {
+    background-color: #2196F3;
+    color: white;
+}
 """
 
-# UI 부분 수정
+# UI 구성
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.HTML("""
         <div class="main-title">
@@ -359,6 +435,21 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     visible=True,
                     scale=1
                 )
+            
+            # 오브젝트 위치와 크기 조정 컨트롤
+            with gr.Row(visible=False) as object_controls:
+                with gr.Column(scale=1):
+                    gr.HTML(create_position_grid())
+                    position = gr.State(value="bottom-center")
+                with gr.Column(scale=1):
+                    scale_slider = gr.Slider(
+                        minimum=10,
+                        maximum=200,
+                        value=100,
+                        step=10,
+                        label="Object Size (%)"
+                    )
+            
             process_btn = gr.Button(
                 "Process",
                 variant="primary",
@@ -396,20 +487,42 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         queue=False
     )
 
-    # bg_prompt가 비어있을 때 aspect_ratio를 비활성화하는 함수
-    def update_aspect_ratio(bg_prompt):
-        return gr.update(visible=bool(bg_prompt))
+    def update_controls(bg_prompt):
+        """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
+        is_visible = bool(bg_prompt)
+        return [
+            gr.update(visible=is_visible),  # aspect_ratio
+            gr.update(visible=is_visible),  # object_controls
+        ]
 
     bg_prompt.change(
-        fn=update_aspect_ratio,
+        fn=update_controls,
         inputs=bg_prompt,
-        outputs=aspect_ratio,
+        outputs=[aspect_ratio, object_controls],
         queue=False
     )
 
+    # 위치 선택 버튼 클릭 이벤트
+    def update_position(evt: gr.SelectData) -> str:
+        """위치 선택 업데이트"""
+        return evt.value
+
+    position.change(
+        fn=lambda x: gr.update(value=x),
+        inputs=position,
+        outputs=position
+    )
+
     process_btn.click(
         fn=process_prompt,
-        inputs=[input_image, text_prompt, bg_prompt, aspect_ratio],
+        inputs=[
+            input_image,
+            text_prompt,
+            bg_prompt,
+            aspect_ratio,
+            position,
+            scale_slider
+        ],
         outputs=[combined_image, extracted_image],
         queue=True
     )