Update app.py

app.py

@@ -846,24 +846,28 @@ def process_input(prompt: str, uploaded_files):
 
             # ── ⑤ 이미지 생성
             if st.session_state.generate_image and full_response:
-                idea_sections = re.split(r"(## Idea \d+:)", full_response)
+
+                idea_sections = re.split(r"(##\s*Idea\s*\d+\s*:?)", full_response, flags=re.IGNORECASE)
                 pairs = [(idea_sections[i].strip(),
                           idea_sections[i+1].strip() if i+1 < len(idea_sections) else "")
                          for i in range(1, len(idea_sections), 2)]
 
+# ❷ 이미지 프롬프트 추출 – 표/콜론/하이픈 모든 변형 지원
                 for title, text_block in pairs:
+    # 표 형태  | Image Prompt | ... |   (굵게 표기 유/무)
                     table_match = re.search(
-                        r"\|\s*\*\*Image\s+Prompt\*\*\s*\|\s*([^\n\|]+)",
+                        r"\|\s*(?:\*\*)?Image\s+Prompt(?:\*\*)?\s*\|\s*([^|\n]+)",  # ← 수정
                         text_block, flags=re.IGNORECASE)
+
+    # 텍스트 형태  Image Prompt : ...   또는  Image Prompt - ...
                     if not table_match:
                         table_match = re.search(
-                            r"(?i)Image\s+Prompt\s*[:|-]\s*([^\n]+)", text_block)
+                            r"(?i)Image\s+Prompt\s*[:\-]\s*([^\n]+)", text_block)
 
                     if not table_match:
-                        continue
+                        continue  # 이 아이디어엔 프롬프트 없음 → 건너뜀
 
-                    raw_prompt = re.sub(r"[\r\n\|`'\"\\]", " ",
-                                        table_match.group(1).strip())
+                    raw_prompt = re.sub(r"[\r\n`\"'\\]", " ", table_match.group(1)).strip()
 
                     with st.spinner(f"Generating image for {title}…"):
                         img, cap = generate_image(raw_prompt)
@@ -877,6 +881,8 @@ def process_input(prompt: str, uploaded_files):
                             "image_caption": f"{title} – {cap}"
                         })
 
+  
+
             # ── ⑥ 결과 저장
             st.session_state.messages.append(
                 {"role": "assistant", "content": full_response})