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,7 +294,6 @@ physical_transformation_categories_en = {
         "Data-driven decision making / AI adoption",
         "Convergence of new technologies / Innovative investments"
     ]
-    
 }
 
 # ──────────────────────────────── Logging ────────────────────────────────
@@ -302,7 +301,6 @@ 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."""
@@ -326,8 +324,6 @@ def get_idea_system_prompt(selected_category: str | None = None) -> str:
         f'이 카테고리의 항목들을 2단계와 3단계 모두에서 우선적으로 고려하십시오.\n'
     ) if selected_category else ""
 
-
-    
     prompt = f"""
 반드시 한글(한국어)로 답변하라. 당신은 혁신 컨설턴트로서 CCM(크로스 카테고리 매트릭스) 방법론을 활용하여 창의적 아이디어를 도출합니다.
 
@@ -645,7 +641,7 @@ def idea_generator_app():
 
     # 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:
@@ -668,8 +664,7 @@ def idea_generator_app():
     if web_search_enabled:
         sb.info("✅ Web search results will be integrated.")
 
-    # 예시 주제들 (원래 예시 블로그 토픽 -> 이제는 예시 아이디어 주제로 전환)
-
+    # 예시 주제들
     example_topics = {
         "example1": "도시 물 부족 문제 해결을 위한 혁신적 방안",
         "example2": "노인 돌봄 서비스의 디지털 전환",
@@ -685,7 +680,6 @@ def idea_generator_app():
         index=0  # 기본값 "(None)"
     )
 
-    
     sb.subheader("Example Prompts")
     c1, c2, c3 = sb.columns(3)
     if c1.button("도시 물 부족 문제", key="ex1"):
@@ -803,13 +797,10 @@ 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, [])
 
-# ── 수정된 process_input 전체 ────────────────────────────────────────────
-
 # ──────────────────────────────── 헬퍼: 결과 기록·다운로드·자동저장 ──────────
 def write_output(md_text: str, prompt: str):
     """
@@ -818,8 +809,7 @@ def write_output(md_text: str, prompt: str):
     • 자동 JSON 백업
     """
     # ① 채팅 기록에 추가
-    st.session_state.messages.append(
-        {"role": "assistant", "content": md_text})
+    st.session_state.messages.append({"role": "assistant", "content": md_text})
 
     # ② 다운로드 버튼
     st.subheader("Download This Output")
@@ -840,21 +830,20 @@ def write_output(md_text: str, prompt: str):
     # ③ 자동 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
-            )
-
+        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})
@@ -862,55 +851,47 @@ def process_input(prompt: str, uploaded_files):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # ── 1. 어시스턴트 응답 영역 ──────────────────────────────────────────
+    # GPT 호출을 위한 초기화
+    use_web_search = st.session_state.web_search_enabled
+    has_uploaded   = bool(uploaded_files)
+    full_response  = ""
+
     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:
-            # 1-A. 모델·상태 초기화
-            status = st.status("Preparing to generate ideas…")
-            status.update(label="Initializing model…")
-            client = get_openai_client()
+            with st.spinner("Preparing to generate ideas..."):
+                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())
 
-            # 1-C. (선택) 웹 검색 + 파일 내용
-            search_content = None
+            # 웹 검색 및 파일 내용 결합
+            user_content = prompt
             if use_web_search:
-                status.update(label="Searching the web…")
-                with st.spinner("Searching…"):
+                with st.spinner("Searching the web..."):
                     search_content = do_web_search(keywords(prompt, top=5))
-
-            file_content = None
+                    user_content += "\n\n" + search_content
             if has_uploaded:
-                status.update(label="Reading uploaded files…")
-                with st.spinner("Processing files…"):
+                with st.spinner("Processing uploaded 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",
@@ -918,112 +899,44 @@ def process_input(prompt: str, uploaded_files):
                 {"role": "user", "content": user_content},
             ]
 
-            # ── 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:
-                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)
-                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"아이디어 시각화 – {cap}")
-                        st.session_state.messages.append({
-                            "role": "assistant",
-                            "content": "",
-                            "image": img,
-                            "image_caption": f"아이디어 시각화 – {cap}"
-                        })
-
-            # ── 4. 결과를 **한 번만** 기록·다운로드·백업 ────────────────
-            write_output(full_response, prompt)
-
-        except Exception as e:
-            # 예외 발생 시: 로그 기록 + 사용자 알림만 수행
-            logging.error("process_input error", exc_info=True)
-            placeholder.error(f"⚠️ 작업 중 오류가 발생했습니다: {e}")
-            st.session_state.messages.append(
-                {"role": "assistant", "content": f"⚠️ 오류: {e}"})
-
-
-            # ---- 카테고리/하위항목 컨텍스트 추가 ---------------------------
-            def category_context(sel):
-                if sel:   # 특정 카테고리 선택 시
-                    return json.dumps(
-                        {sel: physical_transformation_categories[sel]},
-                        ensure_ascii=False)
-                # (None) → 키 목록만 전달
-                return "ALL_CATEGORIES: " + ", ".join(
-                    physical_transformation_categories.keys())
+            # 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 + "▌")
 
-            api_messages = [
-                {"role": "system", "content": sys_prompt},
-                {"role": "system", "name": "category_db",
-                 "content": category_context(selected_cat)},
-                {"role": "user",   "content": user_content},
-            ]
-            # --------------------------------------------------------------
-
-            # ④ 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:
-                    full_response += chunk.choices[0].delta.content
-                    message_placeholder.markdown(full_response + "▌")
+            # 최종 출력
             message_placeholder.markdown(full_response)
-            status.update(label="Ideas created!", state="complete")
 
-            # ⑤ 이미지 생성 (CCM 헤딩 or 레거시 패턴 모두 지원)
+            # 이미지 자동 생성
             if st.session_state.generate_image and full_response:
-                # CCM 헤딩
-                ccm_match = re.search(
-                    r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)",
-                    full_response, flags=re.IGNORECASE)
+                # 패턴1: "### 이미지 프롬프트" 형태 탐색
+                ccm_match = re.search(r"###\s*이미지\s*프롬프트\s*\n+([^\n]+)",
+                                      full_response, flags=re.IGNORECASE)
+                # 패턴2: 예전 형식 "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) \
-                        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("아이디어 이미지 생성 중…"):
+                    raw_prompt = re.sub(r"[\r\n`\"'\\]", " ", match.group(1)).strip()
+                    with st.spinner("Generating idea image..."):
                         img, cap = generate_image(raw_prompt)
                     if img:
                         st.image(img, caption=f"아이디어 시각화 – {cap}")
@@ -1034,48 +947,14 @@ def process_input(prompt: str, uploaded_files):
                             "image_caption": f"아이디어 시각화 – {cap}"
                         })
 
-        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}"
-            })
-
-
-            # Download buttons
-            if full_response:
-                st.subheader("Download This Output")
-                c1, c2 = st.columns(2)
-                c1.download_button(
-                    "Markdown", 
-                    data=full_response, 
-                    file_name=f"{prompt[:30]}.md",
-                    mime="text/markdown"
-                )
-                c2.download_button(
-                    "HTML",
-                    data=md_to_html(full_response, prompt[:30]),
-                    file_name=f"{prompt[:30]}.html",
-                    mime="text/html"
-                )
-
-            # Auto-save
-            if st.session_state.auto_save and st.session_state.messages:
-                try:
-                    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:
-                    logging.error(f"Auto-save failed: {e}")
+            # 결과 기록·다운로드·백업
+            write_output(full_response, prompt)
 
         except Exception as e:
-            error_message = str(e)
-            placeholder.error(f"An error occurred: {error_message}")
-            logging.error(f"Process input error: {error_message}")
-            ans = f"An error occurred while processing your request: {error_message}"
-            st.session_state.messages.append({"role": "assistant", "content": ans})
+            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})
 
 # ──────────────────────────────── main ────────────────────────────────────
 def main():
@@ -1083,4 +962,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-