Spaces:
Running
on
Zero
Running
on
Zero
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]
|