Update app.py

app.py

@@ -878,9 +878,27 @@ def process_input(prompt: str, uploaded_files):
 
                 # 각 아이디어마다 이미지 생성
                 for idx, (title, text_block) in enumerate(pairs, start=1):
-                    image_prompt_match = re.search(r"(?i)(image prompt\s*\:\s*)(.+)", text_block)
-                    if image_prompt_match:
-                        raw_prompt = image_prompt_match.group(2).strip()
+
+                    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}"
+                                })
+
+                        
                         # 만약 문장 끝에 불필요한 구두점이나 줄바꿈이 있을 시 제거
                         raw_prompt = re.sub(r"[\r\n]+", " ", raw_prompt)
                         raw_prompt = re.sub(r"[\"'`]", "", raw_prompt)