Update app.py

app.py

@@ -3,7 +3,7 @@ import random
 from vocab import get_sources, get_words_from_source
 from sentences import generate_sentences
 from ai_sentence import MODEL_LIST
-from quiz import generate_fill_in_blank_exam, check_exam
+from quiz import generate_fill_in_blank_exam, check_exam, render_exam_interface
 
 def process_sentence(mode, word, source, num, use_ai, model_name):
     try:
@@ -36,6 +36,7 @@ def project_description():
     - 查詢單字 → 獲取例句
     - 抽取單字 → 批量獲取例句
     - 可選 AI 生成句子（模型：GPT2 / Pythia）
+    - 英文小考 → 單字填空選擇題，自動批改計分
     ## 🧑‍💻 技術架構
     - Gradio Blocks + Transformers (Hugging Face)
     - SQLite 句庫管理
@@ -114,12 +115,19 @@ with gr.Blocks(css="""
         )
 
     with gr.Tab("英文小考"):
+        model_note = gr.Markdown(visible=True)
+
         quiz_source_dropdown = gr.Dropdown(
             choices=get_sources(),
             value="common3000",
             label="選擇單字庫"
         )
         quiz_num_slider = gr.Slider(minimum=1, maximum=5, value=2, step=1, label="題目數量")
+        quiz_model_dropdown = gr.Dropdown(
+            choices=MODEL_LIST,
+            value="gpt2",
+            label="選擇 AI 模型"
+        )
 
         quiz_generate_button = gr.Button("📄 生成試卷")
         quiz_submit_button = gr.Button("✅ 提交試卷")
@@ -127,21 +135,20 @@ with gr.Blocks(css="""
         quiz_score_display = gr.HTML()
         quiz_questions_state = gr.State([])
 
-        radios = []
-        for i in range(5):
-            radios.append(
-                gr.Radio(choices=[], label=f"第 {i + 1} 題", visible=False, interactive=True)
-            )
+        radios = [gr.Radio(choices=[], label=f"第 {i + 1} 題", visible=False, interactive=True) for i in range(5)]
+
+        def update_model_note(model):
+            return f"本功能目前使用 **{model}** 作為語言模型生成題目。"
+
+        quiz_model_dropdown.change(update_model_note, inputs=[quiz_model_dropdown], outputs=[model_note])
 
-        def display_exam(source, num):
+        def display_exam(source, num, model):
             questions = generate_fill_in_blank_exam(source, num)
             quiz_questions_state.value = questions
 
             updates = []
             for i, q in enumerate(questions):
-                updates.append(
-                    gr.update(choices=q["options"], label=f"第 {i + 1} 題：{q['sentence']}", visible=True)
-                )
+                updates.append(gr.update(choices=q["options"], label=f"第 {i + 1} 題：{q['sentence']}", visible=True))
             for i in range(len(questions), 5):
                 updates.append(gr.update(visible=False))
 
@@ -152,16 +159,7 @@ with gr.Blocks(css="""
             score_html = check_exam(user_answers[:len(questions)], questions)
             return score_html
 
-        quiz_generate_button.click(
-            display_exam,
-            inputs=[quiz_source_dropdown, quiz_num_slider],
-            outputs=radios
-        )
-
-        quiz_submit_button.click(
-            submit_exam,
-            inputs=radios,
-            outputs=quiz_score_display
-        )
+        quiz_generate_button.click(display_exam, inputs=[quiz_source_dropdown, quiz_num_slider, quiz_model_dropdown], outputs=radios)
+        quiz_submit_button.click(submit_exam, inputs=radios, outputs=quiz_score_display)
 
     demo.launch()