Update app.py

app.py

@@ -6,9 +6,7 @@ import logging
 import re
 import time
 import json
-
-from google import genai
-from google.genai import types
+import google.generativeai as genai
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -19,7 +17,6 @@ logger = logging.getLogger(__name__)
 
 # Gemini API 키 설정
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
-genai.configure(api_key=GEMINI_API_KEY)
 
 # 배경 JSON 파일 경로 설정
 BACKGROUNDS_DIR = "./background"
@@ -29,87 +26,37 @@ if not os.path.exists(BACKGROUNDS_DIR):
     os.makedirs(BACKGROUNDS_DIR)
     logger.info(f"배경 디렉토리를 생성했습니다: {BACKGROUNDS_DIR}")
 
-# JSON 파일 로드 함수
-def load_background_json(filename):
-    file_path = os.path.join(BACKGROUNDS_DIR, filename)
-    try:
-        with open(file_path, 'r', encoding='utf-8') as f:
-            data = json.load(f)
-            logger.info(f"{filename} 파일을 성공적으로 로드했습니다. {len(data)} 항목 포함.")
-            return data
-    except FileNotFoundError:
-        logger.warning(f"경고: {filename} 파일을 찾을 수 없습니다. 기본값을 사용합니다.")
-        return {}
-    except json.JSONDecodeError:
-        logger.warning(f"경고: {filename} 파일의 JSON 형식이 올바르지 않습니다. 기본값을 사용합니다.")
-        return {}
-    except Exception as e:
-        logger.warning(f"경고: {filename} 파일 로드 중 오류 발생: {str(e)}. 기본값을 사용합니다.")
-        return {}
+# 기본 배경 생성
+SIMPLE_BACKGROUNDS = {"화이트 배경": "white background", "블랙 배경": "black background", "그레이 배경": "gray background"}
+STUDIO_BACKGROUNDS = {"제품 사진 스튜디오": "product photography studio", "미니멀 스튜디오": "minimal studio setup"}
+NATURE_BACKGROUNDS = {"열대 해변": "tropical beach", "숲속": "lush forest", "산악 풍경": "mountain landscape"}
+INDOOR_BACKGROUNDS = {"모던 리빙룸": "modern living room", "럭셔리 쇼룸": "luxury showroom", "카페": "cozy cafe"}
+ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights", "그라데이션": "gradient background", "기하학적 패턴": "geometric pattern"}
 
-# 배경 데이터 로드
-SIMPLE_BACKGROUNDS = load_background_json("simple_backgrounds.json")
-STUDIO_BACKGROUNDS = load_background_json("studio_backgrounds.json")
-NATURE_BACKGROUNDS = load_background_json("nature_backgrounds.json")
-INDOOR_BACKGROUNDS = load_background_json("indoor_backgrounds.json")
-ABSTRACT_BACKGROUNDS = load_background_json("abstract_backgrounds.json")
-
-# 배경이 로드되지 않은 경우 기본값 설정
-if not SIMPLE_BACKGROUNDS:
-    SIMPLE_BACKGROUNDS = {"화이트 배경": "white background"}
-if not STUDIO_BACKGROUNDS:
-    STUDIO_BACKGROUNDS = {"제품 사진 스튜디오": "product photography studio"}
-if not NATURE_BACKGROUNDS:
-    NATURE_BACKGROUNDS = {"열대 해변": "tropical beach"}
-if not INDOOR_BACKGROUNDS:
-    INDOOR_BACKGROUNDS = {"모던 리빙룸": "modern living room"}
-if not ABSTRACT_BACKGROUNDS:
-    ABSTRACT_BACKGROUNDS = {"네온 조명": "neon lights"}
+def save_binary_file(file_name, data):
+    with open(file_name, "wb") as f:
+        f.write(data)
 
 def generate_system_instruction():
     return """당신은 상품 이미지의 배경을 변경하기 위한 고품질 프롬프트를 생성하는 전문가입니다.
-사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 미드저니(Midjourney)에 사용할 수 있는 
-상세하고 전문적인 프롬프트를 영어로 생성해주세요.
+사용자가 제공하는 상품명, 배경 유형, 추가 요청사항을 바탕으로 고품질 프롬프트를 영어로 생성해주세요.
 다음 가이드라인을 반드시 따라야 합니다:
 1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
 2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
-3. *** 상품의 본질적 특성은 유지하되, 자연스러운 환경 통합을 위한 조명과 그림자는 허용합니다: ***
-   - 상품 자체의 색상, 디자인, 형태, 텍스처는 절대 수정하지 않습니다.
-   - 환경과 자연스럽게 어울리는 그림자, 주변 조명 효과는 허용됩니다.
-   - 상품에 물방울, 응축, 금, 은과 같은 추가 요소나 물리적 효과는 적용하지 않습니다.
-   - 환경에 어울리는 자연스러운 빛 반사, 주변 조명, 그림자는 사실적 통합감을 위해 적용할 수 있습니다.
-4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다. 프롬프트에 "square format", "1:1 ratio" 또는 "aspect ratio 1:1"을 명시적으로 포함합니다.
+3. *** 상품의 본질적 특성은 유지하되, 자연스러운 환경 통합을 위한 조명과 그림자는 허용합니다 ***
+4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다.
 5. 상품은 반드시 정사각형 구도의 정중앙에 배치되어야 합니다.
 6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 크게 차지하도록 합니다.
 7. 상품 이미지 컷아웃(#1)의 기본 형태와 색상은 유지하면서, 선택한 환경에 자연스럽게 통합되도록 합니다.
-8. 고급스러운 상업적 이미지를 위한 다음 환경 요소들을 포함하세요:
-   - 상품과 어울리는 주변 환경/배경 요소를 추가합니다. 예를 들어, 화장품 주변에 꽃이나 허브, 음료 제품 옆에 과일, 전자제품 근처에 현대적 소품 등.
-   - 환경의 조명 효과(림 라이트, 백라이트, 소프트박스 등)를 설명합니다.
-   - 상품이 환경에 자연스럽게 존재하는 것처럼 보이도록 적절한 그림자와 빛 표현을 포함합니다.
-   - 상품의 용도나 장점을 간접적으로 암시하는 배경 요소를 포함합니다.
-   - 프로페셔널한 상업 사진 효과(선택적 피사계 심도, 소프트 포커스, 스튜디오 조명 등)를 명시합니다.
+8. 고급스러운 상업적 이미지를 위한 환경 요소들을 포함하세요.
 9. 프롬프트에 다음 요소들을 명시적으로 포함하세요:
    - "highly detailed commercial photography"
    - "award-winning product photography"
    - "professional advertising imagery"
-   - "studio quality"
-   - "magazine advertisement quality"
-10. 배경 환경 요소를 상품 카테고리에 맞게 선택합니다:
-    - 스킨케어 제품: 깨끗한 욕실 선반, 우아한 화장대, 스파 같은 환경 등
-    - 음료 제품: 세련된 테이블, 파티 환경, 야외 피크닉 장면 등
-    - 전자 제품: 세련된 작업 공간, 현대적인 거실, 미니멀한 책상 등
-    - 패션/의류: 세련된 쇼룸, 도시 거리, 엘레강스한 라이프스타일 환경 등
-    - 식품 제품: 깔끔한 주방, 식탁, 요리 환경 등
-11. 사용자가 제공한 구체적인 배경과 추가 요청사항을 정확히 반영합니다.
-12. 프롬프트는 미드저니 AI에 최적화되어야 합니다.
-13. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
-출력 형식은 영어로 된 단일 단락의 상세한 프롬프트여야 하며, 끝에 미드저니 파라미터가 포함되어야 합니다.
+10. 사용자가 제공한 구체적인 배경과 추가 요청사항을 정확히 반영합니다.
+출력 형식은 영어로 된 단일 단락의 상세한 프롬프트여야 합니다.
 """
 
-def save_binary_file(file_name, data):
-    with open(file_name, "wb") as f:
-        f.write(data)
-
 def generate_prompt_with_gemini(product_name, background_info, additional_info=""):
     if not GEMINI_API_KEY:
         return "Gemini API 키가 설정되지 않았습니다. 환경 변수 GEMINI_API_KEY를 설정하거나 코드에 직접 입력하세요."
@@ -128,25 +75,14 @@ def generate_prompt_with_gemini(product_name, background_info, additional_info="
 5. 환경과의 자연스러운 통합을 위한 조명 효과와 그림자는 포함해주세요.
 6. 상품을 더 돋보이게 하는 배경 환경을 설명해주세요.
 7. 고급스러운 상업 광고 품질의 이미지가 되도록 환경 설명을 해주세요.
-8. 프롬프트 끝에 미드저니 파라미터 "--ar 1:1 --s 750 --q 2"를 추가해주세요.
 한국어 입력 내용을 영어로 적절히 번역하여 반영해주세요.
 """
-        model = genai.GenerativeModel(
-            'gemini-2.0-flash',
-            system_instruction=generate_system_instruction()
-        )
-        response = model.generate_content(
-            prompt_request,
-            generation_config=genai.types.GenerationConfig(
-                temperature=0.7,
-                top_p=0.95,
-                top_k=64,
-                max_output_tokens=1024,
-            )
-        )
+        genai.configure(api_key=GEMINI_API_KEY)
+        
+        model = genai.GenerativeModel('gemini-2.0-flash')
+        
+        response = model.generate_content(prompt_request)
         response_text = response.text.strip()
-        if "--ar 1:1" not in response_text:
-            response_text = response_text.rstrip(".") + ". --ar 1:1 --s 750 --q 2"
         
         # 미드저니 파라미터 제거하고 반환 (이미지 생성에서는 필요없음)
         cleaned_response = re.sub(r'--ar \d+:\d+|--s \d+|--q \d+', '', response_text).strip()
@@ -198,7 +134,7 @@ def generate_with_images(prompt, images, variation_index=0):
         if not api_key:
             return None, "API 키가 설정되지 않았습니다. 환경변수를 확인해주세요."
 
-        client = genai.Client(api_key=api_key)
+        genai.configure(api_key=api_key)
         logger.info(f"Gemini API 요청 시작 - 프롬프트: {prompt}, 변형 인덱스: {variation_index}")
 
         variation_suffixes = [
@@ -213,37 +149,44 @@ def generate_with_images(prompt, images, variation_index=0):
         else:
             prompt = prompt + " Do not add any text, watermarks, or labels to the image."
 
-        contents = [prompt]
-        for idx, img in enumerate(images, 1):
+        model = genai.GenerativeModel('gemini-1.5-flash')
+        
+        parts = [prompt]
+        for img in images:
             if img is not None:
-                contents.append(img)
-                logger.info(f"이미지 #{idx} 추가됨")
+                if isinstance(img, str):  # If it's a file path
+                    with open(img, "rb") as f:
+                        img_data = f.read()
+                    parts.append({"mime_type": "image/jpeg", "data": img_data})
+                else:  # If it's a PIL Image
+                    img_bytes = tempfile.NamedTemporaryFile(suffix=".jpg", delete=False)
+                    img.save(img_bytes, format="JPEG")
+                    img_bytes.close()
+                    
+                    with open(img_bytes.name, "rb") as f:
+                        img_data = f.read()
+                    
+                    parts.append({"mime_type": "image/jpeg", "data": img_data})
+                    os.unlink(img_bytes.name)  # Clean up temp file
 
-        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
-            )
-        )
+        response = model.generate_content(parts)
 
         # 임시 파일은 항상 JPG 확장자로 생성
         with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
             temp_path = tmp.name
             result_text = ""
             image_found = False
-            for part in response.candidates[0].content.parts:
+            
+            for part in response.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)
+                    img_data = part.inline_data.data
+                    save_binary_file(temp_path, img_data)
                     image_found = True
                     logger.info("응답에서 이미지 추출 성공")
+            
             if not image_found:
                 return None, f"API에서 이미지를 생성하지 못했습니다. 응답 텍스트: {result_text}"