Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image
 import gradio as gr
 import logging
 import re
-import io
 from io import BytesIO
 
 from google import genai
@@ -69,22 +68,22 @@ def preprocess_prompt(prompt, image1, image2, image3):
     has_img1 = image1 is not None
     has_img2 = image2 is not None
     has_img3 = image3 is not None
-    
+
     if "#1" in prompt and not has_img1:
         prompt = prompt.replace("#1", "첫 번째 이미지(없음)")
     else:
         prompt = prompt.replace("#1", "첫 번째 이미지")
-        
+
     if "#2" in prompt and not has_img2:
         prompt = prompt.replace("#2", "두 번째 이미지(없음)")
     else:
         prompt = prompt.replace("#2", "두 번째 이미지")
-        
+
     if "#3" in prompt and not has_img3:
         prompt = prompt.replace("#3", "세 번째 이미지(없음)")
     else:
         prompt = prompt.replace("#3", "세 번째 이미지")
-    
+
     if "1. 이미지 변경" in prompt:
         desc_match = re.search(r'#1을 "(.*?)"으로 바꿔라', prompt)
         if desc_match:
@@ -92,7 +91,7 @@ def preprocess_prompt(prompt, image1, image2, image3):
             prompt = f"첫 번째 이미지를 {description}으로 변경해주세요. 원본 이미지의 주요 내용은 유지하되 새로운 스타일과 분위기로 재해석해주세요."
         else:
             prompt = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
-    
+
     elif "2. 글자지우기" in prompt:
         text_match = re.search(r'#1에서 "(.*?)"를 지워라', prompt)
         if text_match:
@@ -100,40 +99,37 @@ def preprocess_prompt(prompt, image1, image2, image3):
             prompt = f"첫 번째 이미지에서 '{text_to_remove}' 텍스트를 찾아 자연스럽게 제거해주세요. 텍스트가 있던 부분을 배경과 조화롭게 채워주세요."
         else:
             prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
-    
+
     elif "4. 옷바꾸기" in prompt:
         prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 변경해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
-    
+
     elif "5. 배경바꾸기" in prompt:
         prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
-    
+
     elif "6. 이미지 합성(상품포함)" in prompt:
         prompt = "첫 번째 이미지와 두 번째 이미지(또는 세 번째 이미지)를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 돋보이도록 조화롭게 통합해주세요."
-    
+
     prompt += " 이미지를 생성해주세요."
-    
     return prompt
 
 def generate_with_images(prompt, images):
     """
-    공식 문서에 기반한 올바른 API 호출 방식 구현
+    API 호출을 통해 이미지를 생성하고 결과 이미지를 반환합니다.
     """
     try:
         api_key = os.environ.get("GEMINI_API_KEY")
         if not api_key:
             return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
-            
+
         client = genai.Client(api_key=api_key)
         logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}")
-        
-        contents = []
-        contents.append(prompt)
-        
+
+        contents = [prompt]
         for idx, img in enumerate(images, 1):
             if img is not None:
                 contents.append(img)
                 logger.info(f"이미지 #{idx} 추가됨")
-        
+
         response = client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
             contents=contents,
@@ -145,7 +141,7 @@ def generate_with_images(prompt, images):
                 max_output_tokens=8192
             )
         )
-        
+
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             temp_path = tmp.name
             result_text = ""
@@ -170,16 +166,15 @@ def generate_with_images(prompt, images):
 
 def process_images_with_prompt(image1, image2, image3, prompt):
     """
-    3개의 이미지와 프롬프트를 처리하는 함수.
-    최종적으로 영어로 번역된 프롬프트(final_prompt)를 생성하여 API에 적용하고, 그 결과와 함께 반환합니다.
+    3개의 이미지와 프롬프트를 처리하여 최종 영어 프롬프트(final_prompt)를 생성한 후,
+    API를 호출하여 결과 이미지를 반환합니다.
     """
     try:
         images = [image1, image2, image3]
         valid_images = [img for img in images if img is not None]
         if not valid_images:
             return None, "적어도 하나의 이미지를 업로드해주세요.", ""
-        
-        # 프롬프트가 입력된 경우: 전처리 후 최종 영어 프롬프트 생성
+
         if prompt and prompt.strip():
             processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
             if re.search("[가-힣]", processed_prompt):
@@ -187,7 +182,6 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             else:
                 final_prompt = processed_prompt
         else:
-            # 프롬프트 미입력 시, 이미지 개수에 따른 기본 프롬프트 (영어)
             if len(valid_images) == 1:
                 final_prompt = "Please creatively transform this image into a more vivid and artistic version."
                 logger.info("Default prompt generated for single image")
@@ -197,7 +191,7 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             else:
                 final_prompt = "Please creatively composite these three images, combining their main elements into a cohesive and natural scene."
                 logger.info("Default prompt generated for three images")
-        
+
         result_img, status = generate_with_images(final_prompt, valid_images)
         return result_img, status, final_prompt
     except Exception as e:
@@ -205,9 +199,6 @@ def process_images_with_prompt(image1, image2, image3, prompt):
         return None, f"오류 발생: {str(e)}", prompt
 
 def process_and_show_prompt(image1, image2, image3, prompt):
-    """
-    최종적으로 이미지 생성 및 영어 최종 프롬프트를 반환하는 함수.
-    """
     try:
         result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt)
         return result_img, status, final_prompt
@@ -215,13 +206,42 @@ def process_and_show_prompt(image1, image2, image3, prompt):
         logger.exception("처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}", prompt
 
+# 예제 실행 함수들
+def run_example_1():
+    """예제 1: 이미지 변경 예제"""
+    input_path = os.path.join("down", "1_in-1.png")
+    output_path = os.path.join("down", "1_out-1.webp")
+    try:
+        input_img = Image.open(input_path)
+    except Exception as e:
+        return None, f"입력 이미지 열기 오류: {str(e)}", ""
+    prompt = "#1 이미지의 [청색 상어레고를 검은 고래레고]으로 변경하라."
+    result_img, status, final_prompt = process_and_show_prompt(input_img, None, None, prompt)
+    if result_img is not None:
+        result_img.save(output_path, "WEBP")
+    return result_img, status, final_prompt
+
+def run_example_2():
+    """예제 2: 글자지우기 예제"""
+    input_path = os.path.join("down", "2_in-1.png")
+    output_path = os.path.join("down", "2_out-1.webp")
+    try:
+        input_img = Image.open(input_path)
+    except Exception as e:
+        return None, f"입력 이미지 열기 오류: {str(e)}", ""
+    prompt = "#1 이미지의 [중국어를 모두]를 제거하라."
+    result_img, status, final_prompt = process_and_show_prompt(input_img, None, None, prompt)
+    if result_img is not None:
+        result_img.save(output_path, "WEBP")
+    return result_img, status, final_prompt
+
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>간단한 이미지 생성기</h1>
-            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다.</p>
+            <h1>간단한 이미지 생성기 (예제 포함)</h1>
+            <p>이미지를 업로드하거나 아래 예제 버튼을 클릭하면 처리된 결과가 down 폴더에 저장됩니다.</p>
         </div>
         """
     )
@@ -237,7 +257,6 @@ with gr.Blocks() as demo:
                 placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
                 label="프롬프트 (선택 사항)"
             )
-
             with gr.Row():
                 image_change_btn = gr.Button("이미지 변경")
                 text_remove_btn = gr.Button("글자지우기")
@@ -250,7 +269,7 @@ with gr.Blocks() as demo:
             output_text = gr.Textbox(label="상태 메시지")
             prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
     
-    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구 입력)
+    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
     image_change_btn.click(
         fn=lambda: "#1 이미지의 [다른 모습]으로 변경하라.",
         inputs=[],
@@ -282,16 +301,30 @@ with gr.Blocks() as demo:
         inputs=[image1_input, image2_input, image3_input, prompt_input],
         outputs=[output_image, output_text, prompt_display],
     )
-
+    
+    with gr.Row():
+        example1_btn = gr.Button("예제 1 실행")
+        example2_btn = gr.Button("예제 2 실행")
+    
+    example1_btn.click(
+        fn=run_example_1,
+        inputs=[],
+        outputs=[output_image, output_text, prompt_display]
+    )
+    example2_btn.click(
+        fn=run_example_2,
+        inputs=[],
+        outputs=[output_image, output_text, prompt_display]
+    )
+    
     gr.Markdown(
         """
         ### 사용 방법:
         
-        1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
+        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
         2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
-        3. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다.
-        4. **선택 옵션**: 위의 마크다운 형식의 옵션에서 원하는 항목을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고,
-           이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 실제 API 호출 프롬프트로 적용됩니다.
+        3. **예제 실행**: 아래 "예제 1 실행" 또는 "예제 2 실행" 버튼을 누르면 down 폴더의 입력 이미지를 처리하여 결과가 저장됩니다.
+        4. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구가 채워지고, 이후 Gemini‑2.0‑flash 모델을 통해 영어로 번역되어 실제 API 호출 프롬프트로 적용됩니다.
         
         > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
         """