Update app.py

app.py

@@ -1,59 +1,54 @@
 import gradio as gr
-from vocab import get_sources, get_words_with_sentences
+from vocab import get_sources, get_sentences_by_word, get_words_from_source
+from ai_sentence import generate_sentence, MODEL_LIST
+
+def fetch_sentence(word, source, use_ai, model_name):
+    valid_words = get_words_from_source(source)
+    if word not in valid_words:
+        return f"<p style='color:red;'>❌ 單字 {word} 不在 {source} 單字庫中</p>", "檢查失敗"
+
+    sentence_records = get_sentences_by_word(word)
+    result_display = ""
+    status_log = []
+
+    if not use_ai:
+        for record in sentence_records:
+            _, phonetic, sentence, src, model = record
+            result_display += f"<p>📖 {word} [{phonetic or '無'}]: {sentence} (來源: {src}{f' ({model})' if model else ''})</p>"
+        if not sentence_records:
+            result_display = f"<p style='color:red;'>❌ 查無句子</p>"
+    else:
+        try:
+            sentence = generate_sentence(word, model_name)
+            result_display = f"<p>🤖 AI 生成: {sentence} (模型: {model_name})</p>"
+        except Exception as e:
+            result_display = f"<p style='color:red;'>❌ AI 生成失敗：{str(e)}</p>"
+
+    return result_display, "完成"
 
 with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        # 📖 英文單字隨機抽取 & GPT 例句生成 API
-
-        ## 📝 專案簡介
-        本專案製作一個 API 服務，能夠從不同的單字庫隨機抽取單字，並使用開源語言模型自動生成例句。  
-        未來適合作為 LINE 單字推播、自學工具、英文教學輔助等用途。
-        因為現在是免費模型，所以生成句子都很鳥，生成一次需要約一分鐘的時間，主要是測試用。
-
-        ## ⚙️ 主要功能
-        - 支援從多個單字庫中選擇，如：`common3000`, `business_words`。
-        - 可自訂抽取單字數量。
-        - 每個單字自動生成一個簡單的英文例句，適合初學者學習。
-
-        ## 📚 使用方式
-        1. 選擇單字庫（例如：`common3000`）。
-        2. 設定抽取單字數量（例如：10個）。
-        3. 點擊「生成例句」按鈕，即可獲得單字 + 音標 + 例句。
-
-        ## 🗂️ 資料來源
-        - **common3000**：常用3000單字表，附音標。
-        - 未來可能會有更多單字庫新增至 `/data/` 資料夾。（可能啦！
-
-        ## 🛠️ 技術架構
-        - **Gradio Blocks** 前端介面 + API。
-        - **Hugging Face Transformers** 語言模型：
-            - 模型：`EleutherAI/pythia-410m`（小型 GPT 模型）
-
-        ## 👨‍💻 開發者資訊
-        - 開發者：余彦志 (大宇 ian)
-        - 信箱：[email protected]
-        - GitHub：[https://github.com/dayuian](https://github.com/dayuian)
-        """
-    )
-
-    source_dropdown = gr.Dropdown(
-        choices=get_sources(),
-        value="common3000",
-        label="選擇單字庫",
-        interactive=True
-    )
-
-    num_input = gr.Number(value=10, label="抽幾個單字")
-    result_output = gr.HTML(label="生成結果")
-    status_output = gr.Textbox(label="狀態更新", lines=8, interactive=False)
-
-    submit_btn = gr.Button("生成例句")
-
-    submit_btn.click(
-        fn=get_words_with_sentences,
-        inputs=[source_dropdown, num_input],
-        outputs=[result_output, status_output]
+    gr.Markdown("# 📖 VocabLine 單字例句查詢 & 生成平台")
+    gr.Markdown("開發者：余彦志 (大宇 ian) | Email: [email protected] | [GitHub](https://github.com/dayuian)")
+
+    mode = gr.Radio(["查詢模式", "隨機抽單字模式"], value="查詢模式")
+
+    with gr.Group(visible=True) as query_mode:
+        source_dropdown_query = gr.Dropdown(choices=get_sources(), value="common3000", label="選擇單字庫")
+        word_input = gr.Textbox(label="輸入單字")
+        use_ai_checkbox = gr.Checkbox(label="使用 AI 生成句子")
+        model_dropdown = gr.Dropdown(choices=MODEL_LIST, value=MODEL_LIST[0], label="選擇 AI 模型", visible=False)
+        query_result = gr.HTML()
+        query_status = gr.Textbox()
+        query_button = gr.Button("查詢例句")
+
+    def toggle_model(use_ai):
+        return gr.update(visible=use_ai)
+
+    use_ai_checkbox.change(toggle_model, inputs=use_ai_checkbox, outputs=model_dropdown)
+    query_button.click(
+        fetch_sentence,
+        inputs=[word_input, source_dropdown_query, use_ai_checkbox, model_dropdown],
+        outputs=[query_result, query_status]
     )
 
 demo.launch()