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]