Update app.py

app.py

@@ -6,16 +6,18 @@ import io
 def merge_images(foreground, background):
     # foreground와 background를 바이트 배열로 변환
     foreground_bytes = io.BytesIO()
-    foreground.save(foreground_bytes, format=foreground.format)
+    foreground.save(foreground_bytes, format="PNG")
+    foreground_bytes.seek(0)  # 포인터를 처음으로 이동
     background_bytes = io.BytesIO()
-    background.save(background_bytes, format=background.format)
+    background.save(background_bytes, format="PNG")
+    background_bytes.seek(0)  # 포인터를 처음으로 이동
     
     # 배경 제거
-    foreground_no_bg_bytes = remove(foreground_bytes.getvalue())
+    foreground_no_bg_bytes = remove(foreground_bytes.read())
     
     # 바이트 데이터를 BytesIO 스트림으로 변환하여 PIL 이미지로 로드
     foreground_no_bg = Image.open(io.BytesIO(foreground_no_bg_bytes)).convert("RGBA")
-    background_image = background.convert("RGBA")
+    background_image = Image.open(background_bytes).convert("RGBA")
     
     # 배경 이미지 크기 조정
     background_image = background_image.resize(foreground_no_bg.size)
@@ -28,8 +30,8 @@ def merge_images(foreground, background):
 
 with gr.Blocks() as demo:
     with gr.Row():
-        foreground_input = gr.Image(type="pil", label="Upload your main image", tool="editor")
-        background_input = gr.Image(type="pil", label="Upload your background image", tool="editor")
+        foreground_input = gr.Image(type="pil", label="Upload your main image")
+        background_input = gr.Image(type="pil", label="Upload your background image")
     image_output = gr.Image(type="pil", label="Merged Image")
     
     btn = gr.Button("Merge Images")