Update app.py

app.py

@@ -25,16 +25,36 @@ def translate_prompt_to_english(prompt):
     """
     입력된 프롬프트에 한글이 포함되어 있으면 Gemini‑2.0‑flash 모델을 사용하여 영어로 번역합니다.
     한글이 없으면 원본 프롬프트를 그대로 반환합니다.
+    중요: #1, #2, #3 태그는 번역 전후에 반드시 보존되어야 합니다.
     """
     if not re.search("[가-힣]", prompt):
         return prompt
+    
+    # #1, #2, #3 태그를 임시 토큰으로 대체하여 보존
+    prompt = prompt.replace("#1", "IMAGE_TAG_ONE")
+    prompt = prompt.replace("#2", "IMAGE_TAG_TWO")
+    prompt = prompt.replace("#3", "IMAGE_TAG_THREE")
+    
     try:
         api_key = os.environ.get("GEMINI_API_KEY")
         if not api_key:
             logger.error("Gemini API 키가 설정되지 않았습니다.")
+            # 임시 토큰을 원래 태그로 복원
+            prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
+            prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
+            prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
             return prompt
+            
         client = genai.Client(api_key=api_key)
-        translation_prompt = f"Translate the following Korean text to English:\n\n{prompt}"
+        translation_prompt = f"""
+        Translate the following Korean text to English:
+        
+        {prompt}
+        
+        IMPORTANT: The tokens IMAGE_TAG_ONE, IMAGE_TAG_TWO, and IMAGE_TAG_THREE are special tags 
+        and must be preserved exactly as is in your translation. Do not translate these tokens.
+        """
+        
         logger.info(f"Translation prompt: {translation_prompt}")
         response = client.models.generate_content(
             model="gemini-2.0-flash",
@@ -47,18 +67,32 @@ def translate_prompt_to_english(prompt):
                 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
+        
         if translated_text.strip():
+            # 번역된 텍스트에서 임시 토큰을 원래 태그로 복원
+            translated_text = translated_text.replace("IMAGE_TAG_ONE", "#1")
+            translated_text = translated_text.replace("IMAGE_TAG_TWO", "#2")
+            translated_text = translated_text.replace("IMAGE_TAG_THREE", "#3")
             logger.info(f"Translated text: {translated_text.strip()}")
             return translated_text.strip()
         else:
             logger.warning("번역 결과가 없습니다. 원본 프롬프트 사용")
+            # 임시 토큰을 원래 태그로 복원
+            prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
+            prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
+            prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
             return prompt
     except Exception as e:
         logger.exception("번역 중 오류 발생:")
+        # 임시 토큰을 원래 태그로 복원
+        prompt = prompt.replace("IMAGE_TAG_ONE", "#1")
+        prompt = prompt.replace("IMAGE_TAG_TWO", "#2")
+        prompt = prompt.replace("IMAGE_TAG_THREE", "#3")
         return prompt
 
 def preprocess_prompt(prompt, image1, image2, image3):