Update app.py

app.py

@@ -15,6 +15,35 @@ output_dir = "./question_bank"
 # 載入題庫字典(question_bank_dict)，格式為 { "年級_學期": [題庫檔名列表] }
 question_bank_dict = json.load(open(f"{output_dir}/question_bank_dict.json", "r"))
 
+print(question_bank_dict)
+
+def random_questions_with_limit(data, limit=20000):
+    """
+    隨機從 data 中挑選題目，並將總字串長度限制在 limit 字元數內（至少不小於 5000）。
+    會傳回：
+    - result_list：篩選後的題目列表
+    - result_str：將篩選後的題目以兩行空白分隔串接的字串
+    - count：所選題卷的數量
+    """
+    # 確保 limit 不小於 5000
+    limit = max(limit, 5000)
+
+    # 將題目列表隨機洗牌
+    random.shuffle(data)
+
+    result_list = []
+    current_length = 0
+    count = 0
+
+    for item in data:
+        # 如果加入下一個題目後長度不超過 limit，則加入結果列表
+        if current_length + len(item) <= limit:
+            result_list.append(item)
+            current_length += len(item)
+            count += 1
+
+    return result_list, "\n\n".join(result_list), count
+
 def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     """
     根據年級、學期以及指定的題型(qtype)和題目數量(num_questions)，
@@ -33,17 +62,14 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     grade_semester = f"{grades[grade]}年級_{terms[term]}學期"
 
     # 根據題庫字典從指定年級學期中取得所有檔名，並嘗試讀取其對應的 markdown 檔案內容
-    # [完成程式區塊]
-    raw_questions = []# question_bank_dict.get(grade_semester, [])
+    raw_questions = [
+        open(f"{output_dir}/md/{doc_path.replace('.pdf', '.md')}").read()
+        for doc_path in question_bank_dict[grade_semester]
+        if os.path.exists(f"{output_dir}/md/{doc_path.replace('.pdf', '.md')}")
+    ]
 
-    for filename in question_bank_dict.get(grade_semester, []):
-        filename = filename.replace('.pdf', '.md')
-        if not os.path.exists(f"{output_dir}/md/{filename}"): continue
-            
-        raw_questions.append(open(f"{output_dir}/md/{filename}").read())
+    print(raw_questions)
 
-    print(len(raw_questions))
-    
     # 隨機選取並限制字串長度的題庫內容
     input_question_bank = random_questions_with_limit(raw_questions, 20000)
 
@@ -98,7 +124,6 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     ]
 
     # 使用 InferenceClient 呼叫 API 模型產生新題目
-    # 請自行挑選使用哪一個開源大語言模型，並自行調整模型輸入參數，看看會產生什麼結果
     completion = client.chat.completions.create(
         model="mistralai/Mistral-Nemo-Instruct-2407",
         messages=messages,
@@ -108,33 +133,6 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     # 傳回模型產生的文本
     return completion.choices[0].message.content
 
-def random_questions_with_limit(data, limit=20000):
-    """
-    隨機從 data 中挑選題目，並將總字串長度限制在 limit 字元數內（至少不小於 5000）。
-    會傳回：
-    - result_list：篩選後的題目列表
-    - result_str：將篩選後的題目以兩行空白分隔串接的字串
-    - count：所選題卷的數量
-    """
-    # 確保 limit 不小於 5000
-    limit = max(limit, 5000)
-
-    # 將題目列表隨機洗牌
-    random.shuffle(data)
-
-    result_list = []
-    current_length = 0
-    count = 0
-
-    for item in data:
-        # 如果加入下一個題目後長度不超過 limit，則加入結果列表
-        if current_length + len(item) <= limit:
-            result_list.append(item)
-            current_length += len(item)
-            count += 1
-
-    return result_list, "\n\n".join(result_list), count
-
 # 建立 Gradio 介面
 with gr.Blocks() as app:
     # 介面標題區