Update app.py

app.py

@@ -4,8 +4,8 @@ from PIL import Image
 import gradio as gr
 import logging
 import re
+import io
 from io import BytesIO
-import time
 
 from google import genai
 from google.genai import types
@@ -22,162 +22,111 @@ def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def translate_prompt_to_english(prompt):
-    """
-    입력된 프롬프트에 한글이 포함되어 있으면 Gemini‑2.0‑flash 모델을 사용하여 영어로 번역합니다.
-    한글이 없으면 원본 프롬프트를 그대로 반환합니다.
-    중요: #1, #2, #3 태그는 번역 전후에 반드시 보존되어야 합니다.
-    """
-    if not re.search("[가-힣]", prompt):
-        return prompt
-    
-    # #1, #2, #3 태그를 임시 토큰으로 대체하여 보존
-    prompt = prompt.replace("#1", "IMAGE_TAG_ONE")
-    prompt = prompt.replace("#2", "IMAGE_TAG_TWO")
-    prompt = prompt.replace("#3", "IMAGE_TAG_THREE")
-    
-    try:
-        api_key = os.environ.get("GEMINI_API_KEY")
-        if not api_key:
-            logger.error("Gemini API 키가 설정되지 않았습니다.")
-            # 임시 토큰을 원래 태그로 복원
-            prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
-            prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
-            prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
-            return prompt
-            
-        client = genai.Client(api_key=api_key)
-        translation_prompt = f"""
-        Translate the following Korean text to English:
-        
-        {prompt}
-        
-        IMPORTANT: The tokens IMAGE_TAG_ONE, IMAGE_TAG_TWO, and IMAGE_TAG_THREE are special tags 
-        and must be preserved exactly as is in your translation. Do not translate these tokens.
-        """
-        
-        logger.info(f"Translation prompt: {translation_prompt}")
-        response = client.models.generate_content(
-            model="gemini-2.0-flash",
-            contents=[translation_prompt],
-            config=types.GenerateContentConfig(
-                response_modalities=['Text'],
-                temperature=0.2,
-                top_p=0.95,
-                top_k=40,
-                max_output_tokens=512
-            )
-        )
-        
-        translated_text = ""
-        for part in response.candidates[0].content.parts:
-            if hasattr(part, 'text') and part.text:
-                translated_text += part.text
-        
-        if translated_text.strip():
-            # 번역된 텍스트에서 임시 토큰을 원래 태그로 복원
-            translated_text = translated_text.replace("IMAGE_TAG_ONE", "#1")
-            translated_text = translated_text.replace("IMAGE_TAG_TWO", "#2")
-            translated_text = translated_text.replace("IMAGE_TAG_THREE", "#3")
-            logger.info(f"Translated text: {translated_text.strip()}")
-            return translated_text.strip()
-        else:
-            logger.warning("번역 결과가 없습니다. 원본 프롬프트 사용")
-            # 임시 토큰을 원래 태그로 복원
-            prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
-            prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
-            prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
-            return prompt
-    except Exception as e:
-        logger.exception("번역 중 오류 발생:")
-        # 임시 토큰을 원래 태그로 복원
-        prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
-        prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
-        prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
-        return prompt
-
 def preprocess_prompt(prompt, image1, image2, image3):
     """
     프롬프트를 처리하고 기능 명령을 해석
     """
+    # 기존 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:
-        prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 변경해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
-
+        # 여러 이미지 참조 처리
+        if "#3" in prompt or "또는 #3" in prompt:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 또는 세 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 참조 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+        else:
+            prompt = "첫 번째 이미지의 인물 의상을 두 번째 이미지의 의상으로 자연스럽게 교체해주세요. 의상의 스타일과 색상은 두 번째 이미지를 따르되, 신체 비율과 포즈는 첫 번째 이미지를 유지해주세요."
+    
     elif "5. 배경바꾸기" in prompt:
-        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 변경해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
-
+        prompt = "첫 번째 이미지의 배경을 두 번째 이미지의 배경으로 자연스럽게 교체해주세요. 첫 번째 이미지의 주요 피사체는 유지하고, 두 번째 이미지의 배경과 조화롭게 합성해주세요."
+    
     elif "6. 이미지 합성(상품포함)" in prompt:
-        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, variation_index=0):
+def generate_with_images(prompt, images):
     """
-    API 호출을 통해 이미지를 생성하고 결과 이미지를 반환합니다.
-    variation_index로 다양한 변화를 줍니다.
-    생성된 이미지는 일관된 크기로 조정됩니다.
+    공식 문서에 기반한 올바른 API 호출 방식 구현
     """
     try:
+        # API 키 확인
         api_key = os.environ.get("GEMINI_API_KEY")
         if not api_key:
             return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
-
+            
+        # Gemini 클라이언트 초기화
         client = genai.Client(api_key=api_key)
-        logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
-
-        # 변형 인덱스에 따라 프롬프트에 약간의 변화 추가
-        variation_suffixes = [
-            " 첫 번째 변형으로 만들어주세요.",
-            " 두 번째 변형으로 더 생생하게 만들어주세요.",
-            " 세 번째 변형으로 더 창의적으로 만들어주세요.",
-            " 네 번째 변형으로 색감을 더 강조해서 만들어주세요."
-        ]
         
-        if variation_index < len(variation_suffixes):
-            prompt = prompt + variation_suffixes[variation_index]
-
-        contents = [prompt]
+        logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}")
+        
+        # 컨텐츠 준비
+        contents = []
+        
+        # 텍스트 프롬프트 추가
+        contents.append(prompt)
+        
+        # 이미지 추가
         for idx, img in enumerate(images, 1):
             if img is not None:
                 contents.append(img)
                 logger.info(f"이미지 #{idx} 추가됨")
-
+        
+        # 생성 설정 - 공식 문서에 따라 responseModalities 설정
         response = client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
             contents=contents,
@@ -189,11 +138,15 @@ def generate_with_images(prompt, images, variation_index=0):
                 max_output_tokens=8192
             )
         )
-
+        
+        # 임시 파일 생성
         with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
             temp_path = tmp.name
+            
             result_text = ""
             image_found = False
+            
+            # 응답 처리
             for part in response.candidates[0].content.parts:
                 if hasattr(part, 'text') and part.text:
                     result_text += part.text
@@ -202,210 +155,183 @@ def generate_with_images(prompt, images, variation_index=0):
                     save_binary_file(temp_path, part.inline_data.data)
                     image_found = True
                     logger.info("응답에서 이미지 추출 성공")
+            
             if not image_found:
                 return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
+                    
+            # 결과 이미지 반환
             result_img = Image.open(temp_path)
             if result_img.mode == "RGBA":
                 result_img = result_img.convert("RGB")
                 
-            # 이미지 크기를 512x512로 조정하여 일관된 출력 크기 유지
-            target_size = (512, 512)
-            result_img = result_img.resize(target_size, Image.Resampling.LANCZOS)
-            
             return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
+            
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0):
+def process_images_with_prompt(image1, image2, image3, prompt):
     """
-    3개의 이미지와 프롬프트를 처리하여 최종 영어 프롬프트(final_prompt)를 생성한 후,
-    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 prompt and prompt.strip():
-            processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
-            if re.search("[가-힣]", processed_prompt):
-                final_prompt = translate_prompt_to_english(processed_prompt)
-            else:
-                final_prompt = processed_prompt
-        else:
+            return None, "적어도 하나의 이미지를 업로드해주세요."
+        
+        # 프롬프트 처리
+        if not prompt or not prompt.strip():
+            # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성
             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")
+                prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+                logger.info("단일 이미지 프롬프트 자동 생성")
             elif len(valid_images) == 2:
-                final_prompt = "Please seamlessly composite these two images, integrating their key elements harmoniously into a single image."
-                logger.info("Default prompt generated for two images")
+                prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+                logger.info("두 이미지 합성 프롬프트 자동 생성")
             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, variation_index)
-        return result_img, status, final_prompt
+                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("이미지 처리 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}", prompt
+        return None, f"오류 발생: {str(e)}"
 
-def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progress()):
-    """
-    여러 개의 이미지를 차례대로 생성합니다.
-    """
-    results = []
-    statuses = []
-    prompts = []
+# 기능 선택 콜백 (기존 코드 유지)
+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를 스타일로 변환하라"
+    }
     
-    num_images = 4  # 생성할 이미지 수
-    
-    progress(0, desc="이미지 생성 준비 중...")
-    
-    for i in range(num_images):
-        progress((i / num_images), desc=f"{i+1}/{num_images} 이미지 생성 중...")
-        result_img, status, final_prompt = process_images_with_prompt(image1, image2, image3, prompt, i)
-        
-        if result_img is not None:
-            results.append(result_img)
-            statuses.append(f"이미지 #{i+1}: {status}")
-            prompts.append(f"이미지 #{i+1}: {final_prompt}")
-        else:
-            # 에러가 발생한 경우에도 결과 목록에 None을 추가
-            results.append(None)
-            statuses.append(f"이미지 #{i+1} 생성 실패: {status}")
-            prompts.append(f"이미지 #{i+1}: {final_prompt}")
-        
-        # API 호출 사이에 약간의 간격을 두어 속도 제한 방지
-        time.sleep(1)
-    
-    progress(1.0, desc="이미지 생성 완료!")
-    
-    # 결과가 없는 경우 빈 슬롯 채우기
-    while len(results) < 4:
-        results.append(None)
-    
-    # 상태 메시지와 프롬프트 정보 결합
-    combined_status = "\n".join(statuses)
-    combined_prompts = "\n".join(prompts)
-    
-    return results[0], results[1], results[2], results[3], combined_status, combined_prompts
+    return function_templates.get(function_choice, "")
 
-# Gradio 인터페이스
-with gr.Blocks(css="""
-    .gradio-container {max-width: 650px; margin: 0 auto;}
-    .output-image {height: 300px; width: 300px;}
-    .input-image {height: 200px; width: 200px;}
-    .image-row {display: flex; justify-content: center;}
-""") as demo:
-    # 메인 컨테이너 시작
-    with gr.Column():
-        gr.HTML(
-            """
-            <div style="text-align: center; margin-bottom: 1rem;">
-                <h1 style="margin-bottom: 0.5rem;">4장 이미지 생성기</h1>
-                <p style="margin-bottom: 0.2rem;">이미지를 업로드하고 "이미지 생성" 버튼을 클릭하면</p>
-                <p>차례로 4장의 이미지가 생성됩니다.</p>
-            </div>
-            """
-        )
-        
-        # 입력 섹션
-        gr.HTML("<h3 style='text-align: center; margin-top: 1rem; margin-bottom: 0.5rem;'>입력 이미지</h3>")
-        
-        # 입력 이미지 UI를 적절한 간격으로 세로 배치
+# Gradio 인터페이스 (기존 코드 유지)
+with gr.Blocks() as demo:
+    gr.HTML(
+        """
+        <div style="text-align: center; margin-bottom: 1rem;">
+            <h1>간단한 이미지 생성기</h1>
+            <p>이미지를 업로드하�� 바로 실행하면 자동으로 합성합니다. 또는 아래 기능을 선택할 수 있습니다.</p>
+        </div>
+        """
+    )
+    
+    with gr.Row():
         with gr.Column():
+            # 3개의 이미지 입력
             with gr.Row():
-                image1_input = gr.Image(type="pil", label="#1", height=200, width=200)
-                image2_input = gr.Image(type="pil", label="#2", height=200, width=200)
-            with gr.Row():
-                image3_input = gr.Image(type="pil", label="#3", height=200, width=200)
-                placeholder_input = gr.Textbox(visible=False)  # 레이아웃 조정용 더미 컴포넌트
-        
-        # 프롬프트 입력
-        prompt_input = gr.Textbox(
-            lines=3,
-            placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
-            label="프롬프트 (선택 사항)"
-        )
-        
-        # 버튼 스타일 개선 
-        with gr.Column(variant="panel"):
-            gr.HTML("<h4 style='margin-bottom: 0.5rem; text-align: center; width: 100%;'>작업 선택</h4>")
+                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():
-                image_change_btn = gr.Button("이미지변경", scale=1)
-                text_remove_btn = gr.Button("글자지우기", scale=1)
-                clothes_change_btn = gr.Button("가상 상품착용", scale=1)
-        
-        with gr.Row():
-                background_change_btn = gr.Button("배경바꾸기", scale=1)
-                composite_product_btn = gr.Button("부분지우기", scale=1)
-        
-        # 생성 버튼을 더 눈에 띄게 만들기
-        submit_btn = gr.Button("이미지 생성 (4장)", variant="primary", size="lg")
-        
-        # 출력 섹션
-        gr.HTML("<h3 style='text-align: center; margin-top: 1.5rem; margin-bottom: 0.5rem;'>생성된 이미지</h3>")
+                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="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
+                label="프롬프트 (선택 사항)"
+            )
+            
+            # 생성 버튼
+            submit_btn = gr.Button("이미지 생성", variant="primary")
         
-        # 출력 이미지 UI를 세로로 배치
         with gr.Column():
-            with gr.Row():
-                output_image1 = gr.Image(label="생성된 이미지 #1", height=300, width=300)
-                output_image2 = gr.Image(label="생성된 이미지 #2", height=300, width=300)
-            with gr.Row():
-                output_image3 = gr.Image(label="생성된 이미지 #3", height=300, width=300)
-                output_image4 = gr.Image(label="생성된 이미지 #4", height=300, width=300)
-        
-        # 상태 및 프롬프트 정보
-        output_text = gr.Textbox(label="상태 메시지")
-        prompt_display = gr.Textbox(label="사용된 프롬프트 (영어)", visible=True)
+            # 결과 출력
+            output_image = gr.Image(label="생성된 이미지")
+            output_text = gr.Textbox(label="상태 메시지")
+            
+            # 사용된 프롬프트 표시
+            prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
     
-    # 선택 옵션 버튼 클릭 시 프롬프트 입력란 업데이트 (한국어 문구)
-    image_change_btn.click(
-        fn=lambda: "(#1의 여성)의 이미지를 살짝 뒤로 돌아보는 모습으로 최대한 이전 seed를 유지한테 자연스럽게 변경하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    text_remove_btn.click(
-        fn=lambda: "(#1 이미지)에 있는 중국어를 모두 제거하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    clothes_change_btn.click(
-        fn=lambda: "(#1의 여성모델)이 신체 비육과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    background_change_btn.click(
-        fn=lambda: "(#1의 여성모델)이 (#2 이미지의 배경)을 주요 피사체는 그대로 유지하여 두 이미지의 분위기가 자연스럽게 어우러지도록 생성하라.",
-        inputs=[],
-        outputs=prompt_input
-    )
-    composite_product_btn.click(
-        fn=lambda: "#1 이미지에서 [제거할 영역 또는 대상을 자세히 작성해주세요]을 제거한 후, 그 자리를 주변 배경과 자연스럽게 어우러지도록 채워주세요. 단, 이미지의 주요 요소와 전체 분위기는 동일하게 유지해 주세요.",
-        inputs=[],
-        outputs=prompt_input
+    # 기능 적용 버튼 이벤트
+    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):
+        # 이미지 개수 확인
+        images = [image1, image2, image3]
+        valid_images = [img for img in images if img is not None]
+        
+        try:
+            # 자동 프롬프트 생성 또는 프롬프트 전처리
+            auto_prompt = prompt
+            if not prompt or not prompt.strip():
+                if len(valid_images) == 1:
+                    auto_prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+                elif len(valid_images) == 2:
+                    auto_prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
+                else:
+                    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("처리 중 오류 발생:")
+            return None, f"오류 발생: {str(e)}", prompt
+    
     submit_btn.click(
-        fn=generate_multiple_images,
+        fn=process_and_show_prompt,
         inputs=[image1_input, image2_input, image3_input, prompt_input],
-        outputs=[output_image1, output_image2, output_image3, output_image4, output_text, prompt_display],
+        outputs=[output_image, output_text, prompt_display],
     )
-    
+
     gr.Markdown(
         """
         ### 사용 방법:
         
-        1. **자동 합성**: 이미지를 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다.
-        2. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다.
-        3. **선택 옵션**: 위의 버튼을 클릭하면 프롬프트 입력란에 한국어 문구로 입력하시면 됩니다.
-        4. **다양한 이미지**: "이미지 생성" 버튼을 클릭하면 차례로 4장의 이미지가 생성됩니다.
+        1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
+        2. **기능 사용**: 드롭다운에서 원하는 기능을 선택하고 '기능 적용' 버튼을 클릭하세요
+        3. **커스텀 텍스트**: 이미지 변경이나 글자지우기 기능을 선택할 때 추가 설명을 입력할 수 있습니다
+        4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
+        5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
         
-        > **팁**: 프롬프트를 직접 수정할 수도 있습니다.
+        > **팁**: 기능을 선택한 후에도 프롬프트를 직접 수정할 수 있습니다
         """
     )
 
+# 애플리케이션 실행
 if __name__ == "__main__":
     demo.launch(share=True)