ginipick commited on
Commit
1dd4709
ยท
verified ยท
1 Parent(s): 64c13d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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 = combine_with_background(masked_alpha, background)
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(results[2], results[1], position, scale_percent)
 
 
 
 
 
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]