Update app.py

app.py

@@ -877,26 +877,38 @@ def process_input(prompt: str, uploaded_files):
                     pairs.append((idea_title, idea_body))
 
                 # 각 아이디어마다 이미지 생성
-                for idx, (title, text_block) in enumerate(pairs, start=1):
-
-                    if not table_match:
+# 각 아이디어마다 이미지 생성  (!!! 전체 블록 교체 !!!)
+                    for idx, (title, text_block) in enumerate(pairs, start=1):
+    # ① 표(마크다운 테이블) 안에 들어있는 형태  ──>  | **Image Prompt** | prompt |
                         table_match = re.search(
-                            r"(?i)Image\s+Prompt\s*[:|-]\s*([^\n]+)", text_block
+                            r"\|\s*\*\*Image\s+Prompt\*\*\s*\|\s*([^\n|]+)",
+                            text_block,
+                            flags=re.IGNORECASE,
                         )
 
-                    if table_match:
-                        raw_prompt = table_match.group(1).strip()
-                        raw_prompt = re.sub(r"[\r\n\|`'\"]", " ", raw_prompt)  # 불필요 문자 제거
-                        with st.spinner(f"Generating image for {title}..."):
-                            img, cap = generate_image(raw_prompt)
-                                if img:
-                                st.image(img, caption=f"{title} - {cap}")
-                                st.session_state.messages.append({
-                                    "role": "assistant",
-                                    "content": "",
-                                    "image": img,
-                                    "image_caption": f"{title} - {cap}"
-                                })
+    # ② 백업 패턴:  "Image Prompt: ..."  또는  "Image Prompt - ..."
+                        if not table_match:
+                            table_match = re.search(
+                                r"(?i)Image\s+Prompt\s*[:|-]\s*([^\n]+)", text_block
+                            )
+
+                        if table_match:
+                            raw_prompt = table_match.group(1).strip()
+                            raw_prompt = re.sub(r"[\r\n\|`'\"\\]", " ", raw_prompt)  # 개행·구두점 정리
+
+        # 이미지 생성
+                            with st.spinner(f"Generating image for {title}..."):
+                                img, cap = generate_image(raw_prompt)
+                            if img:
+                                st.image(img, caption=f"{title} – {cap}")
+                                st.session_state.messages.append(
+                                    {
+                                        "role": "assistant",
+                                        "content": "",
+                                        "image": img,
+                                        "image_caption": f"{title} – {cap}",
+                                    }
+                                )
 
                         
                         # 만약 문장 끝에 불필요한 구두점이나 줄바꿈이 있을 시 제거