Update app.py

app.py

@@ -19,7 +19,7 @@ BRAVE_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
 IMAGE_API_URL  = "http://211.233.58.201:7896"  # 이미지 생성용 API
 MAX_TOKENS     = 7999
 
-# ──────────────────────────────── Physical Transformation Categories (KR & EN) ────────────────
+# ──────────────────────────────── Physical Transformation Categories (KR & EN) ─────────────────
 physical_transformation_categories = {
     "센서 기능": [
         "시각 센서/감지", "청각 센서/감지", "촉각 센서/감지", "미각 센서/감지", "후각 센서/감지",
@@ -294,6 +294,7 @@ physical_transformation_categories_en = {
         "Data-driven decision making / AI adoption",
         "Convergence of new technologies / Innovative investments"
     ]
+    
 }
 
 # ──────────────────────────────── Logging ────────────────────────────────
@@ -301,6 +302,7 @@ logging.basicConfig(level=logging.INFO,
                     format="%(asctime)s - %(levelname)s - %(message)s")
 
 # ──────────────────────────────── OpenAI Client ──────────────────────────
+
 @st.cache_resource
 def get_openai_client():
     """Create an OpenAI client with timeout and retry settings."""
@@ -324,6 +326,8 @@ def get_idea_system_prompt(selected_category: str | None = None) -> str:
         f'이 카테고리의 항목들을 2단계와 3단계 모두에서 우선적으로 고려하십시오.\n'
     ) if selected_category else ""
 
+
+    
     prompt = f"""
 반드시 한글(한국어)로 답변하라. 당신은 혁신 컨설턴트로서 CCM(크로스 카테고리 매트릭스) 방법론을 활용하여 창의적 아이디어를 도출합니다.
 
@@ -635,13 +639,49 @@ def keywords(text: str, top=5):
     cleaned = re.sub(r"[^가-힣a-zA-Z0-9\s]", "", text)
     return " ".join(cleaned.split()[:top])
 
+# ──────────────────────────────── 헬퍼: 결과 기록·다운로드·자동저장 ──────────
+def write_output(md_text: str, prompt: str):
+    """
+    • 대화 기록에 마크다운 답변 저장
+    • 다운로드 버튼(마크다운·HTML) 제공
+    • 자동 JSON 백업
+    """
+    # ① 채팅 기록에 추가
+    st.session_state.messages.append(
+        {"role": "assistant", "content": md_text})
+
+    # ② 다운로드 버튼
+    st.subheader("Download This Output")
+    col_md, col_html = st.columns(2)
+    col_md.download_button(
+        label="Markdown",
+        data=md_text,
+        file_name=f"{prompt[:30]}.md",
+        mime="text/markdown"
+    )
+    col_html.download_button(
+        label="HTML",
+        data=md_to_html(md_text, prompt[:30]),
+        file_name=f"{prompt[:30]}.html",
+        mime="text/html"
+    )
+
+    # ③ 자동 JSON 저장
+    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
+            )
+
 # ──────────────────────────────── Streamlit UI ────────────────────────────
 def idea_generator_app():
     st.title("Creative Idea Generator")
 
     # Set default session state
     if "ai_model" not in st.session_state:
-        st.session_state.ai_model = "gpt-4.1-mini"
+        st.session_state.ai_model = "gpt-4.1-mini"  
     if "messages" not in st.session_state:
         st.session_state.messages = []
     if "auto_save" not in st.session_state:
@@ -664,7 +704,8 @@ def idea_generator_app():
     if web_search_enabled:
         sb.info("✅ Web search results will be integrated.")
 
-    # 예시 주제들
+    # 예시 주제들 (원래 예시 블로그 토픽 -> 이제는 예시 아이디어 주제로 전환)
+
     example_topics = {
         "example1": "도시 물 부족 문제 해결을 위한 혁신적 방안",
         "example2": "노인 돌봄 서비스의 디지털 전환",
@@ -680,6 +721,7 @@ def idea_generator_app():
         index=0  # 기본값 "(None)"
     )
 
+    
     sb.subheader("Example Prompts")
     c1, c2, c3 = sb.columns(3)
     if c1.button("도시 물 부족 문제", key="ex1"):
@@ -797,53 +839,19 @@ def idea_generator_app():
     sb.markdown("---")
     sb.markdown("Created by [Ginigen.com](https://ginigen.com) | [YouTube](https://www.youtube.com/@ginipickaistudio)")
 
+
 def process_example(topic):
     """Handle example prompts."""
     process_input(topic, [])
 
-# ───────────────���──────────────── 헬퍼: 결과 기록·다운로드·자동저장 ──────────
-def write_output(md_text: str, prompt: str):
-    """
-    • 대화 기록에 마크다운 답변 저장
-    • 다운로드 버튼(마크다운·HTML) 제공
-    • 자동 JSON 백업
-    """
-    # ① 채팅 기록에 추가
-    st.session_state.messages.append({"role": "assistant", "content": md_text})
-
-    # ② 다운로드 버튼
-    st.subheader("Download This Output")
-    col_md, col_html = st.columns(2)
-    col_md.download_button(
-        label="Markdown",
-        data=md_text,
-        file_name=f"{prompt[:30]}.md",
-        mime="text/markdown"
-    )
-    col_html.download_button(
-        label="HTML",
-        data=md_to_html(md_text, prompt[:30]),
-        file_name=f"{prompt[:30]}.html",
-        mime="text/html"
-    )
-
-    # ③ 자동 JSON 저장
-    if st.session_state.auto_save:
-        fn = f"chat_history_auto_{datetime.now():%Y%m%d_%H%M%S}.json"
-        try:
-            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:
-            logging.error(f"Auto-save failed: {e}")
-
-# ──────────────────────────────── process_input ────────────────────────────
+# ──────────────────────────────── process_input (중복 출력 제거) ────────────
 def process_input(prompt: str, uploaded_files):
     """
     1) 사용자 입력을 GPT-4로 보내 창의적 아이디어 보고서 생성
     2) 선택적으로 이미지 생성
-    3) 결과를 한 번만 기록·다운로드·백업 (중복 출력 방지)
+    3) 결과를 한 번만 저장·다운로드·백업
     """
-    # 사용자 메시지 기록
+    # ── 0. 사용자 메시지 기록 ──────────────────────────────────────────────
     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})
@@ -851,47 +859,55 @@ def process_input(prompt: str, uploaded_files):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # GPT 호출을 위한 초기화
-    use_web_search = st.session_state.web_search_enabled
-    has_uploaded   = bool(uploaded_files)
-    full_response  = ""
-
+    # ── 1. 어시스턴트 응답 영역 ──────────────────────────────────────────
     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   = bool(uploaded_files)
 
         try:
-            with st.spinner("Preparing to generate ideas..."):
-                client = get_openai_client()
+            # 1-A. 모델·상태 초기화
+            status = st.status("Preparing to generate ideas…")
+            status.update(label="Initializing model…")
+            client = get_openai_client()
 
-            # 시스템 프롬프트 구성
+            # 1-B. 시스템 프롬프트 + 카테고리 DB
             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)
 
-            # 카테고리 정보 (JSON 형식)
             def category_context(sel):
                 if sel:
                     return json.dumps(
                         {sel: physical_transformation_categories[sel]},
-                        ensure_ascii=False
-                    )
-                return "ALL_CATEGORIES: " + ", ".join(physical_transformation_categories.keys())
+                        ensure_ascii=False)
+                return "ALL_CATEGORIES: " + ", ".join(
+                    physical_transformation_categories.keys())
 
-            # 웹 검색 및 파일 내용 결합
-            user_content = prompt
+            # 1-C. (선택) 웹 검색 + 파일 내용
+            search_content = None
             if use_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))
-                    user_content += "\n\n" + search_content
+
+            file_content = None
             if has_uploaded:
-                with st.spinner("Processing uploaded files..."):
+                status.update(label="Reading uploaded files…")
+                with st.spinner("Processing files…"):
                     file_content = process_uploaded_files(uploaded_files)
-                    if file_content:
-                        user_content += "\n\n" + file_content
 
-            # 대화 메시지 배열
+            # 1-D. 사용자 메시지 결합
+            user_content = prompt
+            if search_content:
+                user_content += "\n\n" + search_content
+            if file_content:
+                user_content += "\n\n" + file_content
+
             api_messages = [
                 {"role": "system", "content": sys_prompt},
                 {"role": "system", "name": "category_db",
@@ -899,44 +915,40 @@ def process_input(prompt: str, uploaded_files):
                 {"role": "user", "content": user_content},
             ]
 
-            # GPT-4 스트리밍 호출
-            with st.spinner("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.get("content"):
-                        token_text = chunk.choices[0].delta["content"]
-                        full_response += token_text
-                        message_placeholder.markdown(full_response + "▌")
-
-            # 최종 출력
+            # ── 2. GPT-4 스트리밍 호출 ──────────────────────────────────
+            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:
+                    full_response += chunk.choices[0].delta.content
+                    message_placeholder.markdown(full_response + "▌")
             message_placeholder.markdown(full_response)
+            status.update(label="Ideas created!", state="complete")
 
-            # 이미지 자동 생성
+            # ── 3. 이미지 생성 (선택) ───────────────────────────────────
             if st.session_state.generate_image and full_response:
-                # 패턴1: "### 이미지 프롬프트" 형태 탐색
-                ccm_match = re.search(r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)",
-                                      full_response, flags=re.IGNORECASE)
-                # 패턴2: 예전 형식 "Image Prompt" 등
+                ccm_match = re.search(
+                    r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)",
+                    full_response, flags=re.IGNORECASE)
                 legacy_match = None
                 if not ccm_match:
                     legacy_match = re.search(
                         r"\|\s*(?:\*\*)?Image\s+Prompt(?:\*\*)?\s*\|\s*([^|\n]+)",
-                        full_response, flags=re.IGNORECASE
-                    ) or re.search(
-                        r"(?i)Image\s+Prompt\s*[:\-]\s*([^\n]+)",
-                        full_response
-                    )
+                        full_response, flags=re.IGNORECASE) or \
+                        re.search(r"(?i)Image\s+Prompt\s*[:\-]\s*([^\n]+)",
+                                  full_response)
                 match = ccm_match or legacy_match
                 if match:
-                    raw_prompt = re.sub(r"[\r\n`\"'\\]", " ", match.group(1)).strip()
-                    with st.spinner("Generating idea image..."):
+                    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"아이디어 시각화 – {cap}")
@@ -947,18 +959,19 @@ def process_input(prompt: str, uploaded_files):
                             "image_caption": f"아이디어 시각화 – {cap}"
                         })
 
-            # 결과 기록·다운로드·백업
+            # ── 4. 결과를 **한 번만** 기록·다운로드·백업 ────────────────
             write_output(full_response, prompt)
 
         except Exception as e:
-            logging.error("Error in process_input", exc_info=True)
-            err_msg = f"⚠️ 오류가 발생했습니다: {e}"
-            st.error(err_msg)
-            st.session_state.messages.append({"role": "assistant", "content": err_msg})
+            # 예외 발생 시: 로그 기록 + 사용자 알림만 수행
+            logging.error("process_input error", exc_info=True)
+            placeholder.error(f"⚠️ 작업 중 오류가 발생했습니다: {e}")
+            st.session_state.messages.append(
+                {"role": "assistant", "content": f"⚠️ 오류: {e}"})
 
 # ──────────────────────────────── main ────────────────────────────────────
 def main():
     idea_generator_app()
 
 if __name__ == "__main__":
-    main()
+    main()