Update app.py

app.py

@@ -688,11 +688,11 @@ def idea_generator_app():
     
     sb.subheader("Example Prompts")
     c1, c2, c3 = sb.columns(3)
-    if c1.button("AI & 비즈니스", key="ex1"):
+    if c1.button("도시 물 부족 문제", key="ex1"):
         process_example(example_topics["example1"])
-    if c2.button("스마트 소재 건축", key="ex2"):
+    if c2.button("노인 돌봄 서비스", key="ex2"):
         process_example(example_topics["example2"])
-    if c3.button("미래형 교육", key="ex3"):
+    if c3.button("지속가능한 식품 포장", key="ex3"):
         process_example(example_topics["example3"])
 
     # Download the latest ideas
@@ -880,42 +880,46 @@ def process_input(prompt: str, uploaded_files):
             status.update(label="Ideas created!", state="complete")
 
             # ── ⑤ 이미지 생성
+# ── ⑤ 이미지 생성  (기존 블록 전부 교체) ─────────────────────────────
             if st.session_state.generate_image and 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:
-    
-                    table_match = re.search(
-                        r"\|\s*(?:\*\*)?Image\s+Prompt(?:\*\*)?\s*\|\s*([^|\n]+)",  # ← 수정
-                        text_block, flags=re.IGNORECASE)
-
-                    if not table_match:
-                        table_match = re.search(
-                            r"(?i)Image\s+Prompt\s*[:\-]\s*([^\n]+)", text_block)
-
-                    if not table_match:
-                        continue  # 이 아이디어엔 프롬프트 없음 → 건너뜀
-
-                    raw_prompt = re.sub(r"[\r\n`\"'\\]", " ", table_match.group(1)).strip()
-
-                    with st.spinner(f"Generating image for {title}…"):
+    # 1️⃣ ❶ CCM 스타일: "### 이미지 프롬프트"  헤딩 아래 줄
+                ccm_match = re.search(
+                    r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)",
+                    full_response, flags=re.IGNORECASE)
+
+    # 2️⃣ ❷ 이전 스타일: 테이블 or "Image Prompt:"
+                legacy_match = None
+                if not ccm_match:
+                    legacy_match = re.search(
+                        r"\|\s*(?:\*\*)?Image\s+Prompt(?:\*\*)?\s*\|\s*([^|\n]+)",
+                        full_response, flags=re.IGNORECASE)
+                    if not legacy_match:
+                        legacy_match = re.search(
+                            r"(?i)Image\s+Prompt\s*[:\-]\s*([^\n]+)",
+                            full_response)
+
+    # 3️⃣ 최종 프롬프트 추출
+                match = ccm_match or legacy_match
+                if match:
+                    raw_prompt = re.sub(r"[\r\n`\"'\\]", " ",
+                                        match.group(1)).strip()
+
+                    with st.spinner("아이디어 이미지 생성 중…"):
                         img, cap = generate_image(raw_prompt)
 
                     if img:
-                        st.image(img, caption=f"{title} – {cap}")
+                        st.image(img, caption=f"아이디어 시각화 – {cap}")
+            # 대화 기록에 이미지도 저장
                         st.session_state.messages.append({
                             "role": "assistant",
                             "content": "",
                             "image": img,
-                            "image_caption": f"{title} – {cap}"
+                            "image_caption": f"아이디어 시각화 – {cap}"
                         })
+            
+
 
-  
 
             # ── ⑥ 결과 저장
             st.session_state.messages.append(