Update app.py

app.py

@@ -12,7 +12,7 @@ difficulties = ["簡單", "中等", "困難"]
 
 user_errors = {}
 error_history = {}
-reference_answers = {}  # 儲存參考答案
+reference_answers = {}
 
 DEFAULT_PDF_PATH = "教材.pdf"
 
@@ -43,15 +43,13 @@ def generate_question(topic, difficulty):
             ]
         )
         question = response["choices"][0]["message"]["content"]
-        
         # 儲存參考答案
-        reference_answers[topic] = question  # 假設問題即為參考答案，根據實際情況調整
-        
-        return question  # 返回生成的問題
+        reference_answers[question] = "這是參考答案。"  # 這裡可以根據實際情況獲取正確答案
+        return question
     except Exception as e:
         return f"⚠️ 發生錯誤：{e}"
 
-def analyze_answer(user_input, topic):
+def analyze_answer(user_input, question):
     global user_errors, error_history
     if not user_input.strip():
         return "⚠️ 請輸入回答。"
@@ -59,6 +57,7 @@ def analyze_answer(user_input, topic):
         return "⚠️ 教材內容未載入，請確認 PDF。"
     
     current_date = datetime.today().strftime("%Y-%m-%d")
+    correct_answer = reference_answers.get(question, "無法獲取正確答案")
     prompt = f"請根據以下教材內容，檢查學生的回答是否正確，並提供正確答案與講解：\n{pdf_text}\n\n學生回答：'{user_input}'"
 
     try:
@@ -73,17 +72,15 @@ def analyze_answer(user_input, topic):
     except Exception as e:
         return f"⚠️ 發生錯誤：{e}"
 
-    # 檢查用戶的回答是否正確
-    correct_answer = reference_answers.get(topic, "無參考答案")
     if "❌" in feedback or "錯" in feedback:
-        user_errors[topic] = user_errors.get(topic, 0) + 1
+        user_errors[question] = user_errors.get(question, 0) + 1
         error_history.setdefault(current_date, []).append({
-            "題目": topic,
+            "題目": question,
             "回答": user_input,
-            "AI 分析": feedback,
-            "參考答案": correct_answer  # 儲存參考答案
+            "正確答案": correct_answer,
+            "AI 分析": feedback
         })
-    return f"{feedback}\n\n參考答案：{correct_answer}"  # 這裡可以選擇不返回參考答案
+    return feedback
 
 def get_errors_by_date_safe(date_str):
     try:
@@ -93,19 +90,10 @@ def get_errors_by_date_safe(date_str):
     errors = error_history.get(date_str)
     if not errors:
         return "✅ 該日無錯題紀錄。"
-    return "\n\n".join([f"🔹 題目: {e['題目']}\n📝 回答: {e['回答']}\n📖 AI 分析: {e['AI 分析']}\n📜 參考答案: {e['參考答案']}" for e in errors])
+    return "\n\n".join([f"🔹 題目: {e['題目']}\n📝 回答: {e['回答']}\n📖 正確答案: {e['正確答案']}\n📖 AI 分析: {e['AI 分析']}" for e in errors])
 
-def clear_all():
-    # 清空問題、回答、參考答案與錯題歷史，但保留PDF文本
-    global reference_answers, user_errors, error_history
-    reference_answers = {}
-    user_errors = {}
-    error_history.clear()
-    return "", "", "", get_errors_text()
-
-def clear_question_and_answer():
-    # 清空生成的問題和用戶的回答
-    return "", ""
+def clear_fields():
+    return "", "", "", ""
 
 with gr.Blocks() as demo:
     gr.Markdown("# 👨‍🏫 教師檢定智慧陪讀家教 🚀")
@@ -113,29 +101,28 @@ with gr.Blocks() as demo:
     with gr.Row():
         topic_input = gr.Dropdown(choices=topics, label="選擇複習主題")
         difficulty_input = gr.Dropdown(choices=difficulties, label="選擇難度等級")
-    with gr.Row():
-        ask_btn = gr.Button("🎯 生成問題")
-        clear_question_btn = gr.Button("❌ 清空問題")
+    ask_btn = gr.Button("🎯 生成問題")
+    clear_btn = gr.Button("🧹 清空")
     question_output = gr.Textbox(label="AI 生成的問題", lines=4)
-    
+    ask_btn.click(fn=lambda t, d: generate_question(t, d),
+                  inputs=[topic_input, difficulty_input],
+                  outputs=question_output)
+    clear_btn.click(fn=clear_fields, outputs=[question_output, user_answer, analysis_result])
+
     user_answer = gr.Textbox(label="你的回答", lines=3)
     analyze_btn = gr.Button("📊 分析回答")
     analysis_result = gr.Textbox(label="AI 分析與講解", lines=5)
-    
+    analyze_btn.click(fn=lambda ans, q: analyze_answer(ans, q),
+                      inputs=[user_answer, question_output],
+                      outputs=analysis_result)
+
     gr.Markdown("---")
-    gr.Markdown("📋 錯題自動記錄")
-    error_history_output = gr.Textbox(label="錯題紀錄", lines=10)
-    
-    # 生成問題按鈕事件
-    ask_btn.click(fn=lambda t, d: generate_question(t, d),
-                  inputs=[topic_input, difficulty_input],
-                  outputs=[question_output])
-    # 清空問題按鈕事件
-    clear_question_btn.click(fn=clear_question_and_answer,
-                             inputs=[],
-                             outputs=[question_output, user_answer])
-    # 分析回答按鈕事件
-    analyze_btn.click(fn=analyze_answer,
-                      inputs=[user_answer, topic_input],
-                      outputs=[analysis_result, error_history_output])
+    gr.Markdown("📅 查詢錯題紀錄")
+    date_input = gr.Textbox(label="輸入日期（YYYY-MM-DD）")
+    search_errors_btn = gr.Button("🔍 查看該日期錯題")
+    error_history_output = gr.Textbox(label="錯題紀錄", lines=5)
+    search_errors_btn.click(fn=get_errors_by_date_safe,
+                            inputs=date_input,
+                            outputs=error_history_output)
+
 demo.launch()