Update app.py

app.py

@@ -33,19 +33,30 @@ pdf_text = extract_text_from_pdf()
 def generate_question(topic, difficulty):
     if not pdf_text.strip():
         return "⚠️ 無法載入教材內容，請確認 PDF 是否存在。"
-    prompt = f"請根據以下教育學教材內容，設計一個屬於「{topic}」主題、「{difficulty}」難度的考題：\n\n{pdf_text}"
+    
+    # 修改提示以生成選擇題或填空題
+    prompt = f"請根據以下教育學教材內容，設計一個屬於「{topic}」主題、「{difficulty}」難度的選擇題或填空題擇一：\n\n{pdf_text}"
+    
     try:
         response = openai.ChatCompletion.create(
             model="gpt-4o-mini-2024-07-18",
             messages=[
-                {"role": "system", "content": "你是一位教育專家，請根據教材內容設計問題。"},
+                {"role": "system", "content": "你是一位教育專家，請根據教材內容設計題目。(不需要包含解析)"},
                 {"role": "user", "content": prompt}
             ]
         )
         question = response["choices"][0]["message"]["content"]
-        
-        # 獲取參考答案
-        answer_prompt = f"請根據以下問題提供正確答案：\n問題：{question}\n\n教材內容：\n{pdf_text}"
+        return question.strip()  # 返回生成的問題
+    except Exception as e:
+        return f"⚠️ 發生錯誤：{e}"
+
+def save_answer(question):
+    if not pdf_text.strip():
+        return "⚠️ 教材內容未載入，請確認 PDF。"
+    
+    # 獲取參考答案
+    answer_prompt = f"請根據以下問題提供正確答案：\n問題：{question}\n\n教材內容：\n{pdf_text}"
+    try:
         answer_response = openai.ChatCompletion.create(
             model="gpt-4o-mini-2024-07-18",
             messages=[
@@ -54,10 +65,7 @@ def generate_question(topic, difficulty):
             ]
         )
         correct_answer = answer_response["choices"][0]["message"]["content"]
-        
-        # 儲存參考答案
         reference_answers[question] = correct_answer.strip()  # 儲存正確答案
-        return question
     except Exception as e:
         return f"⚠️ 發生錯誤：{e}"
 
@@ -74,10 +82,12 @@ def analyze_answer(user_input, question):
     global user_errors
     if not user_input.strip():
         return "⚠️ 請輸入回答。"
-    if not pdf_text.strip():
-        return "⚠️ 教材內容未載入，請確認 PDF。"
     
     correct_answer = reference_answers.get(question, "無法獲取正確答案")
+    
+    # 獲取參考答案
+    save_answer(question)  # 在分析之前儲存答案
+    
     prompt = f"請根據以下教材內容，檢查學生的回答是否正確，並提供正確答案與講解：\n{pdf_text}\n\n問題：{question}\n學生回答：'{user_input}'"
 
     try: