Update app.py

app.py

@@ -65,14 +65,13 @@ with gr.Blocks(css="""
                     interactive=True
                 )
 
-            with gr.Group(elem_id="card-group"):
-                word_input = gr.Textbox(label="輸入單字", visible=True)
-                num_input = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="抽取單字數量")
-                source_dropdown = gr.Dropdown(
-                    choices=get_sources(),
-                    value="common3000",
-                    label="選擇單字庫"
-                )
+            word_input = gr.Textbox(label="輸入單字", visible=True)
+            num_input = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="抽取單字數量", visible=False)
+            source_dropdown = gr.Dropdown(
+                choices=get_sources(),
+                value="common3000",
+                label="選擇單字庫"
+            )
 
         with gr.Group(elem_id="card-group"):
             use_ai_checkbox = gr.Checkbox(label="使用 AI 生成句子（較慢，約 30 秒）", elem_id="use-ai-checkbox")
@@ -99,6 +98,21 @@ with gr.Blocks(css="""
         with gr.Row():
             generate_button = gr.Button("✨ 生成句子", elem_id="generate-button")
 
+        def switch_mode(mode):
+            return gr.update(visible=(mode == '查詢單字')), gr.update(visible=(mode == '隨機抽單字'))
+
+        mode_radio.change(
+            switch_mode,
+            inputs=[mode_radio],
+            outputs=[word_input, num_input]
+        )
+
+        generate_button.click(
+            process_sentence,
+            inputs=[mode_radio, word_input, source_dropdown, num_input, use_ai_checkbox, model_dropdown],
+            outputs=[result_output, status_output]
+        )
+
     with gr.Tab("英文小考"):
         quiz_source_dropdown = gr.Dropdown(
             choices=get_sources(),
@@ -119,16 +133,7 @@ with gr.Blocks(css="""
             questions = generate_fill_in_blank_exam(source, num)
             quiz_questions_state.value = questions
 
-            radios = []
-            for i, q in enumerate(questions):
-                radios.append(
-                    gr.Radio(
-                        choices=q['options'],
-                        label=f"第 {i + 1} 題：{q['sentence']}",
-                        interactive=True
-                    )
-                )
-
+            radios = render_exam_interface(questions)
             return radios
 
         def submit_exam(*user_answers):
@@ -148,5 +153,4 @@ with gr.Blocks(css="""
             outputs=quiz_score_display
         )
 
-
     demo.launch()