Update app.py

app.py

@@ -353,7 +353,7 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
         if img is None or prompt.strip() == "":
             raise gr.Error("Please provide both image and prompt")
         
-        print(f"Processing with position: {position}, scale: {scale_percent}")  # 디버깅용
+        print(f"Processing with position: {position}, scale: {scale_percent}")
         
         try:
             prompt = translate_to_english(prompt)
@@ -366,8 +366,7 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
         
         if bg_prompt:
             try:
-                print(f"Using position: {position}")  # 디버깅용
-                # 위치 값 검증
+                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"]
@@ -381,12 +380,13 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
                     position=position,
                     scale_percent=scale_percent
                 )
-                return combined, results[2]
+                # 이미지를 RGB 모드로 변환하여 반환
+                return combined.convert('RGB'), results[2].convert('RGB')
             except Exception as e:
                 print(f"Combination error: {str(e)}")
-                return results[1], results[2]
+                return results[1].convert('RGB'), results[2].convert('RGB')
         
-        return results[1], results[2]  # 기본 반환 추가
+        return results[1].convert('RGB'), results[2].convert('RGB')
     except Exception as e:
         print(f"Error in process_prompt: {str(e)}")
         raise gr.Error(str(e))
@@ -888,29 +888,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             outputs=position
         )
 
-
-# 이벤트 바인딩
-    bg_prompt.change(
-        fn=update_controls,
-        inputs=bg_prompt,
-        outputs=[aspect_ratio, object_controls],
-        queue=False
-    )
-    
-    input_image.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn,
-        queue=False
-    )
-    
-    text_prompt.change(
-        fn=update_process_button,
-        inputs=[input_image, text_prompt],
-        outputs=process_btn,
-        queue=False
-    )
-
     process_btn.click(
         fn=process_prompt,
         inputs=[
@@ -922,7 +899,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             scale_slider
         ],
         outputs=[combined_image, extracted_image],
-        queue=True
+        queue=True,
+        show_progress=True
     )
 
     add_text_btn.click(
@@ -940,8 +918,46 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             font_choice
         ],
         outputs=combined_image,
+        queue=True,
         show_progress=True
     )
+
+    # 추가: 결과 이미지 업데이트를 위한 이벤트
+    combined_image.change(
+        fn=lambda x: x,
+        inputs=combined_image,
+        outputs=combined_image,
+        queue=False
+    )
+
+
+
+    
+# 이벤트 바인딩
+    bg_prompt.change(
+        fn=update_controls,
+        inputs=bg_prompt,
+        outputs=[aspect_ratio, object_controls],
+        queue=False
+    )
+    
+    input_image.change(
+        fn=update_process_button,
+        inputs=[input_image, text_prompt],
+        outputs=process_btn,
+        queue=False
+    )
+    
+    text_prompt.change(
+        fn=update_process_button,
+        inputs=[input_image, text_prompt],
+        outputs=process_btn,
+        queue=False
+    )
+
+
+
+
     
 demo.queue(max_size=5)
 demo.launch(