Update app.py

app.py

@@ -775,48 +775,47 @@ def process_example(topic):
 
 def process_input(prompt: str, uploaded_files):
     # Add user's message
-    if not any(m["role"] == "user" and m["content"] == prompt for m in st.session_state.messages):
+    if not any(m["role"] == "user" and m["content"] == prompt
+               for m in st.session_state.messages):
         st.session_state.messages.append({"role": "user", "content": prompt})
 
     with st.chat_message("user"):
         st.markdown(prompt)
-    
+
     with st.chat_message("assistant"):
         placeholder = st.empty()
         message_placeholder = st.empty()
         full_response = ""
 
-        use_web_search = st.session_state.web_search_enabled
-        has_uploaded_files = bool(uploaded_files) and len(uploaded_files) > 0
-        
+        use_web_search   = st.session_state.web_search_enabled
+        has_uploaded     = bool(uploaded_files)
+
         try:
-            status = st.status("Preparing to generate ideas...")
-            status.update(label="Initializing client...")
+            status = st.status("Preparing to generate ideas…")
+            status.update(label="Initializing model…")
 
             client = get_openai_client()
-            
-            # Prepare system prompt
 
+            # ── ① 시스템 프롬프트 ──────────────────────────────
             selected_cat = st.session_state.get("category_focus", "(None)")
             if selected_cat == "(None)":
                 selected_cat = None
-
             sys_prompt = get_idea_system_prompt(selected_category=selected_cat)
-            # Optional: gather search results
+
+            # ── ② (선택) 웹 검색 & 파일내용 ───────────────────
             search_content = None
             if use_web_search:
-                status.update(label="Performing web search...")
-                with st.spinner("Searching the web..."):
+                status.update(label="Searching the web…")
+                with st.spinner("Searching…"):
                     search_content = do_web_search(keywords(prompt, top=5))
 
-            # File contents
             file_content = None
-            if has_uploaded_files:
-                status.update(label="Processing uploaded files...")
-                with st.spinner("Analyzing files..."):
+            if has_uploaded:
+                status.update(label="Reading uploaded files…")
+                with st.spinner("Processing files…"):
                     file_content = process_uploaded_files(uploaded_files)
 
-            # Build messages
+            # ── ③ 대화 메시지 구성 ─────────────────────────────
             user_content = prompt
             if search_content:
                 user_content += "\n\n" + search_content
@@ -825,108 +824,91 @@ def process_input(prompt: str, uploaded_files):
 
             api_messages = [
                 {"role": "system", "content": sys_prompt},
-                {"role": "user", "content": user_content}
+                {"role": "user",   "content": user_content},
             ]
 
-            # OpenAI API streaming
-            status.update(label="Generating creative ideas...")
-            try:
-                stream = client.chat.completions.create(
-                    model="gpt-4.1-mini",
-                    messages=api_messages,
-                    temperature=1,
-                    max_tokens=MAX_TOKENS,
-                    top_p=1,
-                    stream=True
-                )
-                for chunk in stream:
-                    if (chunk.choices 
-                        and len(chunk.choices) > 0 
+            # ── ④ OpenAI 스트리밍 호출 ────────────────────────
+            status.update(label="Generating ideas…")
+            stream = client.chat.completions.create(
+                model       = "gpt-4.1-mini",
+                messages    = api_messages,
+                temperature = 1,
+                max_tokens  = MAX_TOKENS,
+                top_p       = 1,
+                stream      = True
+            )
+            for chunk in stream:
+                if (chunk.choices
                         and chunk.choices[0].delta.content is not None):
-                        content_delta = chunk.choices[0].delta.content
-                        full_response += content_delta
-                        message_placeholder.markdown(full_response + "▌")
-                message_placeholder.markdown(full_response)
-                status.update(label="Ideas created!", state="complete")
-
-            except Exception as api_error:
-                error_message = str(api_error)
-                logging.error(f"API error: {error_message}")
-                status.update(label=f"Error: {error_message}", state="error")
-                raise Exception(f"Idea generation error: {error_message}")
-
-            # Store final text
-            answer_entry_saved = False
-
-            # 자동 이미지 생성이 활성화되어 있다면
+                    full_response += chunk.choices[0].delta.content
+                    message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_response)
+            status.update(label="Ideas created!", state="complete")
+
+            # ── ⑤ 이미지 생성 ────────────────────────────────
             if st.session_state.generate_image and full_response:
-                # 아이디어를 3개로 구분하고, 각 아이디어마다 이미지 프롬프트를 추출
-                # 간단한 방식: 정규식으로 `Image Prompt:` 형태를 찾는다고 가정 (시스템프롬프트에 지시)
-                # 혹은 단순히 한 번만 추출 -> 여기서는 3개 아이디어 각각을 찾기 위해 나눠본다.
-                # 일단은 전체 답변에서 "English prompt for generating an image" 부분을 찾는다.
-                
-                # 매우 단순한 파싱 예시 (개선 가능)
                 idea_sections = re.split(r"(## Idea \d+:)", full_response)
-                # idea_sections는 ['', '## Idea 1:', ' ...', '## Idea 2:', ' ...', '## Idea 3:', ' ...'] 형태
-                
-                # 다시 묶어서 [('## Idea 1:', idea_text), ('## Idea 2:', idea_text), ...] 형태로
-                pairs = []
-                for i in range(1, len(idea_sections), 2):
-                    idea_title = idea_sections[i].strip()
-                    idea_body = idea_sections[i+1].strip() if i+1 < len(idea_sections) else ""
-                    pairs.append((idea_title, idea_body))
-
-                # 각 아이디어마다 이미지 생성
-# 각 아이디어마다 이미지 생성  (!!! 전체 블록 교체 !!!)
-                    for idx, (title, text_block) in enumerate(pairs, start=1):
-    # ① 표(마크다운 테이블) 안에 들어있는 형태  ──>  | **Image Prompt** | prompt |
+                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 idx, (title, text_block) in enumerate(pairs, start=1):
+                    # 표 형태: | **Image Prompt** | prompt |
+                    table_match = re.search(
+                        r"\|\s*\*\*Image\s+Prompt\*\*\s*\|\s*([^\n\|]+)",
+                        text_block, flags=re.IGNORECASE)
+                    # 백업 형태: Image Prompt: …
+                    if not table_match:
                         table_match = re.search(
-                            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
-                            )
-
-                        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}",
-                                    }
-                                )
+                            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}…"):
+                        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}"
+                        })
+
+            # ── ⑥ 결과 저장 & 다운로드 ──────���──────────────────
+            st.session_state.messages.append(
+                {"role": "assistant", "content": full_response})
+
+            st.subheader("Download This Output")
+            c1, c2 = st.columns(2)
+            c1.download_button("Markdown", full_response,
+                               file_name=f"{prompt[:30]}.md",
+                               mime="text/markdown")
+            c2.download_button("HTML",
+                               md_to_html(full_response, prompt[:30]),
+                               file_name=f"{prompt[:30]}.html",
+                               mime="text/html")
+
+            # ── ⑦ 자동 저장 ─────────────────────────────────
+            if st.session_state.auto_save:
+                fn = f"chat_history_auto_{datetime.now():%Y%m%d_%H%M%S}.json"
+                with open(fn, "w", encoding="utf-8") as fp:
+                    json.dump(st.session_state.messages, fp,
+                              ensure_ascii=False, indent=2)
+
+        except Exception as e:
+            err = str(e)
+            placeholder.error(f"Error: {err}")
+            logging.error(err)
+            st.session_state.messages.append(
+                {"role": "assistant",
+                 "content": f"⚠️ 작업 중 오류가 발생했습니다: {err}"})
 
-                        
-                        # 만약 문장 끝에 불필요한 구두점이나 줄바꿈이 있을 시 제거
-                        raw_prompt = re.sub(r"[\r\n]+", " ", raw_prompt)
-                        raw_prompt = re.sub(r"[\"'`]", "", 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}"
-                                })
-                
                 # 3개 이미지 생성 프로세스를 마친 후 최종 텍스트를 저장
                 st.session_state.messages.append({"role": "assistant", "content": full_response})
                 answer_entry_saved = True