Update app.py

app.py

@@ -88,6 +88,122 @@ def preprocess_prompt(prompt, image1, image2, image3):
     
     return prompt
 
+def generate_with_images(prompt, images, max_retries=2):
+    """
+    공식 문서에 기반한 올바른 API 호출 방식 구현
+    실패 시 재시도 기능 추가
+    """
+    retries = 0
+    last_error = None
+    
+    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)
+            
+            logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt} (시도: {retries+1}/{max_retries+1})")
+            
+            # 이미지 추가 (이미지가 있는 경우만)
+            image_parts = []
+            for idx, img in enumerate(images, 1):
+                if img is not None:
+                    # 이미지를 바이트로 변환
+                    buffered = BytesIO()
+                    img.save(buffered, format="PNG")
+                    image_bytes = buffered.getvalue()
+                    
+                    # 이미지 파트 생성
+                    image_part = types.Content({
+                        'inline_data': {
+                            'mime_type': 'image/png',
+                            'data': image_bytes
+                        }
+                    })
+                    image_parts.append(image_part)
+                    logger.info(f"이미지 #{idx} 추가됨")
+            
+            # 이미지와 텍스트 컨텐츠 결합
+            contents = image_parts + [types.Content({'text': prompt})]
+            
+            # 생성 설정 - 공식 문서에 따라 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 response or not response.candidates:
+                if retries < max_retries:
+                    retries += 1
+                    logger.warning(f"유효한 응답을 받지 못했습니다. 재시도 중... ({retries}/{max_retries})")
+                    continue
+                return None, "이미지 생성에 실패했습니다. 유효한 응답을 받지 못했습니다."
+            
+            # 임시 파일 생성
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+                temp_path = tmp.name
+                
+                result_text = ""
+                image_found = False
+                
+                # 응답 처리
+                for candidate in response.candidates:
+                    if not candidate.content:
+                        continue
+                    
+                    # 텍스트 추출
+                    if candidate.content.text:
+                        result_text += candidate.content.text
+                        logger.info(f"응답 텍스트: {candidate.content.text}")
+                    
+                    # 이미지 추출
+                    for part in candidate.content.parts:
+                        if hasattr(part, 'inline_data') and part.inline_data:
+                            save_binary_file(temp_path, part.inline_data.data)
+                            image_found = True
+                            logger.info("응답에서 이미지 추출 성공")
+                            break
+                    
+                    if image_found:
+                        break
+                
+                if not image_found:
+                    if retries < max_retries:
+                        retries += 1
+                        logger.warning(f"API에서 이미지를 생성하지 못했습니다. 재시도 중... ({retries}/{max_retries})")
+                        continue
+                    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"이미지가 성공적으로 생성되었습니다. {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.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):
     """
@@ -119,14 +235,14 @@ def process_images_with_prompt(image1, image2, image3, prompt):
             # 프롬프트 전처리 및 기능 명령 해석
             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)}"
 
-# 기능 선택 콜백 (수정됨)
+# 기능 선택 콜백
 def update_prompt_from_function(function_choice):
     function_templates = {
         "1. 이미지 변경": "#1을 창의적으로 바꿔라",
@@ -140,7 +256,7 @@ def update_prompt_from_function(function_choice):
     
     return function_templates.get(function_choice, "")
 
-# Gradio 인터페이스 (수정됨)
+# Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
         """
@@ -159,7 +275,7 @@ with gr.Blocks() as demo:
                 image2_input = gr.Image(type="pil", label="#2", image_mode="RGB")
                 image3_input = gr.Image(type="pil", label="#3", image_mode="RGB")
             
-            # 기능 선택 드롭다운 (커스텀 텍스트 입력 제거)
+            # 기능 선택 드롭다운
             function_dropdown = gr.Dropdown(
                 choices=[
                     "1. 이미지 변경",