Update app.py

app.py

@@ -248,90 +248,41 @@ def _llm_structured_extract(user_text: str):
 # ──────────────────────────────── Streamlit용 LLM 모드 UI ────────────────────────────────
 def render_llm_followup(chat_container, inline=False):
     _ensure_llm_state()
-    MAX_TURNS = 6
 
-    # ── 인라인: 구분선 먼저, 그 다음 안내 버블, 그 다음 입력창 ──
-    if inline:
-        st.divider()
-        if st.session_state.pop("llm_intro_needed", False):
-            log_and_render(
-                "🧠 이제부터는 자유 질문 모드예요. 여행 외에도 뭐든 한국어로 물어보세요!",
-                sender="bot",
-                chat_container=chat_container,
-                key=f"llm_intro_{random.randint(1,999999)}"
-            )
-        st.markdown("#### 🤖 LLM 질문")
-    else:
-        # 풀스크린 모드에서는 안내 버블 먼저
-        if st.session_state.pop("llm_intro_needed", False):
-            log_and_render(
-                "🧠 이제부터는 자유 질문 모드예요. 여행 외에도 뭐든 한국어로 물어보세요!",
-                sender="bot",
-                chat_container=chat_container,
-                key=f"llm_intro_{random.randint(1,999999)}"
-            )
-        st.markdown("### 🤖 LLM 질문")
+    st.markdown("### ◎ LLM 질문")
 
-    # 입력창 (항상 여기서만 노출)
-    q = st.text_input(
-        "LLM 질문",
-        placeholder="무엇이든 물어보세요 (종료하려면 '종료' 입력)",
-        key="llm_input"
-    )
+    # 기존 대화 렌더
+    for m in st.session_state.get("llm_msgs", []):
+        with st.chat_message(m["role"]):
+            st.markdown(m["content"])
 
-    if q:
-        if q.strip() in ("종료", "quit", "exit"):
-            if inline:
-                # ✅ 인라인 종료(버블은 유지, 새로고침 없음)
-                st.session_state["llm_inline"] = False
-                st.session_state["llm_input"] = ""
-                log_and_render(
-                    "LLM 모드를 종료할게요. 필요하실 때 다시 질문해 주세요! ✨",
-                    sender="bot", chat_container=chat_container,
-                    key=f"llm_end_inline_{random.randint(1,999999)}"
-                )
-                return
-            else:
-                # 풀스크린 종료
-                st.session_state["llm_mode"] = False
-                st.session_state["llm_input"] = ""
-                log_and_render(
-                    "LLM 모드를 종료할게요. 필요하실 때 다시 질문해 주세요! ✨",
-                    sender="bot", chat_container=chat_container,
-                    key=f"llm_end_full_{random.randint(1,999999)}"
-                )
-                st.rerun()
-                return
+    # ⚠️ 위젯 key는 'llm_query'로 사용 (이 key에 대해 어디서도 직접 대입 금지)
+    user_msg = st.chat_input("무엇이든 물어보세요 (종료하려면 '종료' 입력)", key="llm_query")
+    if not user_msg:
+        return
 
-        # 일반 질의 처리
-        log_and_render(q, sender="user", chat_container=chat_container,
-                       key=f"llm_user_{random.randint(1,999999)}")
-        st.session_state.llm_history.append({"role": "user", "content": q})
+    text = user_msg.strip()
 
-        msgs = st.session_state.llm_history[-(MAX_TURNS-1):]
-        a = _call_ollama_chat(
-            messages=msgs,
-            system_prompt=KOREAN_SYSTEM_PROMPT,
-            temperature=0.8, top_p=0.9, top_k=40, repeat_penalty=1.1
-        )
-        if not a:
-            log_and_render("⚠️ LLM 응답을 받지 못했습니다. Ollama 서버를 확인해 주세요.",
-                           sender="bot", chat_container=chat_container,
-                           key=f"llm_err_{random.randint(1,999999)}")
-        else:
-            log_and_render(a, sender="bot", chat_container=chat_container,
-                           key=f"llm_bot_{random.randint(1,999999)}")
-            st.session_state.llm_history.append({"role": "assistant", "content": a})
-        st.session_state["llm_input"] = ""
-
-    # 하단 버튼: 인라인은 'LLM 패널 종료'만, 풀스크린은 'LLM 모드 종료'만
-    if inline:
-        if st.button("🔚 LLM 모드 종료", key="llm_close_inline"):
-            st.session_state["llm_inline"] = False
-    else:
-        if st.button("🔚 LLM 모드 종료", key="llm_close_full"):
-            st.session_state["llm_mode"] = False
-            st.rerun()
+    # 종료 명령 처리
+    if text in {"종료", "quit", "exit"}:
+        st.session_state["llm_inline"] = False
+        st.session_state["llm_open"] = False
+        st.rerun()
+        return
+
+    # 대화 저장
+    st.session_state.setdefault("llm_msgs", [])
+    st.session_state["llm_msgs"].append({"role": "user", "content": text})
+
+    # LLM 호출 (예외 안전)
+    try:
+        bot = call_llm(text)  # 기존 함수 사용
+    except Exception as e:
+        bot = "⚠️ LLM 응답을 받지 못했습니다. Ollama 서버를 확인해 주세요."
+    st.session_state["llm_msgs"].append({"role": "assistant", "content": bot})
+
+    # ⛔ 여기서 st.session_state['llm_query']나 'llm_input'을 직접 대입/삭제하지 마세요.
+    st.rerun()
 
 def render_llm_inline_if_open(chat_container):
     """llm_inline 플래그가 켜져 있으면 인라인 LLM 패널을 그립니다."""