Update app.py

app.py

@@ -98,7 +98,7 @@ def preprocess_prompt(prompt, image1, image2, image3):
 
 def generate_with_images(prompt, images):
     """
-    문제가 된 API 호출 부분을 새로운 방식으로 구현
+    새로운 API 호출 방식으로 구현 - 응답 모달리티 수정
     """
     try:
         # API 키 확인
@@ -151,10 +151,12 @@ def generate_with_images(prompt, images):
             ),
         ]
         
-        # 생성 설정
+        # 생성 설정 - response_modalities 제거
         generate_content_config = types.GenerateContentConfig(
             temperature=1,
-            response_modalities=["image"],
+            top_p=0.95,
+            top_k=40,
+            max_output_tokens=8192,
         )
         
         # 임시 파일 생성
@@ -162,48 +164,85 @@ def generate_with_images(prompt, images):
             temp_path = tmp.name
             
             # Gemini 모델로 요청 전송
-            response_stream = client.models.generate_content_stream(
-                model="gemini-2.0-flash-exp-image-generation",
+            response = client.models.generate_content(
+                model="gemini-pro-vision",  # 모델을 gemini-pro-vision으로 변경
                 contents=contents,
                 config=generate_content_config,
             )
             
-            # 응답에서 이미지 추출
-            image_found = False
-            for chunk in response_stream:
-                if not chunk.candidates or not chunk.candidates[0].content or not chunk.candidates[0].content.parts:
-                    logger.warning("청크에 데이터가 없음. 건너뜀.")
-                    continue
-                
-                for part in chunk.candidates[0].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("응답에서 이미지 추출 성공")
-            
-            # 임시 파일 정리
-            for path in temp_image_paths:
-                try:
-                    os.unlink(path)
-                except:
-                    pass
+            # 응답이 이미지 대신 텍스트일 것으로 예상
+            response_text = ""
+            for part in response.candidates[0].content.parts:
+                if hasattr(part, 'text') and part.text:
+                    response_text += part.text
             
-            # 파일 URI 정리
-            for file in file_uris:
-                try:
-                    client.files.delete(fileLocation=file.uri)
-                except:
-                    pass
+            logger.info(f"응답 텍스트: {response_text}")
             
-            if not image_found:
-                return None, "API에서 이미지를 생성하지 못했습니다. 다른 프롬프트로 시도해보세요."
+            # 이제 이 텍스트 응답을 사용하여 gemini-2.0-flash-exp-image-generation 모델로 이미지 생성
+            # 첫 번째 이미지만 사용 (여러 이미지 사용은 복잡해질 수 있음)
+            if len(temp_image_paths) > 0:
+                with open(temp_image_paths[0], 'rb') as f:
+                    image_data = f.read()
+                
+                # gemini-2.0-flash-exp-image-generation 모델을 사용한 단일 이미지 요청
+                response_stream = client.models.generate_content_stream(
+                    model="gemini-2.0-flash-exp-image-generation",
+                    contents=[
+                        types.Content(
+                            role="user",
+                            parts=[
+                                types.Part.from_data(
+                                    mime_type="image/png",
+                                    data=image_data
+                                ),
+                                types.Part.from_text(text=f"{prompt} {response_text}")
+                            ]
+                        )
+                    ],
+                    config=types.GenerateContentConfig(
+                        temperature=1,
+                        response_mime_type="image/png",
+                    )
+                )
+                
+                # 응답에서 이미지 추출
+                image_found = False
+                for chunk in response_stream:
+                    if not chunk.candidates or not chunk.candidates[0].content or not chunk.candidates[0].content.parts:
+                        logger.warning("청크에 데이터가 없음. 건너뜀.")
+                        continue
                     
-            # 결과 이미지 반환
-            result_img = Image.open(temp_path)
-            if result_img.mode == "RGBA":
-                result_img = result_img.convert("RGB")
+                    for part in chunk.candidates[0].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("응답에서 이미지 추출 성공")
+                
+                # 임시 파일 정리
+                for path in temp_image_paths:
+                    try:
+                        os.unlink(path)
+                    except:
+                        pass
+                
+                # 파일 URI 정리
+                for file in file_uris:
+                    try:
+                        client.files.delete(fileLocation=file.uri)
+                    except:
+                        pass
                 
-            return result_img, "이미지가 성공적으로 생성되었습니다."
+                if not image_found:
+                    return None, "API에서 이미지를 생성하지 못했습니다. 다른 프롬프트로 시도해보세요."
+                        
+                # 결과 이미지 반환
+                result_img = Image.open(temp_path)
+                if result_img.mode == "RGBA":
+                    result_img = result_img.convert("RGB")
+                    
+                return result_img, "이미지가 성공적으로 생성되었습니다."
+            else:
+                return None, "처리할 이미지가 없습니다."
             
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")