Update app.py

app.py

@@ -18,15 +18,52 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# LLM 모델 초기화 (번역 기능용)
+def initialize_gemini():
+    api_key = os.environ.get("GEMINI_API_KEY")
+    if not api_key:
+        logger.error("GEMINI_API_KEY가 설정되지 않았습니다.")
+        return None
+    
+    genai.configure(api_key=api_key)
+    return genai.GenerativeModel(
+        model_name="gemini-2.0-flash",
+        generation_config={
+            "temperature": 0.2,  # 낮은 temperature로 정확한 번역 유도
+            "max_output_tokens": 1024,
+            "top_p": 0.9,
+        }
+    )
+
+# LLM 모델로 한국어 프롬프트를 영어로 번역
+def translate_to_english(korean_prompt, model):
+    if not korean_prompt or not model:
+        return korean_prompt
+    
+    try:
+        translation_prompt = f"""
+        다음 한국어 텍스트를 영어로 정확하게 번역해주세요. 맥락과 의미를 최대한 보존하세요.
+        번역할 텍스트: {korean_prompt}
+        
+        번역만 제공하고 다른 설명은 하지 마세요.
+        """
+        
+        response = model.generate_content(translation_prompt)
+        english_prompt = response.text.strip()
+        logger.info(f"번역 결과: {english_prompt}")
+        return english_prompt
+    except Exception as e:
+        logger.exception(f"번역 중 오류 발생: {str(e)}")
+        return korean_prompt  # 오류 발생 시 원본 한국어 프롬프트 반환
+
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def preprocess_prompt(prompt, image1, image2, image3):
+def preprocess_prompt(prompt, image1=None, image2=None, image3=None):
     """
     프롬프트를 처리하고 기능 명령을 해석
     """
-    # 기존 preprocess_prompt 함수 코드 유지
     # 이미지 없는 참조 확인 및 처리
     has_img1 = image1 is not None
     has_img2 = image2 is not None
@@ -50,22 +87,10 @@ def preprocess_prompt(prompt, image1, image2, image3):
     
     # 기능 명령 해석
     if "1. 이미지 변경" in prompt:
-        # 설명 추출을 시도하지만 실패해도 기본 프롬프트 제공
-        desc_match = re.search(r'#1을 "(.*?)"으로 바꿔라', prompt)
-        if desc_match:
-            description = desc_match.group(1)
-            prompt = f"첫 번째 이미지를 {description}으로 변경해주세요. 원본 이미지의 주요 내용은 유지하되 새로운 스타일과 분위기로 재해석해주세요."
-        else:
-            prompt = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
+        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 = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
+        prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
     
     elif "3. 얼굴바꾸기" in prompt:
         prompt = "첫 번째 이미지의 인물 얼굴을 두 번째 이미지의 얼굴로 자연스럽게 교체해주세요. 얼굴의 표정과 특징은 두 번째 이미지를 따르되, 나머지 부분은 첫 번째 이미지를 유지해주세요."
@@ -99,96 +124,112 @@ def preprocess_prompt(prompt, image1, image2, image3):
     
     return prompt
 
-def generate_with_images(prompt, images):
+def generate_with_images(prompt, images, gemini_model):
     """
     공식 문서에 기반한 올바른 API 호출 방식 구현
+    재시도 로직 추가
     """
-    try:
-        # API 키 확인
-        api_key = os.environ.get("GEMINI_API_KEY")
-        if not api_key:
-            return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
+    max_retries = 2
+    retries = 0
+    
+    while retries <= max_retries:
+        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)
             
-        # Gemini 클라이언트 초기화
-        client = genai.Client(api_key=api_key)
-        
-        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,
-            config=types.GenerateContentConfig(
-                response_modalities=['Text', 'Image'],
-                temperature=1,
-                top_p=0.95,
-                top_k=40,
-                max_output_tokens=8192
-            )
-        )
-        
-        # 임시 파일 생성
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            temp_path = tmp.name
+            # 프롬프트 번역
+            english_prompt = translate_to_english(prompt, gemini_model) if gemini_model else prompt
+            logger.info(f"원본 프롬프트: {prompt}")
+            logger.info(f"번역된 프롬프트: {english_prompt}")
             
-            result_text = ""
-            image_found = False
+            # 컨텐츠 준비
+            contents = []
             
-            # 응답 처리
-            for part in response.candidates[0].content.parts:
-                if hasattr(part, 'text') and part.text:
-                    result_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("응답에서 이미지 추출 성공")
+            # 텍스트 프롬프트 추가
+            contents.append(english_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,
+                config=types.GenerateContentConfig(
+                    response_modalities=['Text', 'Image'],
+                    temperature=1,
+                    top_p=0.95,
+                    top_k=40,
+                    max_output_tokens=8192
+                )
+            )
             
-            if not image_found:
-                return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
+            # 임시 파일 생성
+            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
+                        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("응답에서 이미지 추출 성공")
+                
+                if not image_found:
+                    if retries < max_retries:
+                        retries += 1
+                        logger.warning(f"이미지 생성 실패, 재시도 {retries}/{max_retries}")
+                        continue
+                    else:
+                        return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
+                        
+                # 결과 이미지 반환
+                result_img = Image.open(temp_path)
+                if result_img.mode == "RGBA":
+                    result_img = result_img.convert("RGB")
                     
-            # 결과 이미지 반환
-            result_img = Image.open(temp_path)
-            if result_img.mode == "RGBA":
-                result_img = result_img.convert("RGB")
+                return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
                 
-            return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
-            
-    except Exception as e:
-        logger.exception("이미지 생성 중 오류 발생:")
-        return None, f"오류 발생: {str(e)}"
+        except Exception as e:
+            if retries < max_retries:
+                retries += 1
+                logger.warning(f"이미지 생성 중 오류 발생, 재시도 {retries}/{max_retries}: {str(e)}")
+            else:
+                logger.exception("이미지 생성 중 오류 발생:")
+                return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt):
+def process_images_with_prompt(image1, image2, image3, prompt, gemini_model):
     """
     3개의 이미지와 프롬프트를 처리하는 함수
     """
     try:
         # 이미지 개수 확인
-        images = [image1, image2, image3]
-        valid_images = [img for img in images if img is not None]
-        
-        if not valid_images:
-            return None, "적어도 하나의 이미지를 업로드해주세요."
+        images = [img for img in [image1, image2, image3] if img is not None]
         
         # 프롬프트 처리
         if not prompt or not prompt.strip():
             # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성
-            if len(valid_images) == 1:
+            if len(images) == 0:
+                prompt = "아름다운 자연 풍경 이미지를 생성해주세요. 산, 호수, 하늘이 포함된 평화로운 장면이면 좋겠습니다."
+                logger.info("이미지 없음, 기본 생성 프롬프트 사용")
+            elif len(images) == 1:
                 prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
                 logger.info("단일 이미지 프롬프트 자동 생성")
-            elif len(valid_images) == 2:
+            elif len(images) == 2:
                 prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
                 logger.info("두 이미지 합성 프롬프트 자동 생성")
             else:
@@ -198,18 +239,18 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             # 프롬프트 전처리 및 기능 명령 해석
             prompt = preprocess_prompt(prompt, image1, image2, image3)
         
-        # 새로운 API 호출 방식 사용
-        return generate_with_images(prompt, valid_images)
+        # 이미지 생성 API 호출
+        return generate_with_images(prompt, images, gemini_model)
             
     except Exception as e:
         logger.exception("이미지 처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-# 기능 선택 콜백 (기존 코드 유지)
-def update_prompt_from_function(function_choice, custom_text=""):
+# 기능 선택 콜백 (수정된 버전)
+def update_prompt_from_function(function_choice):
     function_templates = {
-        "1. 이미지 변경": f'#1을 "{custom_text if custom_text else "원하는 설명"}"으로 바꿔라',
-        "2. 글자지우기": f'#1에서 "{custom_text if custom_text else "지울 텍스트"}"를 지워라',
+        "1. 이미지 변경": "#1을 창의적으로 변형해주세요",
+        "2. 글자지우기": "#1에서 텍스트를 지워주세요",
         "3. 얼굴바꾸기": "#1의 인물을 #2의 얼굴로 바꿔라",
         "4. 옷바꾸기": "#1의 인물에 #2 또는 #3의 옷으로 변경하라",
         "5. 배경바꾸기": "#1의 이미지에 #2의 배경으로 자연스럽게 바꿔라",
@@ -219,119 +260,107 @@ def update_prompt_from_function(function_choice, custom_text=""):
     
     return function_templates.get(function_choice, "")
 
-# Gradio 인터페이스 (기존 코드 유지)
-with gr.Blocks() as demo:
-    gr.HTML(
-        """
-        <div style="text-align: center; margin-bottom: 1rem;">
-            <h1>간단한 이미지 생성기</h1>
-            <p>이미지를 업로드하고 바로 실행하면 자동으로 합성합니다. 또는 아래 기능을 선택할 수 있습니다.</p>
-        </div>
-        """
-    )
+# Gradio 인터페이스 (수정된 버전)
+def create_interface():
+    # Gemini 모델 초기화
+    gemini_model = initialize_gemini()
     
-    with gr.Row():
-        with gr.Column():
-            # 3개의 이미지 입력
-            with gr.Row():
-                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="예: 붉은색, 수채화 스타일, 중국어..."
+    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 (선택사항)", 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
+                    )
+                
+                # 프롬프트 입력
+                prompt_input = gr.Textbox(
+                    lines=3,
+                    placeholder="프롬프트를 입력하거나 비워두면 자동 생성됩니다. 한국어로 입력하세요.",
+                    label="프롬프트"
                 )
+                
+                # 생성 버튼
+                submit_btn = gr.Button("이미지 생성", variant="primary")
             
-            apply_function_btn = gr.Button("기능 적용")
-            
-            # 프롬프트 입력 (선택 사항)
-            prompt_input = gr.Textbox(
-                lines=3,
-                placeholder="프롬프트를 입력하거나 비워두면 자동 합성됩니다.",
-                label="프롬프트 (선택 사항)"
-            )
-            
-            # 생성 버튼
-            submit_btn = gr.Button("이미지 생성", variant="primary")
+            with gr.Column():
+                # 결과 출력
+                output_image = gr.Image(label="생성된 이미지")
+                output_text = gr.Textbox(label="상태 메시지")
+                
+                # 사용된 프롬프트 표시
+                prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
         
-        with gr.Column():
-            # 결과 출력
-            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):
-        # 이미지 개수 확인
-        images = [image1, image2, image3]
-        valid_images = [img for img in images if img is not None]
+        # 기능 선택 이벤트 (선택만으로 프롬프트 변경)
+        function_dropdown.change(
+            fn=update_prompt_from_function,
+            inputs=[function_dropdown],
+            outputs=[prompt_input]
+        )
         
-        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)
+        # 이미지 생성 버튼 클릭 이벤트
+        def process_and_show_prompt(image1, image2, image3, prompt):
+            try:
+                # 이미지 생성 함수 호출
+                result_img, status = process_images_with_prompt(image1, image2, image3, prompt, gemini_model)
+                
+                # 프롬프트 전처리
+                processed_prompt = preprocess_prompt(prompt, image1, image2, image3)
+                
+                return result_img, status, processed_prompt
+            except Exception as e:
+                logger.exception("처리 중 오류 발생:")
+                return None, f"오류 발생: {str(e)}", prompt
+        
+        submit_btn.click(
+            fn=process_and_show_prompt,
+            inputs=[image1_input, image2_input, image3_input, prompt_input],
+            outputs=[output_image, output_text, prompt_display],
+        )
+
+        gr.Markdown(
+            """
+            ### 사용 방법:
             
-            # 이미지 생성 함수 호출
-            result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
+            1. **이미지 없이 생성**: 프롬프트만 입력하여 이미지를 생성할 수 있습니다
+            2. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
+            3. **기능 사용**: 드롭다운에서 원하는 기능을 선택하면 자동으로 프롬프트가 변경됩니다
+            4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
+            5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
             
-            return result_img, status, auto_prompt
-        except Exception as e:
-            logger.exception("처리 중 오류 발생:")
-            return None, f"오류 발생: {str(e)}", prompt
+            > **팁**: 한국어로 입력한 프롬프트는 자동으로 영어로 번역되어 처리됩니다
+            """
+        )
     
-    submit_btn.click(
-        fn=process_and_show_prompt,
-        inputs=[image1_input, image2_input, image3_input, prompt_input],
-        outputs=[output_image, output_text, prompt_display],
-    )
-
-    gr.Markdown(
-        """
-        ### 사용 방법:
-        
-        1. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
-        2. **기능 사용**: 드롭다운에서 원하는 기능을 선택하고 '기능 적용' 버튼을 클릭하세요
-        3. **커스텀 텍스트**: 이미지 변경이나 글자지우기 기능을 선택할 때 추가 설명을 입력할 수 있습니다
-        4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
-        5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
-        
-        > **팁**: 기능을 선택한 후에도 프롬프트를 직접 수정할 수 있습니다
-        """
-    )
+    return demo
 
 # 애플리케이션 실행
 if __name__ == "__main__":
+    demo = create_interface()
     demo.launch(share=True)