Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import openai
+import PyPDF2
 import os
 
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
@@ -11,13 +12,27 @@ difficulties = ["簡單", "中等", "困難"]
 # 學習者錯誤統計
 user_errors = {}
 
-# AI 生成問題函數（完全由 AI 創建）
+# **開發者預設教材 PDF 檔案**
+DEFAULT_PDF_PATH = "教材.pdf"  # 確保此檔案存放在專案目錄中
+
+# 解析 PDF 並擷取文本（使用開發者預設的教材）
+def extract_text_from_pdf():
+    with open(DEFAULT_PDF_PATH, "rb") as pdf_file:
+        reader = PyPDF2.PdfReader(pdf_file)
+        text = ""
+        for page in reader.pages:
+            text += page.extract_text() + "\n"
+    return text
+
+pdf_text = extract_text_from_pdf()  # 讀取教材
+
+# AI 生成問題函數（基於預設教材）
 def generate_question(topic, difficulty):
-    prompt = f"請設計一個有關'{topic}'的考題，難度為 '{difficulty}'，適用於教師檢定。"
+    prompt = f"請根據以下教育心理學教材內容，設計一個'{difficulty}'難度的考題：\n{pdf_text}"
     
     response = openai.ChatCompletion.create(
         model="gpt-4",
-        messages=[{"role": "system", "content": "你是一位教育心理學專家，請根據主題與難度提供合適的考題。"},
+        messages=[{"role": "system", "content": "你是一位教育心理學專家，請根據教材內容生成考題。"},
                   {"role": "user", "content": prompt}]
     )
     return response['choices'][0]['message']['content']