Update app.py

app.py

@@ -268,14 +268,41 @@ def process_images_with_prompt(image1, prompt, variation_index=0, max_retries=3)
 
 # ------------------- 프롬프트 관련 함수 -------------------
 def filter_prompt_only(prompt):
-    idx = prompt.find("Highly detailed commercial photography")
-    if idx != -1:
-        prompt_text = prompt[idx:].strip()
-        end_idx = prompt_text.rfind("--ar 1:1 --s 750 --q 2")
-        if end_idx != -1:
-            end_idx += len("--ar 1:1 --s 750 --q 2")
-            prompt_text = prompt_text[:end_idx].strip()
-        return prompt_text
+    """Gemini의 설명 및 불필요한 메시지를 제거하고 실제 프롬프트만 추출하는 함수"""
+    
+    # 코드 블록 내부의 프롬프트 찾기
+    code_block_pattern = r"```\s*(.*?)```"
+    code_match = re.search(code_block_pattern, prompt, re.DOTALL)
+    if code_match:
+        return code_match.group(1).strip()
+    
+    # Midjourney 파라미터를 포함하는 프롬프트 부분 찾기
+    if "--ar 1:1" in prompt:
+        lines = prompt.split('\n')
+        prompt_lines = []
+        in_prompt = False
+        
+        for line in lines:
+            # 프롬프트 시작 부분 인식 (일반적인 제품 설명이나 'Magazine-worthy' 같은 키워드로 시작)
+            if (not in_prompt and 
+                ("product" in line.lower() or 
+                 "magazine" in line.lower() or 
+                 "commercial" in line.lower() or 
+                 "photography" in line.lower())):
+                in_prompt = True
+                prompt_lines.append(line)
+            # 이미 프롬프트 영역에 있는 경우 계속 추가
+            elif in_prompt:
+                # 설명이나 메타 텍스트가 시작되면 중단
+                if "explanation" in line.lower() or "let me know" in line.lower():
+                    break
+                prompt_lines.append(line)
+        
+        # 프롬프트 라인 합치기
+        if prompt_lines:
+            return '\n'.join(prompt_lines).strip()
+    
+    # 위 방법으로 찾지 못한 경우 원본 반환
     return prompt.strip()
 
 def get_selected_background_info(bg_type, simple, studio, nature, indoor, tech, colorful, abstract):
@@ -338,22 +365,14 @@ def generate_enhanced_system_instruction():
 다음 가이드라인을 반드시 준수해야 합니다:
 
 1. 상품을 "#1"로 지정하여 참조합니다. (예: "skincare tube (#1)")
-
 2. *** 매우 중요: 상품의 원래 특성(디자인, 색상, 형태, 로고, 패키지 등)은 어떤 상황에서도 절대 변경하지 않습니다. ***
-
 3. *** 상품의 본질적 특성을 유지하되, 상품에 포커스를 맞춰 모든 세부 사항이 선명하게 드러나도록 하며, 
    8K 해상도(8K resolution), 오버샤프닝 없는 초고화질(ultra high definition without oversharpening)로 렌더링되어야 합니다. ***
-
 4. 이미지 비율은 정확히 1:1(정사각형) 형식으로 지정합니다. 프롬프트에 "square format", "1:1 ratio" 또는 "aspect ratio 1:1"을 명시적으로 포함합니다.
-
 5. 상품은 반드시 정사각형 구도의 정중앙에 배치되어야 하며, 적절한 크기로 표현하여 디테일이 완벽하게 보이도록 합니다.
-
 6. 상품을 이미지의 주요 초점으로 부각시키고, 상품의 비율이 전체 이미지에서 60-70% 이상 차지하도록 합니다.
-
 7. 조명 설명을 매우 구체적으로 해주세요. 예: "soft directional lighting from left side", "dramatic rim lighting", "diffused natural light through windows"
-
 8. 배경의 재질과 질감을 상세히 설명해주세요. 예: "polished marble surface", "rustic wooden table with visible grain", "matte concrete wall with subtle texture"
-
 9. 프롬프트에 다음 요소들을 명시적으로 포함하되, 사용 맥락에 적절하게 변형하세요:
    - "award-winning product photography" 
    - "magazine-worthy commercial product shot"
@@ -363,10 +382,9 @@ def generate_enhanced_system_instruction():
    - "commercial product photography with precise detail rendering"
    - "ultra high definition"
    - "crystal clear details"
-
 10. 사용자가 제공한 구체적인 배경과 추가 요청사항을 프롬프트에 정확히 반영하고 확장합니다.
-
 11. 프롬프트 끝에 "--ar 1:1 --s 750 --q 2 --v 5.2" 파라미터를 추가하여 미드저니에서 고품질 정사각형 비율을 강제합니다.
+12. 매우 중요: 프롬프트 외에 다른 설명이나 메타 텍스트를 포함하지 마세요. 예를 들어 "Here's a prompt for you" 또는 "Let me know if you need adjustments" 같은 메시지나 설명을 포함하지 마세요. 오직 프롬프트 자체만 제공하세요.
 """
 
 def generate_prompt_with_gemini(product_name, background_info, additional_info=""):