Update app.py

app.py

@@ -268,16 +268,14 @@ def resize_object(image: Image.Image, scale_percent: float) -> Image.Image:
 def combine_with_background(foreground: Image.Image, background: Image.Image, 
                           position: str = "bottom-center", scale_percent: float = 100) -> Image.Image:
     """전경과 배경 합성 함수"""
-    # 배경 이미지 준비
-    result = background.convert('RGBA')
+    print(f"Combining with position: {position}, scale: {scale_percent}")
     
-    # 오브젝트 크기 조정
+    result = background.convert('RGBA')
     scaled_foreground = resize_object(foreground, scale_percent)
     
-    # 오브젝트 위치 계산
     x, y = calculate_object_position(position, result.size, scaled_foreground.size)
+    print(f"Calculated position coordinates: ({x}, {y})")
     
-    # 합성
     result.paste(scaled_foreground, (x, y), scaled_foreground)
     return result
 
@@ -348,12 +346,11 @@ def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str
     return gr.update(interactive=bool(img and prompt))
 
 
-        
-# process_prompt 함수 수정
 def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None, 
                   aspect_ratio: str = "1:1", position: str = "bottom-center", 
                   scale_percent: float = 100) -> tuple[Image.Image, Image.Image]:
-    try:
+    try:        
+
         if img is None or prompt.strip() == "":
             raise gr.Error("Please provide both image and prompt")
         
@@ -368,9 +365,18 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
         
         results, _ = _process(img, prompt, bg_prompt, aspect_ratio)
         
+        
         if bg_prompt:
             try:
                 print(f"Using position: {position}")  # 디버깅용
+                # 위치 값 검증
+                valid_positions = ["top-left", "top-center", "top-right",
+                                 "middle-left", "middle-center", "middle-right",
+                                 "bottom-left", "bottom-center", "bottom-right"]
+                if position not in valid_positions:
+                    position = "bottom-center"
+                    print(f"Invalid position, using default: {position}")
+                
                 combined = combine_with_background(
                     foreground=results[2],
                     background=results[1],
@@ -381,14 +387,8 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
             except Exception as e:
                 print(f"Combination error: {str(e)}")
                 return results[1], results[2]
-        
-        return results[1], results[2]
-    except Exception as e:
-        print(f"Error in process_prompt: {str(e)}")
-        raise gr.Error(str(e))
-    finally:
-        clear_memory()
-        
+
+    
 def process_bbox(img: Image.Image, box_input: str) -> tuple[Image.Image, Image.Image]:
     try:
         if img is None or box_input.strip() == "":
@@ -675,15 +675,10 @@ def add_text_to_image(
         return input_image
 
     
-# 전역 변수 설정
-current_position = None  # position 상태를 전역으로 관리하기 위한 변수
-
-def update_position(new_position, *buttons):
-    print(f"Position updated to: {new_position}")  # 디버깅용
-    updates = []
-    for btn in buttons:
-        updates.append(gr.update(elem_classes=["position-btn", "selected"] if btn == new_position else ["position-btn"]))
-    return [new_position] + updates
+def update_position(new_position):
+    """위치 업데이트 함수"""
+    print(f"Position updated to: {new_position}")
+    return new_position
 
 def update_controls(bg_prompt):
     """배경 프롬프트 입력 여부에 따라 컨트롤 표시 업데이트"""
@@ -694,6 +689,8 @@ def update_controls(bg_prompt):
     ]
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    position = gr.State(value="bottom-center")  # 여기로 이동
+
     gr.HTML("""
         <div class="main-title">
             <h1>🎨 GiniGen Canvas-o3</h1>
@@ -859,20 +856,27 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         </style>
     """)
 
-    # 버튼 클릭 이벤트 바인딩
-    all_position_buttons = [btn_top_left, btn_top_center, btn_top_right,
-                          btn_middle_left, btn_middle_center, btn_middle_right,
-                          btn_bottom_left, btn_bottom_center, btn_bottom_right]
+# 버튼 클릭 이벤트 바인딩
+    position_mapping = {
+        btn_top_left: "top-left",
+        btn_top_center: "top-center",
+        btn_top_right: "top-right",
+        btn_middle_left: "middle-left",
+        btn_middle_center: "middle-center",
+        btn_middle_right: "middle-right",
+        btn_bottom_left: "bottom-left",
+        btn_bottom_center: "bottom-center",
+        btn_bottom_right: "bottom-right"
+    }
 
-    for btn, pos in zip(all_position_buttons, 
-                       ["top-left", "top-center", "top-right",
-                        "middle-left", "middle-center", "middle-right",
-                        "bottom-left", "bottom-center", "bottom-right"]):
+    for btn, pos in position_mapping.items():
         btn.click(
             fn=lambda p=pos: update_position(p),
-            outputs=[position] + all_position_buttons
+            outputs=position
         )
 
+ 
+
     # 이벤트 바인딩
     bg_prompt.change(
         fn=update_controls,