Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
 from openai import OpenAI
-import os
-import random
+from datasets import load_dataset, Dataset
+import pandas as pd
+import time, os, random
 
-API_KEY = os.getenv("API_KEY")
+# --- API設定 ---
+API_KEY = os.getenv("API_KEY")  # Hugging FaceではSpaces Secretsに設定
 BASE_URL = "https://openrouter.ai/api/v1"
+HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face Token（Secretsに設定）
+DATASET_REPO = "your-username/reading_test_logs"  # ← 自分のDataset名に変更
 
 # --- Lexile難易度別教材（各段階3つずつ） ---
 texts = {
@@ -43,7 +47,7 @@ client = OpenAI(
     api_key=API_KEY
 )
 
-# --- AIに問題生成を依頼 ---
+# --- 問題生成 ---
 def generate_question(text):
     prompt = f"""
     Read the following passage and create ONE multiple-choice question with 4 options (A–D).
@@ -54,11 +58,9 @@ def generate_question(text):
     B. <option>
     C. <option>
     D. <option>
-
     Passage:
     {text}
     """
-
     response = client.chat.completions.create(
         model="google/gemma-3-27b-it:free",
         messages=[{"role": "user", "content": prompt}],
@@ -67,22 +69,17 @@ def generate_question(text):
     )
     return response.choices[0].message.content.strip()
 
-# --- AIで正解を判定（別プロンプト） ---
+# --- AIで正誤判定 ---
 def check_answer_with_ai(text, question, user_answer):
     prompt = f"""
     Read the passage and question below. Decide if the user's answer is correct.
-
     Passage:
     {text}
-
     Question:
     {question}
-
     User Answer: {user_answer}
-
     Respond with only one word: "Correct" or "Incorrect".
     """
-
     response = client.chat.completions.create(
         model="google/gemma-3-27b-it:free",
         messages=[{"role": "user", "content": prompt}],
@@ -91,16 +88,38 @@ def check_answer_with_ai(text, question, user_answer):
     )
     return response.choices[0].message.content.strip().lower() == "correct"
 
-# --- 適応型テストの進行 ---
+# --- 適応型テスト ---
 def adaptive_test(prev_level, prev_correct):
     idx = levels.index(prev_level)
     if prev_correct and idx < len(levels) - 1:
-        new_level = levels[idx + 1]
+        return levels[idx + 1]
     elif not prev_correct and idx > 0:
-        new_level = levels[idx - 1]
-    else:
-        new_level = prev_level
-    return new_level
+        return levels[idx - 1]
+    return prev_level
+
+# --- ログ書き込み関数 ---
+def log_to_hf_dataset(user_id, level, passage, question, user_answer, correct, response_time):
+    log_entry = {
+        "user_id": [user_id],
+        "lexile_level": [level],
+        "passage": [passage],
+        "question": [question],
+        "user_answer": [user_answer],
+        "correct": [correct],
+        "response_time": [response_time],
+        "timestamp": [pd.Timestamp.now().strftime("%Y-%m-%d %H:%M:%S")],
+    }
+
+    new_df = pd.DataFrame(log_entry)
+    try:
+        dataset = load_dataset(DATASET_REPO, split="train", use_auth_token=HF_TOKEN)
+        existing_df = dataset.to_pandas()
+        updated_df = pd.concat([existing_df, new_df], ignore_index=True)
+    except Exception:
+        updated_df = new_df
+
+    updated_dataset = Dataset.from_pandas(updated_df)
+    updated_dataset.push_to_hub(DATASET_REPO, token=HF_TOKEN)
 
 # --- 状態管理 ---
 used_texts = set()
@@ -112,39 +131,51 @@ def start_test():
     global used_texts, question_count
     used_texts = set()
     question_count = 0
-
-    level = 850  # 中間レベルから開始
+    level = 850
     text = random.choice(texts[level])
     used_texts.add(text)
     question = generate_question(text)
-    return text, question, level, None, "", "", True  # True = show test
+    start_time = time.time()
+    return text, question, level, None, "", "", True, start_time
 
-# --- 回答処理して次へ ---
-def next_step(prev_level, user_answer, question_text, passage_text):
+# --- 回答処理 ---
+def next_step(prev_level, user_answer, question_text, passage_text, start_time):
     global question_count, used_texts
     question_count += 1
 
+    end_time = time.time()
+    response_time = round(end_time - start_time, 2)
+
     correct = check_answer_with_ai(passage_text, question_text, user_answer)
     new_level = adaptive_test(prev_level, correct)
+
+    # ログ保存
+    log_to_hf_dataset(
+        user_id="test_user",
+        level=prev_level,
+        passage=passage_text,
+        question=question_text,
+        user_answer=user_answer,
+        correct=correct,
+        response_time=response_time
+    )
+
     feedback = "✅ Correct!" if correct else "❌ Incorrect."
 
-    # --- 終了条件 ---
+    # 終了条件
     if question_count >= MAX_QUESTIONS:
-        feedback = f"🎯 **Test finished!**\n\nYour estimated reading level is **{new_level}L**."
-        feedback += "\n\nThank you for participating!"
-        # hide text, question, and answer area
-        return feedback, "", "", new_level, None, "", False
+        feedback = f"🎯 **Test finished!**\n\nYour estimated reading level is **{new_level}L**.\n\nThank you for participating!"
+        return feedback, "", "", new_level, None, "", False, 0.0
 
-    # --- 次の文章を選択（同じものを避ける） ---
+    # 次の文章を選ぶ
     available = [t for t in texts[new_level] if t not in used_texts]
     if not available:
         available = texts[new_level]
     next_text = random.choice(available)
     used_texts.add(next_text)
     next_question = generate_question(next_text)
-
-    feedback += f"\n➡️ Next question loading..."
-    return feedback, next_text, next_question, new_level, None, "", True
+    feedback += "\n➡️ Next question loading..."
+    return feedback, next_text, next_question, new_level, None, "", True, time.time()
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
@@ -152,19 +183,17 @@ with gr.Blocks() as demo:
 
     start_btn = gr.Button("▶️ Start Test")
 
-    # 可視領域
     text_display = gr.Textbox(label="Reading Passage", lines=6, interactive=False, visible=True)
     question_display = gr.Textbox(label="Generated Question", lines=8, interactive=False, visible=True)
     user_answer = gr.Radio(choices=["A", "B", "C", "D"], label="Your Answer", visible=True)
     submit_btn = gr.Button("Submit Answer", visible=True)
 
-    # フィードバック領域
     feedback_display = gr.Markdown()
     hidden_level = gr.Number(visible=False)
     hidden_passage = gr.Textbox(visible=False)
     test_visible = gr.State(True)
+    hidden_start_time = gr.Number(visible=False)
 
-    # --- Start Test ---
     start_btn.click(
         fn=start_test,
         inputs=[],
@@ -175,14 +204,14 @@ with gr.Blocks() as demo:
             user_answer,
             feedback_display,
             hidden_passage,
-            test_visible
+            test_visible,
+            hidden_start_time
         ]
     )
 
-    # --- Submit & Move to Next ---
     submit_btn.click(
         fn=next_step,
-        inputs=[hidden_level, user_answer, question_display, text_display],
+        inputs=[hidden_level, user_answer, question_display, text_display, hidden_start_time],
         outputs=[
             feedback_display,
             text_display,
@@ -190,14 +219,15 @@ with gr.Blocks() as demo:
             hidden_level,
             user_answer,
             hidden_passage,
-            test_visible
+            test_visible,
+            hidden_start_time
         ]
     )
 
-    # --- 表示制御 ---
     def toggle_visibility(show):
-        visible = bool(show)
-        return gr.update(visible=visible), gr.update(visible=visible), gr.update(visible=visible), gr.update(visible=visible)
+        v = bool(show)
+        return (gr.update(visible=v), gr.update(visible=v),
+                gr.update(visible=v), gr.update(visible=v))
 
     test_visible.change(
         fn=toggle_visibility,