Update app.py

app.py

@@ -18,58 +18,11 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# LLM 설정 및 번역 함수
-def get_translation(korean_text):
-    """
-    한국어 텍스트를 영어로 번역
-    """
-    try:
-        api_key = os.environ.get("GEMINI_API_KEY")
-        if not api_key:
-            logger.error("GEMINI_API_KEY가 설정되지 않았습니다.")
-            return korean_text
-        
-        # 클라이언트 초기화
-        client = genai.GenerativeModel(
-            model_name="gemini-2.0-flash",
-            generation_config={
-                "temperature": 0.2,
-                "max_output_tokens": 1024,
-                "top_p": 0.9,
-            },
-            system_instruction="You are a professional translator who translates Korean to English accurately.",
-            api_key=api_key
-        )
-        
-        # 번역 프롬프트
-        translation_prompt = f"""
-        Translate the following Korean text to English accurately:
-        
-        {korean_text}
-        
-        Provide only the translation, no explanations.
-        """
-        
-        # 번역 요청
-        response = client.generate_content(translation_prompt)
-        
-        # 응답에서 텍스트 추출
-        if hasattr(response, 'text'):
-            english_text = response.text.strip()
-            logger.info(f"번역 결과: {english_text}")
-            return english_text
-        else:
-            logger.warning("번역 응답에 text 속성이 없습니다.")
-            return korean_text
-    except Exception as e:
-        logger.exception(f"번역 중 오류 발생: {str(e)}")
-        return korean_text  # 오류 발생 시 원본 한국어 프롬프트 반환
-
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
 
-def preprocess_prompt(prompt, image1=None, image2=None, image3=None):
+def preprocess_prompt(prompt, image1, image2, image3):
     """
     프롬프트를 처리하고 기능 명령을 해석
     """
@@ -96,9 +49,11 @@ def preprocess_prompt(prompt, image1=None, image2=None, image3=None):
     
     # 기능 명령 해석
     if "1. 이미지 변경" in prompt:
+        # 기본 프롬프트 제공
         prompt = "첫 번째 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
     
     elif "2. 글자지우기" in prompt:
+        # 기본 프롬프트 제공
         prompt = "첫 번째 이미지에서 모든 텍스트를 찾아 자연스럽게 제거해주세요. 깔끔한 이미지로 만들어주세요."
     
     elif "3. 얼굴바꾸기" in prompt:
@@ -133,13 +88,13 @@ def preprocess_prompt(prompt, image1=None, image2=None, image3=None):
     
     return prompt
 
-def generate_with_images(prompt, images):
+def generate_with_images(prompt, images, max_retries=2):
     """
     공식 문서에 기반한 올바른 API 호출 방식 구현
-    재시도 로직 추가
+    실패 시 재시도 기능 추가
     """
-    max_retries = 2
     retries = 0
+    last_error = None
     
     while retries <= max_retries:
         try:
@@ -151,18 +106,15 @@ def generate_with_images(prompt, images):
             # Gemini 클라이언트 초기화
             client = genai.Client(api_key=api_key)
             
-            # 프롬프트 번역
-            english_prompt = get_translation(prompt)
-            logger.info(f"원본 프롬프트: {prompt}")
-            logger.info(f"번역된 프롬프트: {english_prompt}")
+            logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt} (시도: {retries+1}/{max_retries+1})")
             
             # 컨텐츠 준비
             contents = []
             
             # 텍스트 프롬프트 추가
-            contents.append(english_prompt)
+            contents.append(prompt)
             
-            # 이미지 추가
+            # 이미지 추가 (이미지가 있는 경우만)
             for idx, img in enumerate(images, 1):
                 if img is not None:
                     contents.append(img)
@@ -201,10 +153,9 @@ def generate_with_images(prompt, images):
                 if not image_found:
                     if retries < max_retries:
                         retries += 1
-                        logger.warning(f"이미지 생성 실패, 재시도 {retries}/{max_retries}")
+                        logger.warning(f"API에서 이미지를 생성하지 못했습니다. 재시도 중... ({retries}/{max_retries})")
                         continue
-                    else:
-                        return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
+                    return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"
                         
                 # 결과 이미지 반환
                 result_img = Image.open(temp_path)
@@ -214,31 +165,37 @@ def generate_with_images(prompt, images):
                 return result_img, f"이미지가 성공적으로 생성되었습니다. {result_text}"
                 
         except Exception as e:
+            last_error = str(e)
+            logger.exception(f"이미지 생성 중 오류 발생 (시도: {retries+1}/{max_retries+1}):")
             if retries < max_retries:
                 retries += 1
-                logger.warning(f"이미지 생성 중 오류 발생, 재시도 {retries}/{max_retries}: {str(e)}")
-            else:
-                logger.exception("이미지 생성 중 오류 발생:")
-                return None, f"오류 발생: {str(e)}"
+                logger.info(f"재시도 중... ({retries}/{max_retries})")
+                continue
+            return None, f"오류 발생: {last_error}"
+    
+    return None, f"최대 재시도 횟수 초과. 마지막 오류: {last_error}"
 
 def process_images_with_prompt(image1, image2, image3, prompt):
     """
     3개의 이미지와 프롬프트를 처리하는 함수
+    이미지 없이 프롬프트만으로도 동작
     """
     try:
         # 이미지 개수 확인
-        images = [img for img in [image1, image2, image3] if img is not None]
+        images = [image1, image2, image3]
+        valid_images = [img for img in images if img is not None]
         
         # 프롬프트 처리
         if not prompt or not prompt.strip():
-            # 프롬프트가 없으면 업로드된 이미지 수에 따라 자동 합성 프롬프트 생성
-            if len(images) == 0:
-                prompt = "아름다운 자연 풍경 이미지를 생성해주세요. 산, 호수, 하늘이 포함된 평화로운 장면이면 좋겠습니다."
-                logger.info("이미지 없음, 기본 생성 프롬프트 사용")
-            elif len(images) == 1:
+            # 프롬프트가 없는 경우
+            if not valid_images:
+                # 이미지도 없고 프롬프트도 없는 경우 기본 프롬프트 사용
+                prompt = "아름다운 풍경 이미지를 생성해주세요. 푸른 하늘, 초록색 언덕, 그리고 맑은 호수가 있는 평화로운 자연 풍경을 만들어주세요."
+                logger.info("이미지와 프롬프트 모두 없음. 기본 프롬프트 생성")
+            elif len(valid_images) == 1:
                 prompt = "이 이미지를 창의적으로 변형해주세요. 더 생생하고 예술적인 버전으로 만들어주세요."
                 logger.info("단일 이미지 프롬프트 자동 생성")
-            elif len(images) == 2:
+            elif len(valid_images) == 2:
                 prompt = "이 두 이미지를 자연스럽게 합성해주세요. 두 이미지의 요소를 조화롭게 통합하여 하나의 이미지로 만들어주세요."
                 logger.info("두 이미지 합성 프롬프트 자동 생성")
             else:
@@ -248,18 +205,18 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             # 프롬프트 전처리 및 기능 명령 해석
             prompt = preprocess_prompt(prompt, image1, image2, image3)
         
-        # 이미지 생성 API 호출
-        return generate_with_images(prompt, images)
+        # 새로운 API 호출 방식 사용 (재시도 기능 포함)
+        return generate_with_images(prompt, valid_images)
             
     except Exception as e:
         logger.exception("이미지 처리 중 오류 발생:")
         return None, f"오류 발생: {str(e)}"
 
-# 기능 선택 콜백 (수정된 버전)
+# 기능 선택 콜백 (수정됨)
 def update_prompt_from_function(function_choice):
     function_templates = {
-        "1. 이미지 변경": "#1을 창의적으로 변형해주세요",
-        "2. 글자지우기": "#1에서 텍스트를 지워주세요",
+        "1. 이미지 변경": "#1을 창의적으로 바꿔라",
+        "2. 글자지우기": "#1에서 텍스트를 지워라",
         "3. 얼굴바꾸기": "#1의 인물을 #2의 얼굴로 바꿔라",
         "4. 옷바꾸기": "#1의 인물에 #2 또는 #3의 옷으로 변경하라",
         "5. 배경바꾸기": "#1의 이미지에 #2의 배경으로 자연스럽게 바꿔라",
@@ -269,104 +226,116 @@ def update_prompt_from_function(function_choice):
     
     return function_templates.get(function_choice, "")
 
-# Gradio 인터페이스 (수정된 버전)
-def create_interface():
-    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")
+# 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", 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.Column():
-                # 결과 출력
-                output_image = gr.Image(label="생성된 이미지")
-                output_text = gr.Textbox(label="상태 메시지")
-                
-                # 사용된 프롬프트 표시
-                prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
-        
-        # 기능 선택 이벤트 (선택만으로 프롬프트 변경)
-        function_dropdown.change(
-            fn=update_prompt_from_function,
-            inputs=[function_dropdown],
-            outputs=[prompt_input]
-        )
+            # 기능 선택 드롭다운 (커스텀 텍스트 입력 제거)
+            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")
         
-        # 이미지 생성 버튼 클릭 이벤트
-        def process_and_show_prompt(image1, image2, image3, prompt):
-            try:
-                # 이미지 생성 함수 호출
-                result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
-                
-                # 프롬프트 전처리
-                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
+        with gr.Column():
+            # 결과 출력
+            output_image = gr.Image(label="생성된 이미지")
+            output_text = gr.Textbox(label="상태 메시지")
+            
+            # 사용된 프롬프트 표시
+            prompt_display = gr.Textbox(label="사용된 프롬프트", visible=True)
+    
+    # 기능 선택 드롭다운 이벤트 (즉시 프롬프트 업데이트)
+    function_dropdown.change(
+        fn=update_prompt_from_function,
+        inputs=[function_dropdown],
+        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]
         
-        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(
-            """
-            ### 사용 방법:
+        try:
+            # 자동 프롬프트 생성 또는 프롬프트 전처리
+            auto_prompt = prompt
+            if not prompt or not prompt.strip():
+                if not valid_images:
+                    # 이미지도 없고 프롬프트도 없는 경우
+                    auto_prompt = "아름다운 풍경 이미지를 생성해주세요. 푸른 하늘, 초록색 언덕, 그리고 맑은 호수가 있는 평화로운 자연 풍경을 만들어주세요."
+                elif 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)
             
-            1. **이미지 없이 생성**: 프롬프트만 입력하여 이미지를 생성할 수 있습니다
-            2. **자동 합성**: 이미지만 업로드하고 프롬프트를 비워두면 자동으로 합성됩니다
-            3. **기능 사용**: 드롭다운에서 원하는 기능을 선택하면 자동으로 프롬프트가 변경됩니다
-            4. **이미지 참조**: #1, #2, #3으로 각 이미지를 참조할 수 있습니다
-            5. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
+            # 이미지 생성 함수 호출
+            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
     
-    return demo
+    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. **일부 이미지만**: 필요한 이미지만 업로드해도 기능 실행이 가능합니다
+        6. **재시도 기능**: 이미지 생성 중 오류 발생 시 자동으로 재시도합니다
+        
+        > **팁**: 기능을 선택한 후에도 프롬프트를 직접 수정할 수 있습니다
+        """
+    )
 
 # 애플리케이션 실행
 if __name__ == "__main__":
-    demo = create_interface()
     demo.launch(share=True)