Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -291,8 +291,9 @@ def _process(img: Image.Image, prompt: str | BoundingBox | None, bg_prompt: str
|
|
291 |
masked_alpha = apply_mask(img, mask, defringe=True)
|
292 |
|
293 |
if bg_prompt:
|
|
|
294 |
background = generate_background(bg_prompt, aspect_ratio)
|
295 |
-
combined =
|
296 |
else:
|
297 |
combined = Image.alpha_composite(Image.new("RGBA", masked_alpha.size, "white"), masked_alpha)
|
298 |
|
@@ -317,7 +318,6 @@ def on_change_prompt(img: Image.Image | None, prompt: str | None, bg_prompt: str
|
|
317 |
return gr.update(interactive=bool(img and prompt))
|
318 |
|
319 |
|
320 |
-
# process_prompt ํจ์ ์์
|
321 |
def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
|
322 |
aspect_ratio: str = "1:1", position: str = "bottom-center",
|
323 |
scale_percent: float = 100) -> tuple[Image.Image, Image.Image]:
|
@@ -334,8 +334,13 @@ def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
|
|
334 |
results, _ = _process(img, prompt, bg_prompt, aspect_ratio)
|
335 |
|
336 |
if bg_prompt:
|
337 |
-
#
|
338 |
-
combined = combine_with_background(
|
|
|
|
|
|
|
|
|
|
|
339 |
return combined, results[2]
|
340 |
|
341 |
return results[1], results[2]
|
|
|
291 |
masked_alpha = apply_mask(img, mask, defringe=True)
|
292 |
|
293 |
if bg_prompt:
|
294 |
+
# ๋ฐฐ๊ฒฝ๋ง ์์ฑํ๊ณ ํฉ์ฑ์ ํ์ง ์์
|
295 |
background = generate_background(bg_prompt, aspect_ratio)
|
296 |
+
combined = background # ๋ฐฐ๊ฒฝ ์ด๋ฏธ์ง๋ง ๋ฐํ
|
297 |
else:
|
298 |
combined = Image.alpha_composite(Image.new("RGBA", masked_alpha.size, "white"), masked_alpha)
|
299 |
|
|
|
318 |
return gr.update(interactive=bool(img and prompt))
|
319 |
|
320 |
|
|
|
321 |
def process_prompt(img: Image.Image, prompt: str, bg_prompt: str | None = None,
|
322 |
aspect_ratio: str = "1:1", position: str = "bottom-center",
|
323 |
scale_percent: float = 100) -> tuple[Image.Image, Image.Image]:
|
|
|
334 |
results, _ = _process(img, prompt, bg_prompt, aspect_ratio)
|
335 |
|
336 |
if bg_prompt:
|
337 |
+
# ์ฌ๊ธฐ์ ํ ๋ฒ๋ง ํฉ์ฑ ์ํ
|
338 |
+
combined = combine_with_background(
|
339 |
+
foreground=results[2], # ์ถ์ถ๋ ์ค๋ธ์ ํธ
|
340 |
+
background=results[1], # ์์ฑ๋ ๋ฐฐ๊ฒฝ
|
341 |
+
position=position,
|
342 |
+
scale_percent=scale_percent
|
343 |
+
)
|
344 |
return combined, results[2]
|
345 |
|
346 |
return results[1], results[2]
|