Update app.py

app.py

@@ -24,15 +24,79 @@ def save_binary_file(file_name, data):
 
 def preprocess_prompt(prompt, image1, image2, image3):
     """
-    입력된 프롬프트가 영어로만 되어 있는지 확인하고,
-    단순히 "Please generate the image."를 덧붙여 반환합니다.
+    프롬프트를 처리하고 기능 명령을 해석
     """
-    # 프롬프트에 한글이 포함되어 있으면 에러 발생
-    if re.search(r'[ㄱ-ㅎ가-힣]', prompt):
-        raise ValueError("Error: Prompt must be in English only.")
-
-    # 불필요한 기능 관련 처리는 제거하고 단순 명령어 추가
-    prompt = prompt.strip() + " Please generate the image."
+    # 기존 preprocess_prompt 함수 코드 유지
+    # 이미지 없는 참조 확인 및 처리
+    has_img1 = image1 is not None
+    has_img2 = image2 is not None
+    has_img3 = image3 is not None
+    
+    # #1, #2, #3 참조를 설명으로 변환 (이미지가 없는 경우 무시)
+    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:
+            description = desc_match.group(1)
+            prompt = f"첫 번째 이미지를 {description}으로 변경해주세요. 원본 이미지의 주요 내용은 유지하되 새로운 스타일과 분위기로 재해석해주세요."
+        else:
+            prompt = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+    
+    elif "2. 글자지우기" in prompt:
+        # 지울 텍스트 추출을 시도하지만 실패해도 기본 프롬프트 제공
+        text_match = re.search(r'#1에서 "(.*?)"를 지워라', prompt)
+        if text_match:
+            text_to_remove = text_match.group(1)
+            prompt = f"첫 번째 이미지에서 '{text_to_remove}' 텍스트를 찾아 자연스럽게 제거해주세요. 텍스트가 있던 부분을 배경과 조화롭게 채워주세요."
+        else:
+            prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
+    
+    elif "3. 얼굴바꾸기" in prompt:
+        prompt = "첫 번째 이미지의 인물 얼굴을 두 번째 이미지의 얼굴로 자연스럽게 교체해주세요. 얼굴의 표정과 특징은 두 번째 이미지를 따르되, 나머지 부분은 첫 번째 이미지를 유지해주세요."
+    
+    elif "4. 옷바꾸기" in prompt:
+        # 여러 이미지 참조 처리
+        if "#3" in prompt or "또는 #3" in prompt:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 또는 세 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 참조 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+        else:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+    
+    elif "5. 배경바꾸기" in prompt:
+        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 자연스럽게 교체해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
+    
+    elif "6. 이미지 합성(상품포함)" in prompt:
+        # 여러 이미지 참조 처리
+        if "#3" in prompt or "또는 #3" in prompt:
+            prompt = "첫 번째 이미지와 두 번째, 세 번째 이미지를 자연스럽게 합성해주세요. 모든 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
+        else:
+            prompt = "첫 번째 이미지와 두 번째 이미지를 자연스럽게 합성해주세요. 두 이미지의 주요 요소를 포함하고, 특히 상품이 잘 보이도록 조화롭게 통합해주세요."
+    
+    elif "7. 이미지 합성(스타일적용)" in prompt:
+        prompt = "첫 번째 이미지의 내용을 두 번째 이미지의 스타일로 변환해주세요. 첫 번째 이미지의 주요 피사체와 구도는 유지하되, 두 번째 이미지의 예술적 스타���, 색상, 질감을 적용해주세요."
+    
+    # 간단한 색상 변경 요청 처리
+    elif "을 붉은색으로 바꿔라" in prompt or "를 붉은색으로 바꿔라" in prompt:
+        prompt = "첫 번째 이미지를 붉은색 톤으로 변경해주세요. 전체적인 색상을 붉은 계열로 조정하고 자연스러운 느낌을 유지해주세요."
+    
+    # 명확한 이미지 생성 요청 추가
+    prompt += " 이미지를 생성해주세요."
+    
     return prompt
 
 def generate_with_images(prompt, images):
@@ -43,7 +107,7 @@ def generate_with_images(prompt, images):
         # API 키 확인
         api_key = os.environ.get("GEMINI_API_KEY")
         if not api_key:
-            return None, "API key is not set. Please check your environment variables."
+            return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
             
         # Gemini 클라이언트 초기화
         client = genai.Client(api_key=api_key)
@@ -56,11 +120,11 @@ def generate_with_images(prompt, images):
         # 텍스트 프롬프트 추가
         contents.append(prompt)
         
-        # 이미지 추가 (이미지가 없으면 텍스트만 전송)
+        # 이미지 추가
         for idx, img in enumerate(images, 1):
             if img is not None:
                 contents.append(img)
-                logger.info(f"Image #{idx} added.")
+                logger.info(f"이미지 #{idx} 추가됨")
         
         # 생성 설정 - 공식 문서에 따라 responseModalities 설정
         response = client.models.generate_content(
@@ -86,94 +150,140 @@ def generate_with_images(prompt, images):
             for part in response.candidates[0].content.parts:
                 if hasattr(part, 'text') and part.text:
                     result_text += part.text
-                    logger.info(f"Response text: {part.text}")
+                    logger.info(f"응답 텍스트: {part.text}")
                 elif hasattr(part, 'inline_data') and part.inline_data:
                     save_binary_file(temp_path, part.inline_data.data)
                     image_found = True
-                    logger.info("Image extracted successfully from response.")
+                    logger.info("응답에서 이미지 추출 성공")
             
             if not image_found:
-                return None, f"API did not generate an image. Response text: {result_text}"
+                return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
                     
             # 결과 이미지 반환
             result_img = Image.open(temp_path)
             if result_img.mode == "RGBA":
                 result_img = result_img.convert("RGB")
                 
-            return result_img, f"Image generated successfully. {result_text}"
+            return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
             
     except Exception as e:
-        logger.exception("Error during image generation:")
-        return None, f"Error occurred: {str(e)}"
+        logger.exception("이미지 생성 중 오류 발생:")
+        return None, f"오류 발생: {str(e)}"
 
 def process_images_with_prompt(image1, image2, image3, prompt):
     """
-    3개의 이미지와 프롬프트를 처리하는 함수.
-    이미지 입력이 없어도 프롬프트만으로 API 호출을 진행합니다.
+    3개의 이미지와 프롬프트를 처리하는 함수
     """
     try:
-        # 이미지 개수 확인 (이미지 없이도 진행)
+        # 이미지 개수 확인
         images = [image1, image2, image3]
         valid_images = [img for img in images if img is not None]
         
-        # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성 (영어)
+        if not valid_images:
+            return None, "적어도 하나의 이미지를 업로드해주세요."
+        
+        # 프롬프트 처리
         if not prompt or not prompt.strip():
-            if len(valid_images) == 0:
-                prompt = "Please generate an image based on the description."
-                logger.info("Auto prompt generated for no image input.")
-            elif len(valid_images) == 1:
-                prompt = "Please creatively transform this image into a more vivid and artistic version."
-                logger.info("Auto prompt generated for a single image.")
+            # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성
+            if len(valid_images) == 1:
+                prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+                logger.info("단일 이미지 프롬프트 자동 생성")
             elif len(valid_images) == 2:
-                prompt = "Please seamlessly blend these two images, integrating their elements harmoniously into a single image."
-                logger.info("Auto prompt generated for two images.")
+                prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+                logger.info("두 이미지 합성 프롬프트 자동 생성")
             else:
-                prompt = "Please creatively composite these three images, incorporating their key elements into a natural and coherent scene."
-                logger.info("Auto prompt generated for three images.")
+                prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
+                logger.info("세 이미지 합성 프롬프트 자동 생성")
         else:
+            # 프롬프트 전처리 및 기능 명령 해석
             prompt = preprocess_prompt(prompt, image1, image2, image3)
         
         # 새로운 API 호출 방식 사용
         return generate_with_images(prompt, valid_images)
             
     except Exception as e:
-        logger.exception("Error during image processing:")
-        return None, f"Error occurred: {str(e)}"
+        logger.exception("이미지 처리 중 오류 발생:")
+        return None, f"오류 발생: {str(e)}"
+
+# 기능 선택 콜백 (기존 코드 유지)
+def update_prompt_from_function(function_choice, custom_text=""):
+    function_templates = {
+        "1. 이미지 변경": f'#1을 "{custom_text if custom_text else "원하는 설명"}"으로 바꿔라',
+        "2. 글자지우기": f'#1에서 "{custom_text if custom_text else "지울 텍스트"}"를 지워라',
+        "3. 얼굴바꾸기": "#1의 인물을 #2의 얼굴로 바꿔라",
+        "4. 옷바꾸기": "#1의 인물에 #2 또는 #3의 옷으로 변경하라",
+        "5. 배경바꾸기": "#1의 이미지에 #2의 배경으로 자연스럽게 바꿔라",
+        "6. 이미지 합성(상품포함)": "#1와 #2 또는 #3의 를 합성하라",
+        "7. 이미지 합성(스타일적용)": "#1와 #2를 스타일로 변환하라"
+    }
+    
+    return function_templates.get(function_choice, "")
 
-# Gradio 인터페이스 (기능 선택, 기능 적용, 커스텀 텍스트 관련 요소 제거)
+# Gradio 인터페이스 (기존 코드 유지)
 with gr.Blocks() as demo:
     gr.HTML(
         """
         <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>Simple Image Generator</h1>
-            <p>Upload an image (or leave empty) and click generate to create an image based on the English prompt.</p>
+            <h1>간단한 이미지 생성기</h1>
+            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 아래 기능을 선택할 수 있습니다.</p>
         </div>
         """
     )
     
     with gr.Row():
         with gr.Column():
-            # 3개의 이미지 입력 (이미지 없이도 실행 가능)
+            # 3개의 이미지 입력
             with gr.Row():
-                image1_input = gr.Image(type="pil", label="Image 1", image_mode="RGB")
-                image2_input = gr.Image(type="pil", label="Image 2", image_mode="RGB")
-                image3_input = gr.Image(type="pil", label="Image 3", image_mode="RGB")
+                image1_input = gr.Image(type="pil", label="#1", image_mode="RGB")
+                image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
+                image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
             
-            # 프롬프트 입력 (영어로만 입력)
+            # 기능 선택 드롭다운과 커스텀 텍스트 입력
+            with gr.Row():
+                function_dropdown = gr.Dropdown(
+                    choices=[
+                        "1. 이미지 변경", 
+                        "2. 글자지우기", 
+                        "3. 얼굴바꾸기", 
+                        "4. 옷바꾸기",
+                        "5. 배경바꾸기", 
+                        "6. 이미지 합성(상품포함)", 
+                        "7. 이미지 합성(스타일적용)"
+                    ],
+                    label="기능 선택",
+                    value=None
+                )
+                custom_text_input = gr.Textbox(
+                    label="커스텀 텍스트 (1, 2번 기능용)", 
+                    placeholder="예: 붉은색, 수채화 스타일, 중국어..."
+                )
+            
+            apply_function_btn = gr.Button("기능 적용")
+            
+            # 프롬프트 입력 (선택 사항)
             prompt_input = gr.Textbox(
                 lines=3,
-                placeholder="Enter the prompt in English.",
-                label="Prompt (Required: English only)"
+                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
+                label="프롬프트 (선택 사항)"
             )
             
             # 생성 버튼
-            submit_btn = gr.Button("Generate Image", variant="primary")
+            submit_btn = gr.Button("이미지 생성", variant="primary")
         
         with gr.Column():
             # 결과 출력
-            output_image = gr.Image(label="Generated Image")
-            output_text = gr.Textbox(label="Status Message")
-            prompt_display = gr.Textbox(label="Used Prompt", visible=True)
+            output_image = gr.Image(label="생성된 이미지")
+            output_text = gr.Textbox(label="상태 메시지")
+            
+            # 사용된 프롬프트 표시
+            prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
+    
+    # 기능 적용 버튼 이벤트
+    apply_function_btn.click(
+        fn=update_prompt_from_function,
+        inputs=[function_dropdown, custom_text_input],
+        outputs=[prompt_input]
+    )
     
     # 이미지 생성 버튼 클릭 이벤트
     def process_and_show_prompt(image1, image2, image3, prompt):
@@ -182,30 +292,25 @@ with gr.Blocks() as demo:
         valid_images = [img for img in images if img is not None]
         
         try:
-            # 만약 사용자가 프롬프트를 입력했다면 영어만 포함되었는지 확인
-            if prompt and re.search(r'[ㄱ-ㅎ가-힣]', prompt):
-                return None, "Error: Prompt must be in English only.", prompt
-            
-            # 프롬프트가 없으면 자동 생성 (영어)
+            # 자동 프롬프트 생성 또는 프롬프트 전처리
             auto_prompt = prompt
             if not prompt or not prompt.strip():
-                if len(valid_images) == 0:
-                    auto_prompt = "Please generate an image based on the description."
-                elif len(valid_images) == 1:
-                    auto_prompt = "Please creatively transform this image into a more vivid and artistic version."
+                if len(valid_images) == 1:
+                    auto_prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
                 elif len(valid_images) == 2:
-                    auto_prompt = "Please seamlessly blend these two images, integrating their elements harmoniously into a single image."
+                    auto_prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
                 else:
-                    auto_prompt = "Please creatively composite these three images, incorporating their key elements into a natural and coherent scene."
+                    auto_prompt = "이 세 이미지를 창의적으로 합성해주세요. 모든 이미지의 주요 요소를 포함하되 자연스럽고 일관된 하나의 장면으로 만들어주세요."
             else:
                 auto_prompt = preprocess_prompt(prompt, image1, image2, image3)
             
+            # 이미지 생성 함수 호출
             result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
             
             return result_img, status, auto_prompt
         except Exception as e:
-            logger.exception("Error during processing:")
-            return None, f"Error occurred: {str(e)}", prompt
+            logger.exception("처리 중 오류 발생:")
+            return None, f"오류 발생: {str(e)}", prompt
     
     submit_btn.click(
         fn=process_and_show_prompt,
@@ -215,14 +320,18 @@ with gr.Blocks() as demo:
 
     gr.Markdown(
         """
-        ### Instructions:
+        ### 사용 방법:
+        
+        1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
+        2. **기능 사용**: 드롭다운에서 원하는 기능을 선택하고 '기능 적용' 버튼을 클릭하세요
+        3. **커스텀 텍스트**: 이미지 변경이나 글자지우기 기능을 선택할 때 추가 설명을 입력할 수 있습니다
+        4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
+        5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
         
-        1. **Auto Generation**: You can leave the image upload empty and the system will generate an image based solely on the prompt.
-        2. **Prompt Requirement**: Enter the prompt in English only.
-        3. **Image Reference**: The app supports up to three image inputs.
+        > **팁**: 기능을 선택한 후에도 프롬프트를 직접 수정할 수 있습니다
         """
     )
 
 # 애플리케이션 실행
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)