Update app.py

app.py

@@ -18,53 +18,52 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-# LLM 모델 초기화 (번역 기능용)
-def initialize_gemini():
-    api_key = os.environ.get("GEMINI_API_KEY")
-    if not api_key:
-        logger.error("GEMINI_API_KEY가 설정되지 않았습니다.")
-        return None
-    
-    # genai 클라이언트 생성
-    client = genai.Client(api_key=api_key)
-    
-    # GenerativeModel 인스턴스 생성
-    model = client.models.get_model("gemini-2.0-flash")
-    
-    return model
-
-# LLM 모델로 한국어 프롬프트를 영어로 번역
-def translate_to_english(korean_prompt, model):
-    if not korean_prompt or not model:
-        return korean_prompt
-    
+# LLM 설정 및 번역 함수
+def get_translation(korean_text):
+    """
+    한국어 텍스트를 영어로 번역
+    """
     try:
+        api_key = os.environ.get("GEMINI_API_KEY")
+        if not api_key:
+            logger.error("GEMINI_API_KEY가 설정되지 않았습니다.")
+            return korean_text
+        
+        # 클라이언트 초기화
+        client = genai.GenerativeModel(
+            model_name="gemini-2.0-flash",
+            generation_config={
+                "temperature": 0.2,
+                "max_output_tokens": 1024,
+                "top_p": 0.9,
+            },
+            system_instruction="You are a professional translator who translates Korean to English accurately.",
+            api_key=api_key
+        )
+        
+        # 번역 프롬프트
         translation_prompt = f"""
-        다음 한국어 텍스트를 영어로 정확하게 번역해주세요. 맥락과 의미를 최대한 보존하세요.
-        번역할 텍스트: {korean_prompt}
+        Translate the following Korean text to English accurately:
         
-        번역만 제공하고 다른 설명은 하지 마세요.
+        {korean_text}
+        
+        Provide only the translation, no explanations.
         """
         
-        # 새로운 API 형식에 맞게 수정
-        response = model.generate_content(
-            types.GenerateContentRequest(
-                contents=[types.Content(parts=[types.Part(text=translation_prompt)])],
-                generation_config=types.GenerationConfig(
-                    temperature=0.2,
-                    max_output_tokens=1024,
-                    top_p=0.9,
-                )
-            )
-        )
+        # 번역 요청
+        response = client.generate_content(translation_prompt)
         
         # 응답에서 텍스트 추출
-        english_prompt = response.candidates[0].content.parts[0].text.strip()
-        logger.info(f"번역 결과: {english_prompt}")
-        return english_prompt
+        if hasattr(response, 'text'):
+            english_text = response.text.strip()
+            logger.info(f"번역 결과: {english_text}")
+            return english_text
+        else:
+            logger.warning("번역 응답에 text 속성이 없습니다.")
+            return korean_text
     except Exception as e:
         logger.exception(f"번역 중 오류 발생: {str(e)}")
-        return korean_prompt  # 오류 발생 시 원본 한국어 프롬프트 반환
+        return korean_text  # 오류 발생 시 원본 한국어 프롬프트 반환
 
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
@@ -134,7 +133,7 @@ def preprocess_prompt(prompt, image1=None, image2=None, image3=None):
     
     return prompt
 
-def generate_with_images(prompt, images, gemini_model):
+def generate_with_images(prompt, images):
     """
     공식 문서에 기반한 올바른 API 호출 방식 구현
     재시도 로직 추가
@@ -153,7 +152,7 @@ def generate_with_images(prompt, images, gemini_model):
             client = genai.Client(api_key=api_key)
             
             # 프롬프트 번역
-            english_prompt = translate_to_english(prompt, gemini_model) if gemini_model else prompt
+            english_prompt = get_translation(prompt)
             logger.info(f"원본 프롬프트: {prompt}")
             logger.info(f"번역된 프롬프트: {english_prompt}")
             
@@ -222,7 +221,7 @@ def generate_with_images(prompt, images, gemini_model):
                 logger.exception("이미지 생성 중 오류 발생:")
                 return None, f"오류 발생: {str(e)}"
 
-def process_images_with_prompt(image1, image2, image3, prompt, gemini_model):
+def process_images_with_prompt(image1, image2, image3, prompt):
     """
     3개의 이미지와 프롬프트를 처리하는 함수
     """
@@ -250,7 +249,7 @@ def process_images_with_prompt(image1, image2, image3, prompt, gemini_model):
             prompt = preprocess_prompt(prompt, image1, image2, image3)
         
         # 이미지 생성 API 호출
-        return generate_with_images(prompt, images, gemini_model)
+        return generate_with_images(prompt, images)
             
     except Exception as e:
         logger.exception("이미지 처리 중 오류 발생:")
@@ -272,9 +271,6 @@ def update_prompt_from_function(function_choice):
 
 # Gradio 인터페이스 (수정된 버전)
 def create_interface():
-    # Gemini 모델 초기화
-    gemini_model = initialize_gemini()
-    
     with gr.Blocks() as demo:
         gr.HTML(
             """
@@ -338,7 +334,7 @@ def create_interface():
         def process_and_show_prompt(image1, image2, image3, prompt):
             try:
                 # 이미지 생성 함수 호출
-                result_img, status = process_images_with_prompt(image1, image2, image3, prompt, gemini_model)
+                result_img, status = process_images_with_prompt(image1, image2, image3, prompt)
                 
                 # 프롬프트 전처리
                 processed_prompt = preprocess_prompt(prompt, image1, image2, image3)