Update app.py

app.py

@@ -5,7 +5,7 @@ import tempfile
 from PIL import Image
 import gradio as gr
 import logging
-from io import BytesIO
+import io
 
 from google import genai
 from google.genai import types
@@ -27,6 +27,12 @@ def save_binary_file(file_name, data):
     logger.debug(f"파일 저장 완료: {file_name}")
 
 
+def image_to_bytes(img_path):
+    """파일 경로에서 이미지 바이트 데이터 읽기"""
+    with open(img_path, "rb") as f:
+        return f.read()
+
+
 def merge_images(person_img_path, product_img_path, background_img_path, prompt, model="gemini-2.0-flash-exp-image-generation"):
     logger.debug(f"merge_images 함수 시작 - 프롬프트: '{prompt}'")
     
@@ -42,37 +48,50 @@ def merge_images(person_img_path, product_img_path, background_img_path, prompt,
         client = genai.Client(api_key=effective_api_key)
         logger.debug("Gemini 클라이언트 초기화 완료.")
 
-        # PIL 이미지 객체로 변환
-        person_img = Image.open(person_img_path)
-        product_img = Image.open(product_img_path)
+        # 이미지 파일 업로드
+        person_file = client.files.upload(file=person_img_path)
+        logger.debug(f"사람 이미지 업로드 완료: {person_file.uri}")
+        
+        product_file = client.files.upload(file=product_img_path)
+        logger.debug(f"상품 이미지 업로드 완료: {product_file.uri}")
         
         # 컨텐츠 리스트 생성
-        contents = []
+        parts = []
         
         # 사람 이미지 추가
-        contents.append(
-            types.Part.from_data(data=person_img, mime_type="image/jpeg")
+        parts.append(
+            types.Part.from_uri(
+                file_uri=person_file.uri,
+                mime_type=person_file.mime_type
+            )
         )
         
         # 상품 이미지 추가
-        contents.append(
-            types.Part.from_data(data=product_img, mime_type="image/jpeg")
+        parts.append(
+            types.Part.from_uri(
+                file_uri=product_file.uri,
+                mime_type=product_file.mime_type
+            )
         )
         
         # 배경 이미지가 있으면 추가
         if background_img_path:
-            background_img = Image.open(background_img_path)
-            contents.append(
-                types.Part.from_data(data=background_img, mime_type="image/jpeg")
+            background_file = client.files.upload(file=background_img_path)
+            logger.debug(f"배경 이미지 업로드 완료: {background_file.uri}")
+            parts.append(
+                types.Part.from_uri(
+                    file_uri=background_file.uri,
+                    mime_type=background_file.mime_type
+                )
             )
             logger.debug("배경 이미지 추가됨")
         
         # 마지막에 프롬프트 추가
-        contents.append(
+        parts.append(
             types.Part.from_text(text=prompt)
         )
         
-        logger.debug(f"컨텐츠 객체 생성 완료: {len(contents)} 아이템")
+        logger.debug(f"컨텐츠 객체 생성 완료: {len(parts)} 아이템")
 
         # 생성 설정
         generate_content_config = types.GenerateContentConfig(
@@ -88,13 +107,13 @@ def merge_images(person_img_path, product_img_path, background_img_path, prompt,
             temp_path = tmp.name
             logger.debug(f"임시 파일 생성됨: {temp_path}")
 
-            # 스트리밍 방식으로 데이터 수신 (원래 코드 방식으로 복원)
+            # 스트리밍 방식으로 데이터 수신
             response_stream = client.models.generate_content_stream(
                 model=model,
                 contents=[
                     types.Content(
                         role="user",
-                        parts=contents,
+                        parts=parts,
                     ),
                 ],
                 config=generate_content_config,
@@ -110,11 +129,11 @@ def merge_images(person_img_path, product_img_path, background_img_path, prompt,
                 logger.debug(f"chunk 수신: {chunk}")
                 
                 # 응답 검증
-                if not hasattr(chunk, 'candidates') or not chunk.candidates:
+                if not hasattr(chunk, 'candidates') or not chunk.candidates or len(chunk.candidates) == 0:
                     logger.warning("chunk에 candidates가 없습니다. 건너뜁니다.")
                     continue
                     
-                if not hasattr(chunk.candidates[0], 'content') or not chunk.candidates[0].content:
+                if not hasattr(chunk.candidates[0], 'content') or chunk.candidates[0].content is None:
                     logger.warning("chunk.candidates[0]에 content가 없습니다. 건너뜁니다.")
                     continue