Update app.py

app.py

@@ -316,17 +316,19 @@ def ginigen_app():
     
     # 예제 주제 선택
     sb.subheader("예제 주제")
-    if sb.button("2025년 부동산 세금 제도", key="example1"):
-        st.session_state.example_topic = EXAMPLE_TOPICS["example1"]
-        st.rerun()
-        
-    if sb.button("여름 지역별 축제와 명소", key="example2"):
-        st.session_state.example_topic = EXAMPLE_TOPICS["example2"]
-        st.rerun()
-        
-    if sb.button("신성장 산업 투자 가이드", key="example3"):
-        st.session_state.example_topic = EXAMPLE_TOPICS["example3"]
-        st.rerun()
+    
+    col1, col2, col3 = sb.columns(3)
+    
+    # 수정: 예제 선택 시 직접 처리하도록 변경
+    if col1.button("부동산 세금", key="ex1"):
+        # 예제 주제를 입력으로 즉시 처리 (rerun 없이)
+        process_example(EXAMPLE_TOPICS["example1"])
+    
+    if col2.button("여름 축제", key="ex2"):
+        process_example(EXAMPLE_TOPICS["example2"])
+    
+    if col3.button("투자 가이드", key="ex3"):
+        process_example(EXAMPLE_TOPICS["example3"])
     
     sb.subheader("기타 설정")
     sb.toggle("자동 저장", key="auto_save")
@@ -375,79 +377,83 @@ def ginigen_app():
             if "image" in m:
                 st.image(m["image"], caption=m.get("image_caption", ""))
 
-    # ── 사용자 입력 또는 예제 주제 처리
+    # ── 사용자 입력 처리
     prompt = st.chat_input("무엇을 도와드릴까요?")
     
-    # 예제 주제가 선택되었으면 처리
-    if hasattr(st.session_state, 'example_topic') and st.session_state.example_topic:
-        prompt = st.session_state.example_topic
-        # 처리 후 예제 주제 초기화
-        del st.session_state.example_topic
-    
     if prompt:
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user"): st.markdown(prompt)
+        process_input(prompt)
 
-        with st.chat_message("assistant"):
-            placeholder = st.empty(); answer = ""
-            
-            # 선택된 템플릿, 톤, 단어 수로 시스템 프롬프트 생성
-            sys_prompt = get_system_prompt(
-                template=st.session_state.blog_template,
-                tone=st.session_state.blog_tone,
-                word_count=st.session_state.word_count
-            )
-
-            if st.session_state.use_web_search:
-                with st.spinner("웹 검색 중…"):
-                    search_md = do_web_search(keywords(prompt))
-                    sys_prompt += f"\n\n검색 결과:\n{search_md}\n"
-
-            # Claude 스트리밍
-            with client.messages.stream(
-                model=st.session_state.ai_model, max_tokens=MAX_TOKENS,
-                system=sys_prompt,
-                messages=[{"role": m["role"], "content": m["content"]}
-                          for m in st.session_state.messages]
-            ) as stream:
-                for t in stream.text_stream:
-                    answer += t or ""
-                    placeholder.markdown(answer + "▌")
-            placeholder.markdown(answer)
-
-            # 이미지 옵션
-            answer_entry_saved = False
-            if st.session_state.generate_image:
-                with st.spinner("이미지 생성 중…"):
-                    ip = extract_image_prompt(answer, prompt)
-                    img, cap = generate_image(ip)
-                    if img:
-                        st.image(img, caption=cap)
-                        st.session_state.messages.append(
-                            {"role": "assistant", "content": answer,
-                             "image": img, "image_caption": cap})
-                        answer_entry_saved = True
-            if not answer_entry_saved:
-                st.session_state.messages.append(
-                    {"role": "assistant", "content": answer})
-
-            # 본문 다운로드 버튼 (MD / HTML)
-            st.subheader("이 블로그 다운로드")
-            b1, b2 = st.columns(2)
-            b1.download_button("마크다운", answer,
-                               file_name=f"{prompt[:30]}.md", mime="text/markdown")
-            b2.download_button("HTML", md_to_html(answer, prompt[:30]),
-                               file_name=f"{prompt[:30]}.html", mime="text/html")
-
-    # ── 자동 백업 저장
-    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"자동 저장 실패: {e}")
+
+def process_example(topic):
+    """예제 주제를 직접 처리하는 함수 (rerun 없이)"""
+    process_input(topic)
+
+
+def process_input(prompt):
+    """사용자 입력 처리 함수 (일반 입력과 예제 입력 모두 처리)"""
+    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(); answer = ""
+        
+        # 선택된 템플릿, 톤, 단어 수로 시스템 프롬프트 생성
+        sys_prompt = get_system_prompt(
+            template=st.session_state.blog_template,
+            tone=st.session_state.blog_tone,
+            word_count=st.session_state.word_count
+        )
+
+        if st.session_state.use_web_search:
+            with st.spinner("웹 검색 중…"):
+                search_md = do_web_search(keywords(prompt))
+                sys_prompt += f"\n\n검색 결과:\n{search_md}\n"
+
+        # Claude 스트리밍
+        with client.messages.stream(
+            model=st.session_state.ai_model, max_tokens=MAX_TOKENS,
+            system=sys_prompt,
+            messages=[{"role": m["role"], "content": m["content"]}
+                      for m in st.session_state.messages]
+        ) as stream:
+            for t in stream.text_stream:
+                answer += t or ""
+                placeholder.markdown(answer + "▌")
+        placeholder.markdown(answer)
+
+        # 이미지 옵션
+        answer_entry_saved = False
+        if st.session_state.generate_image:
+            with st.spinner("이미지 생성 중…"):
+                ip = extract_image_prompt(answer, prompt)
+                img, cap = generate_image(ip)
+                if img:
+                    st.image(img, caption=cap)
+                    st.session_state.messages.append(
+                        {"role": "assistant", "content": answer,
+                         "image": img, "image_caption": cap})
+                    answer_entry_saved = True
+        if not answer_entry_saved:
+            st.session_state.messages.append(
+                {"role": "assistant", "content": answer})
+
+        # 본문 다운로드 버튼 (MD / HTML)
+        st.subheader("이 블로그 다운로드")
+        b1, b2 = st.columns(2)
+        b1.download_button("마크다운", answer,
+                          file_name=f"{prompt[:30]}.md", mime="text/markdown")
+        b2.download_button("HTML", md_to_html(answer, prompt[:30]),
+                          file_name=f"{prompt[:30]}.html", mime="text/html")
+
+        # ── 자동 백업 저장
+        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"자동 저장 실패: {e}")
 
 # ──────────────────────────────── main / requirements ──────────────────────
 def main(): ginigen_app()