Update app.py

app.py

@@ -209,12 +209,13 @@ footer { display: none !important; }
 # ------------------------------------------------------------------
 # 5. App (최종 수정: State 사용 + API 생성 차단)
 # ------------------------------------------------------------------
+# 5. App (최종 수정: State 사용 + API 생성 차단 + 즉시 반응 패치)
+# ------------------------------------------------------------------
 with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as demo:
-    gr.Markdown("# ≪ 하 이 루 ≫")
+    gr.Markdown("# ≪ 어 서 오 세 요 ≫")
     gr.Markdown(">> 01410 접속 성공... [대화실]에 입장하셨습니다.")
 
     # [중요] 채팅 기록을 저장할 '기억 저장소' (화면엔 안 보임)
-    # Chatbot 컴포넌트를 직접 입력으로 쓰지 않기 위해 사용
     history_state = gr.State([])
 
     chatbot = gr.Chatbot(show_label=False, elem_classes="chatbot")
@@ -236,62 +237,63 @@ with gr.Blocks(theme=gr.themes.Base(), css=PC_COM_CSS, title="CHOLLIAN 98") as d
         btn3 = gr.Button("오늘 기분 꿀꿀하네..", elem_classes="example-btn")
         btn4 = gr.Button("야 밥 뭐먹지 추천좀", elem_classes="example-btn")
 
-    # [핵심 수정] 채팅창(chatbot)을 입력으로 쓰지 않고, state를 입력으로 사용
-    def predict(user_input, history):
+    # [수정 1] 사용자 입력을 화면에 '즉시' 띄우는 함수
+    def user(user_input, history):
         history = history or []
+        # 봇 응답 자리에 None을 채워서 일단 화면에 표시 (즉시 반응)
+        new_history = history + [[user_input, None]]
+        return "", new_history, new_history
+
+    # [수정 2] 봇의 응답을 생성해서 업데이트하는 함수
+    def bot(history):
+        if not history:
+            return history, history
+            
+        user_input = history[-1][0]
+        
+        # LLM에 보낼 히스토리 정리
         hist_pairs = []
-        for u, b in history:
-            if u is None or b is None:
-                continue
+        for u, b in history[:-1]:
+            if u is None or b is None: continue
             hist_pairs.append((u, b))
 
+        # 답변 생성 (시간이 좀 걸림)
         bot_out = chat_response(user_input, hist_pairs)
         
-        # 새 히스토리
-        new_history = list(history) + [[user_input, bot_out]]
-        
-        # 리턴: 입력창 비우기, State 업데이트, Chatbot 업데이트
-        return "", new_history, new_history
+        # 마지막 None 자리에 답변 채워넣기
+        history[-1][1] = bot_out
+        return history, history
 
-    # [핵심 수정] api_name=False로 설정하여 스키마 생성 에러 원천 차단
+    # [수정 3] 이벤트 연결 (user 실행 -> bot 실행 순서로 체이닝)
+    # 이렇게 해야 엔터 치자마자 내 글이 먼저 뜨고, 똘배가 고민하는 게 보임
     msg.submit(
-        predict, 
-        [msg, history_state], 
-        [msg, history_state, chatbot], 
-        queue=False, 
-        api_name=False
+        user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+    ).then(
+        bot, [history_state], [history_state, chatbot], queue=False, api_name=False
     )
+
     submit_btn.click(
-        predict, 
-        [msg, history_state], 
-        [msg, history_state, chatbot], 
-        queue=False, 
-        api_name=False
+        user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+    ).then(
+        bot, [history_state], [history_state, chatbot], queue=False, api_name=False
     )
     
-    # 초기화: State와 Chatbot 둘 다 비움
+    # 초기화
     clear.click(
-        lambda: ([], []), 
-        None, 
-        [history_state, chatbot], 
-        queue=False, 
-        api_name=False
+        lambda: ([], []), None, [history_state, chatbot], queue=False, api_name=False
     )
 
-    # 예시 버튼 연결
+    # 예시 버튼 연결 (버튼 클릭 -> 텍스트 입력 -> 유저 함수 -> 봇 함수)
     for btn, text in [
         (btn1, "하이 방가방가"), 
         (btn2, "밸런스게임 ㄱㄱ"), 
         (btn3, "오늘 기분 꿀꿀하네.."), 
         (btn4, "야 밥 뭐먹지 추천좀")
     ]:
-        # 버튼 클릭 시 텍스트 박스에 채우고 -> 바로 전송
         btn.click(lambda: text, None, msg, queue=False, api_name=False).then(
-            predict, 
-            [msg, history_state], 
-            [msg, history_state, chatbot], 
-            queue=False, 
-            api_name=False
+            user, [msg, history_state], [msg, history_state, chatbot], queue=False, api_name=False
+        ).then(
+            bot, [history_state], [history_state, chatbot], queue=False, api_name=False
         )
 
 if __name__ == "__main__":