Update app.py

app.py

@@ -7,8 +7,8 @@ import logging
 import re
 import time
 
-from google import genai
-from google.genai import types
+# 패키지 임포트 방식을 변경
+import google.generativeai as genai
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -19,10 +19,10 @@ logger = logging.getLogger(__name__)
 
 # Gemini API 키 설정
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
-
-# API 키 설정 - 새로운 방식으로 수정
-# genai.configure(api_key=GEMINI_API_KEY) 대신 클라이언트 객체를 생성
-genai_client = genai.Client(api_key=GEMINI_API_KEY) if GEMINI_API_KEY else None
+if GEMINI_API_KEY:
+    genai.configure(api_key=GEMINI_API_KEY)
+else:
+    logger.warning("GEMINI_API_KEY가 설정되지 않았습니다.")
 
 # 배경 JSON 파일 경로 설정 - 상대 경로 사용
 BACKGROUNDS_DIR = "./background"
@@ -110,8 +110,6 @@ def translate_prompt_to_english(prompt):
             prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
             return prompt
             
-        # 기존: client = genai.Client(api_key=GEMINI_API_KEY)
-        client = genai_client  # 전역 변수 사용
         translation_prompt = f"""
         Translate the following Korean text to English:
         
@@ -122,24 +120,20 @@ def translate_prompt_to_english(prompt):
         """
         
         logger.info(f"Translation prompt: {translation_prompt}")
-        response = client.models.generate_content(
+        
+        # 새로운 API 사용 방식
+        response = genai.generate_text(
             model="gemini-2.0-flash",
-            contents=[translation_prompt],
-            config=types.GenerateContentConfig(
-                response_modalities=['Text'],
-                temperature=0.2,
-                top_p=0.95,
-                top_k=40,
-                max_output_tokens=512
-            )
+            prompt=translation_prompt,
+            temperature=0.2,
+            top_p=0.95,
+            top_k=40,
+            max_output_tokens=512
         )
         
-        translated_text = ""
-        for part in response.candidates[0].content.parts:
-            if hasattr(part, 'text') and part.text:
-                translated_text += part.text
+        translated_text = response.result
         
-        if translated_text.strip():
+        if translated_text and translated_text.strip():
             translated_text = translated_text.replace("IMAGE_TAG_ONE", "#1")
             logger.info(f"Translated text: {translated_text.strip()}")
             return translated_text.strip()
@@ -363,20 +357,27 @@ def generate_prompt_with_gemini(product_name, background_info, additional_info="
 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,
-            )
+        
+        # google-generativeai 패키지 사용 방식으로 변경
+        system_instruction = generate_system_instruction()
+        
+        # GenerativeModel 대신 genai.chat 사용
+        chat = genai.chat(
+            model="gemini-2.0-flash",
+            messages=[
+                {"role": "system", "content": system_instruction},
+                {"role": "user", "content": prompt_request}
+            ],
+            generation_config={
+                "temperature": 0.7,
+                "top_p": 0.95,
+                "top_k": 64,
+                "max_output_tokens": 1024,
+            }
         )
-        response_text = response.text.strip()
+        
+        response_text = chat.last.text.strip()
+        
         if "--ar 1:1" not in response_text:
             response_text = response_text.rstrip(".") + ". --ar 1:1 --s 750 --q 2"