Update app.py

app.py

@@ -45,13 +45,12 @@ def load_prompt(level):
 def chat_process(message, display_history, full_logs, prompt_level):
     """
     チャット処理を行う関数
-    display_history: 画面表示用の履歴 (Chatbotに渡す)
-    full_logs: 全保存用の履歴 (CSV保存用)
+    Stateの更新漏れを防ぐため、リストは + 演算子で新しく生成して返す
     """
     if display_history is None: display_history = []
     if full_logs is None: full_logs = []
 
-    # 1. ユーザーメッセージの記録
+    # 1. ユーザーメッセージの作成
     current_time_user = datetime.datetime.now(JST).strftime("%Y-%m-%d %H:%M:%S")
     user_msg_data = {
         "role": "user", 
@@ -59,16 +58,15 @@ def chat_process(message, display_history, full_logs, prompt_level):
         "timestamp": current_time_user
     }
     
-    # 画面用と保存用の両方に追加
-    display_history.append(user_msg_data)
-    full_logs.append(user_msg_data)
+    # 【修正】appendではなく新しいリストを作成して結合 (State更新を確実にするため)
+    temp_display = display_history + [user_msg_data]
+    temp_logs = full_logs + [user_msg_data]
 
-    # APIリクエスト用のメッセージ作成（現在の画面の文脈を使用）
+    # APIリクエスト用のメッセージ作成
     system_prompt = load_prompt(prompt_level)
     messages_for_api = [{"role": "system", "content": system_prompt}]
     
-    for msg in display_history:
-        # roleとcontentだけ抽出してAPIに渡す
+    for msg in temp_display:
         messages_for_api.append({"role": msg["role"], "content": msg["content"]})
 
     ai_response = ""
@@ -92,7 +90,7 @@ def chat_process(message, display_history, full_logs, prompt_level):
         print(error_msg, file=sys.stderr)
         ai_response = f"システムエラーが発生しました: {e}"
 
-    # 2. AIメッセージの記録
+    # 2. AIメッセージの作成
     current_time_ai = datetime.datetime.now(JST).strftime("%Y-%m-%d %H:%M:%S")
     ai_msg_data = {
         "role": "assistant", 
@@ -100,21 +98,19 @@ def chat_process(message, display_history, full_logs, prompt_level):
         "timestamp": current_time_ai
     }
 
-    display_history.append(ai_msg_data)
-    full_logs.append(ai_msg_data)
+    # 【修正】最終的な新しいリストを作成
+    new_display = temp_display + [ai_msg_data]
+    new_logs = temp_logs + [ai_msg_data]
 
-    # 入力欄を空にする("")、更新されたchatbot用履歴、更新された保存用履歴 を返す
-    return "", display_history, full_logs
+    # 戻り値: (入力欄クリア, Chatbot表示更新, 画面用State更新, 保存用State更新)
+    return "", new_display, new_display, new_logs
 
 def change_difficulty_logic(new_level, full_logs):
     """
     難易度変更時の処理
-    - 画面用履歴(display_history)はリセットして新しい通知を入れる
-    - 全保存用履歴(full_logs)は維持して通知を追記する
     """
     if full_logs is None: full_logs = []
 
-    # 通知メッセージ作成
     notification_msg = (
         f"**[システム通知]**\n"
         f"難易度を **{new_level}** に変更しました。\n"
@@ -128,13 +124,14 @@ def change_difficulty_logic(new_level, full_logs):
         "timestamp": current_time
     }
 
-    # 新しい画面用履歴（これまでの会話は消える）
+    # 新しい画面用履歴（これまでの会話は消え、通知だけ入る）
     new_display_history = [msg_data]
     
-    # 全保存用履歴には追記する（これまでの会話は残る）
-    full_logs.append(msg_data)
+    # 【修正】全保存用履歴には追記する（新しいリストを作成）
+    new_full_logs = full_logs + [msg_data]
     
-    return new_display_history, full_logs, new_level
+    # 戻り値: (Chatbot表示更新, 画面用State更新, 保存用State更新, 難易度State更新)
+    return new_display_history, new_display_history, new_full_logs, new_level
 
 def export_csv_logic(user_id, user_name, full_logs):
     print(f"Exporting CSV for UserID: {user_id}, Name: '{user_name}'")
@@ -173,10 +170,10 @@ def export_csv_logic(user_id, user_name, full_logs):
 
 # --- UI Definition ---
 with gr.Blocks() as demo:
-    # ステート管理（グローバル変数ではなく、ここで個別にデータを保持する）
+    # ステート定義
     user_id_state = gr.State(generate_user_id)
-    display_history_state = gr.State([]) # 画面表示用のリスト
-    full_logs_state = gr.State([])       # CSV保存用の全リスト
+    display_history_state = gr.State([]) # 画面用
+    full_logs_state = gr.State([])       # 保存用
     prompt_level_state = gr.State("Beginner")
 
     gr.Markdown("# Multi-Level AI Chatbot")
@@ -200,50 +197,38 @@ with gr.Blocks() as demo:
             interactive=True
         )
 
-    # ChatInterfaceの代わりに、ChatbotとTextboxを配置して手動制御する
+    # Chatbotコンポーネント
     chatbot = gr.Chatbot(type="messages", height=500)
+    
+    # 入力エリア
     msg_input = gr.Textbox(
         placeholder="メッセージを入力してください... (Enterで送信)", 
         label="Chat Input",
         lines=1
     )
-    # 送信ボタン（非表示にしてEnter送信だけでも良いが、あると便利）
     send_btn = gr.Button("送信", variant="primary")
 
-    # --- Event Handling ---
+    # --- Event Handling (ここが重要) ---
 
     # 1. メッセージ送信時の処理
-    # msg_inputのsubmit(Enterキー)と、send_btnのclickの両方で発火
-    msg_input.submit(
-        fn=chat_process,
-        inputs=[msg_input, display_history_state, full_logs_state, prompt_level_state],
-        outputs=[msg_input, chatbot, full_logs_state] # 戻り値でStateを更新する
-    ).then(
-        fn=lambda x: x, # Stateの中身をchatbotに反映させるための繋ぎ
-        inputs=[display_history_state],
-        outputs=[chatbot]
-    )
+    # 入力: [メッセージ, 画面履歴, 保存履歴, 難易度]
+    # 出力: [入力欄, チャットボット, 画面履歴State, 保存履歴State] ← 全て更新する
+    chat_event_args = {
+        "fn": chat_process,
+        "inputs": [msg_input, display_history_state, full_logs_state, prompt_level_state],
+        "outputs": [msg_input, chatbot, display_history_state, full_logs_state]
+    }
 
-    send_btn.click(
-        fn=chat_process,
-        inputs=[msg_input, display_history_state, full_logs_state, prompt_level_state],
-        outputs=[msg_input, chatbot, full_logs_state]
-    ).then(
-        fn=lambda x: x,
-        inputs=[display_history_state],
-        outputs=[chatbot]
-    )
+    msg_input.submit(**chat_event_args)
+    send_btn.click(**chat_event_args)
 
     # 2. 難易度変更時の処理
+    # 入力: [難易度, 保存履歴]
+    # 出力: [チャットボット, 画面履歴State, 保存履歴State, 難易度State] ← 全て更新する
     level_radio.change(
         fn=change_difficulty_logic,
         inputs=[level_radio, full_logs_state],
-        outputs=[chatbot, full_logs_state, prompt_level_state] # chatbotには新しい画面履歴が入る
-    ).then(
-        # 画面履歴Stateも更新しておく
-        fn=lambda history: history,
-        inputs=[chatbot],
-        outputs=[display_history_state]
+        outputs=[chatbot, display_history_state, full_logs_state, prompt_level_state]
     )
 
     # 3. CSV保存処理