Update app.py

app.py

@@ -45,7 +45,6 @@ def load_prompt(level):
 def chat_process(message, display_history, full_logs, prompt_level):
     """
     チャット処理を行う関数
-    Stateの更新漏れを防ぐため、リストは + 演算子で新しく生成して返す
     """
     if display_history is None: display_history = []
     if full_logs is None: full_logs = []
@@ -58,7 +57,6 @@ def chat_process(message, display_history, full_logs, prompt_level):
         "timestamp": current_time_user
     }
     
-    # 【修正】appendではなく新しいリストを作成して結合 (State更新を確実にするため)
     temp_display = display_history + [user_msg_data]
     temp_logs = full_logs + [user_msg_data]
 
@@ -72,7 +70,6 @@ def chat_process(message, display_history, full_logs, prompt_level):
     ai_response = ""
     try:
         model_name = "google/gemini-2.5-flash"
-        # model_name = "google/gemma-3-27b-it:free"
 
         response = client.chat.completions.create(
             model=model_name,
@@ -98,11 +95,9 @@ def chat_process(message, display_history, full_logs, prompt_level):
         "timestamp": current_time_ai
     }
 
-    # 【修正】最終的な新しいリストを作成
     new_display = temp_display + [ai_msg_data]
     new_logs = temp_logs + [ai_msg_data]
 
-    # 戻り値: (入力欄クリア, Chatbot表示更新, 画面用State更新, 保存用State更新)
     return "", new_display, new_display, new_logs
 
 def change_difficulty_logic(new_level, full_logs):
@@ -124,20 +119,17 @@ def change_difficulty_logic(new_level, full_logs):
         "timestamp": current_time
     }
 
-    # 新しい画面用履歴（これまでの会話は消え、通知だけ入る）
     new_display_history = [msg_data]
-    
-    # 【修正】全保存用履歴には追記する（新しいリストを作成）
     new_full_logs = full_logs + [msg_data]
     
-    # 戻り値: (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}'")
 
     if not user_name or not user_name.strip():
-        gr.Warning("名前が入力されていません")
+        # Start画面でチェックしているので基本ここには来ないはずだが念のため
+        gr.Warning("名前が正しく取得できませんでした")
         return None
 
     if not full_logs:
@@ -168,6 +160,18 @@ def export_csv_logic(user_id, user_name, full_logs):
         gr.Warning(f"CSV保存エラー: {e}")
         return None
 
+# --- 新規追加: スタートボタン処理 ---
+def start_app_logic(user_name):
+    """
+    名前が入力されているか確認し、画面表示を切り替える
+    """
+    if not user_name or not user_name.strip():
+        raise gr.Error("お名前を入力してください。")
+    
+    # Start画面を隠し(False)、Main画面を表示(True)
+    return gr.update(visible=False), gr.update(visible=True)
+
+
 # --- UI Definition ---
 with gr.Blocks() as demo:
     # ステート定義
@@ -176,43 +180,64 @@ with gr.Blocks() as demo:
     full_logs_state = gr.State([])       # 保存用
     prompt_level_state = gr.State("Beginner")
 
-    gr.Markdown("# Multi-Level AI Chatbot")
-    
-    with gr.Row():
+    # =========================================
+    # 1. スタート画面 (初期状態: 表示)
+    # =========================================
+    with gr.Column(visible=True) as start_screen:
+        gr.Markdown("# AI Chatbot System")
+        gr.Markdown("学習を始めるには、お名前を入力して「解答開始」を押してください。")
+        
+        # 名前入力欄をここに移動
         user_name_input = gr.Textbox(
-            label="お名前（ファイル名に使用されます）", 
-            placeholder="ここに名前を入力してください", 
+            label="お名前", 
+            placeholder="山田 太郎", 
             scale=2
         )
-        export_btn = gr.DownloadButton("📥 会話をCSVで保存", scale=1)
+        
+        start_btn = gr.Button("解答開始", variant="primary", size="lg")
+
+    # =========================================
+    # 2. メインチャット画面 (初期状態: 非表示)
+    # =========================================
+    with gr.Column(visible=False) as main_screen:
+        gr.Markdown("# Multi-Level AI Chatbot")
+        
+        with gr.Row():
+            # エクスポートボタンのみここに配置（名前はstart_screenのものを使用）
+            export_btn = gr.DownloadButton("📥 会話をCSVで保存", scale=1)
 
-    with gr.Row():
-        gr.Markdown("## 問題の難易度を選択してください（初級から順番に取り組んでください）")
+        with gr.Row():
+            gr.Markdown("## 問題の難易度を選択してください（初級から順番に取り組んでください）")
+            
+        with gr.Row():
+            level_radio = gr.Radio(
+                ["Beginner", "Intermediate", "Advanced"],
+                label="Difficulty Level",
+                value="Beginner",
+                interactive=True
+            )
+
+        # Chatbotコンポーネント
+        chatbot = gr.Chatbot(type="messages", height=500)
         
-    with gr.Row():
-        level_radio = gr.Radio(
-            ["Beginner", "Intermediate", "Advanced"],
-            label="Difficulty Level",
-            value="Beginner",
-            interactive=True
+        # 入力エリア
+        msg_input = gr.Textbox(
+            placeholder="メッセージを入力してください... (Enterで送信)", 
+            label="Chat Input",
+            lines=1
         )
+        send_btn = gr.Button("送信", variant="primary")
 
-    # Chatbotコンポーネント
-    chatbot = gr.Chatbot(type="messages", height=500)
-    
-    # 入力エリア
-    msg_input = gr.Textbox(
-        placeholder="メッセージを入力してください... (Enterで送信)", 
-        label="Chat Input",
-        lines=1
-    )
-    send_btn = gr.Button("送信", variant="primary")
+    # --- Event Handling ---
 
-    # --- Event Handling (ここが重要) ---
+    # A. スタートボタンの処理 (新規)
+    start_btn.click(
+        fn=start_app_logic,
+        inputs=[user_name_input],
+        outputs=[start_screen, main_screen]
+    )
 
-    # 1. メッセージ送信時の処理
-    # 入力: [メッセージ, 画面履歴, 保存履歴, 難易度]
-    # 出力: [入力欄, チャットボット, 画面履歴State, 保存履歴State] ← 全て更新する
+    # B. メッセージ送信時の処理
     chat_event_args = {
         "fn": chat_process,
         "inputs": [msg_input, display_history_state, full_logs_state, prompt_level_state],
@@ -222,16 +247,15 @@ with gr.Blocks() as demo:
     msg_input.submit(**chat_event_args)
     send_btn.click(**chat_event_args)
 
-    # 2. 難易度変更時の処理
-    # 入力: [難易度, 保存履歴]
-    # 出力: [チャットボット, 画面履歴State, 保存履歴State, 難易度State] ← 全て更新する
+    # C. 難易度変更時の処理
     level_radio.change(
         fn=change_difficulty_logic,
         inputs=[level_radio, full_logs_state],
         outputs=[chatbot, display_history_state, full_logs_state, prompt_level_state]
     )
 
-    # 3. CSV保存処理
+    # D. CSV保存処理
+    # 注意: start_screenにある user_name_input の値を読み取ります（非表示でも値は残ります）
     export_btn.click(
         fn=export_csv_logic,
         inputs=[user_id_state, user_name_input, full_logs_state],