Update chat_a.py

chat_a.py

@@ -11,6 +11,18 @@ from sentence_transformers import SentenceTransformer, util
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from css import log_and_render
 
+from css import log_and_render
+import time # time 라이브러리를 import 합니다.
+
+def stream_text_generator(text_to_stream: str):
+    """
+    문자열을 단어 단위로 쪼개서 하나씩 흘려보내주는 생성기(generator) 함수.
+    타이핑 효과를 위해 사용됩니다.
+    """
+    for word in text_to_stream.split(" "):
+        yield word + " "
+        time.sleep(0.05) # 각 단어 사이에 0.05초의 딜레이를 줍니다.
+        
 # ──────────────────────────────── 캐시/데이터셋 설정 ────────────────────────────────
 HOME = pathlib.Path.home()
 
@@ -1340,14 +1352,31 @@ def recommend_packages(
         )
 
     # ✅ 출력
+    # if recommend_texts:
+    #     full_message = "🧳 이런 패키지를 추천드려요:<br><br>" + "<br><br>".join(recommend_texts)
+    #     log_and_render(
+    #         full_message,
+    #         sender="bot",
+    #         chat_container = chat_container,
+    #         key="recommend_package_intro",
+    # )
     if recommend_texts:
-        full_message = "🧳 이런 패키지를 추천드려요:<br><br>" + "<br><br>".join(recommend_texts)
-        log_and_render(
-            full_message,
-            sender="bot",
-            chat_container = chat_container,
-            key="recommend_package_intro",
-    )
+        # ⬇️⬇️⬇️ 여기가 핵심 변경 부분입니다 ⬇️⬇️⬇️
+        intro_message = "🧳 이런 패키지를 추천드려요:"
+        package_list_message = "".join(recommend_texts)
+
+        # 1. 안내 문구는 타이핑 효과로 먼저 보여줍니다.
+        with st.chat_message("assistant"):
+            st.write_stream(stream_text_generator(intro_message))
+        
+        # 2. 패키지 목록은 타이핑 없이 바로 보여주는 것이 더 깔끔합니다.
+        st.markdown(package_list_message) # st.markdown을 사용해 바로 출력
+
+        # 3. 대화 기록에는 전체 메시지를 합쳐서 저장합니다.
+        full_message = intro_message + package_list_message
+        if "chat_log" not in st.session_state:
+            st.session_state.chat_log = []
+        st.session_state.chat_log.append(("bot", full_message))
     else:
         log_and_render(
             "⚠️ 추천 가능한 패키지가 없습니다.",
@@ -1378,6 +1407,21 @@ def handle_selected_place(selected_place, travel_df, external_score_df, festival
         message_lines.append(f"함께 가보면 좋은 여행지: {sample_names}")
     else:
         message_lines.append("⚠️ 함께 가볼 다른 여행지가 없어요.")
+
+    ##
+    full_message = "".join(message_lines)
+
+    # Streamlit의 chat_message 컨텍스트를 사용해 말풍선을 직접 만듭니다.
+    with st.chat_message("assistant"): # 'assistant'는 봇 메시지를 의미합니다.
+        # 우리가 만든 제너레이터와 st.write_stream을 연결해 타이핑 효과를 냅니다.
+        st.write_stream(stream_text_generator(full_message))
+    
+    # 대화 기록(log)을 남기는 로직은 화면 출력 뒤에 따로 실행합니다.
+    if "chat_log" not in st.session_state:
+        st.session_state.chat_log = []
+    st.session_state.chat_log.append(("bot", full_message))
+    ##
+    
     # integrated_theme 추론 추가
     if selected_theme is None:
         theme_row = travel_df[travel_df["여행지"] == selected_place]