Kims12 commited on
Commit
fbf3bce
ยท
verified ยท
1 Parent(s): 853fc8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -31
app.py CHANGED
@@ -206,35 +206,6 @@ def process_and_show_prompt(image1, image2, image3, prompt):
206
  logger.exception("์ฒ˜๋ฆฌ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ:")
207
  return None, f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}", prompt
208
 
209
- # ์˜ˆ์ œ ์‹คํ–‰ ํ•จ์ˆ˜๋“ค
210
- def run_example_1():
211
- """์˜ˆ์ œ 1: ์ด๋ฏธ์ง€ ๋ณ€๊ฒฝ ์˜ˆ์ œ"""
212
- input_path = os.path.join("down", "1_in-1.png")
213
- output_path = os.path.join("down", "1_out-1.webp")
214
- try:
215
- input_img = Image.open(input_path)
216
- except Exception as e:
217
- return None, f"์ž…๋ ฅ ์ด๋ฏธ์ง€ ์—ด๊ธฐ ์˜ค๋ฅ˜: {str(e)}", ""
218
- prompt = "#1 ์ด๋ฏธ์ง€์˜ [์ฒญ์ƒ‰ ์ƒ์–ด๋ ˆ๊ณ ๋ฅผ ๊ฒ€์€ ๊ณ ๋ž˜๋ ˆ๊ณ ]์œผ๋กœ ๋ณ€๊ฒฝํ•˜๋ผ."
219
- result_img, status, final_prompt = process_and_show_prompt(input_img, None, None, prompt)
220
- if result_img is not None:
221
- result_img.save(output_path, "WEBP")
222
- return result_img, status, final_prompt
223
-
224
- def run_example_2():
225
- """์˜ˆ์ œ 2: ๊ธ€์ž์ง€์šฐ๊ธฐ ์˜ˆ์ œ"""
226
- input_path = os.path.join("down", "2_in-1.png")
227
- output_path = os.path.join("down", "2_out-1.webp")
228
- try:
229
- input_img = Image.open(input_path)
230
- except Exception as e:
231
- return None, f"์ž…๋ ฅ ์ด๋ฏธ์ง€ ์—ด๊ธฐ ์˜ค๋ฅ˜: {str(e)}", ""
232
- prompt = "#1 ์ด๋ฏธ์ง€์˜ [์ค‘๊ตญ์–ด๋ฅผ ๋ชจ๋‘]๋ฅผ ์ œ๊ฑฐํ•˜๋ผ."
233
- result_img, status, final_prompt = process_and_show_prompt(input_img, None, None, prompt)
234
- if result_img is not None:
235
- result_img.save(output_path, "WEBP")
236
- return result_img, status, final_prompt
237
-
238
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค
239
  with gr.Blocks() as demo:
240
  gr.HTML(
@@ -262,7 +233,7 @@ with gr.Blocks() as demo:
262
  text_remove_btn = gr.Button("๊ธ€์ž์ง€์šฐ๊ธฐ")
263
  clothes_change_btn = gr.Button("์˜ท๋ฐ”๊พธ๊ธฐ")
264
  background_change_btn = gr.Button("๋ฐฐ๊ฒฝ๋ฐ”๊พธ๊ธฐ")
265
- composite_product_btn = gr.Button("์ด๋ฏธ์ง€ ํ•ฉ์„ฑ(์ƒํ’ˆํฌํ•จ)")
266
  submit_btn = gr.Button("์ด๋ฏธ์ง€ ์ƒ์„ฑ", variant="primary")
267
  with gr.Column():
268
  output_image = gr.Image(label="์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€")
@@ -291,7 +262,7 @@ with gr.Blocks() as demo:
291
  outputs=prompt_input
292
  )
293
  composite_product_btn.click(
294
- fn=lambda: "#1 ์ด๋ฏธ์ง€์™€ #2 ์ด๋ฏธ์ง€(๋˜๋Š” #3 ์ด๋ฏธ์ง€)๋ฅผ ์ž์—ฐ์Šค๋Ÿฝ๊ฒŒ ํ•ฉ์„ฑํ•˜๋ผ.",
295
  inputs=[],
296
  outputs=prompt_input
297
  )
 
206
  logger.exception("์ฒ˜๋ฆฌ ์ค‘ ์˜ค๋ฅ˜ ๋ฐœ์ƒ:")
207
  return None, f"์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}", prompt
208
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
209
  # Gradio ์ธํ„ฐํŽ˜์ด์Šค
210
  with gr.Blocks() as demo:
211
  gr.HTML(
 
233
  text_remove_btn = gr.Button("๊ธ€์ž์ง€์šฐ๊ธฐ")
234
  clothes_change_btn = gr.Button("์˜ท๋ฐ”๊พธ๊ธฐ")
235
  background_change_btn = gr.Button("๋ฐฐ๊ฒฝ๋ฐ”๊พธ๊ธฐ")
236
+ composite_product_btn = gr.Button("์ƒํ’ˆ๋ชจ๋ธํ”ผํŒ…")
237
  submit_btn = gr.Button("์ด๋ฏธ์ง€ ์ƒ์„ฑ", variant="primary")
238
  with gr.Column():
239
  output_image = gr.Image(label="์ƒ์„ฑ๋œ ์ด๋ฏธ์ง€")
 
262
  outputs=prompt_input
263
  )
264
  composite_product_btn.click(
265
+ fn=lambda: "#1 ์ƒํ’ˆ ์ด๋ฏธ์ง€์— #2 ์ด๋ฏธ์ง€ ๋˜๋Š” #3 ์ด๋ฏธ์ง€๋ฅผ [๋ชจ๋“  ์ด๋ฏธ์ง€์˜ ์ฃผ์š” ์š”์†Œ๋ฅผ ํฌํ•จํ•˜๊ณ , ํŠนํžˆ ์ƒํ’ˆ์ด ๋‹๋ณด์ด๋„๋ก ์กฐํ™”๋กญ๊ฒŒ ํ†ตํ•ฉํ•˜์—ฌ]์ž์—ฐ์Šค๋Ÿฝ๊ฒŒ ํ•ฉ์„ฑํ•˜๋ผ."",
266
  inputs=[],
267
  outputs=prompt_input
268
  )