Update app.py

app.py

@@ -1,37 +1,122 @@
 import gradio as gr
-from vocab import get_sources, get_sentences_by_word, get_words_from_source
+from vocab import get_sources, get_words_from_source, get_word_info
+from sentences import get_sentences_by_word, get_words_with_sentences, save_sentence
 from ai_sentence import generate_sentence, MODEL_LIST
 
+
+# 查詢例句模式 - 處理函數
 def fetch_sentence(word, source, use_ai, model_name):
-    valid_words = get_words_from_source(source)
+    status_log = []
+    result_display = ""
+
+    # 1️⃣ 檢查單字是否存在於該單字庫
+    words_data = get_words_from_source(source)
+    valid_words = {w['word'] for w in words_data}
     if word not in valid_words:
         return f"<p style='color:red;'>❌ 單字 {word} 不在 {source} 單字庫中</p>", "檢查失敗"
 
+    # 2️⃣ 查詢單字音標（用於顯示 & AI 生成時存入資料庫）
+    word_info = get_word_info(source, word)
+    phonetic = word_info['phonetic'] if word_info else "無"
+
+    # 3️⃣ 查詢句庫
     sentence_records = get_sentences_by_word(word)
-    result_display = ""
-    status_log = []
 
     if not use_ai:
+        # 不使用 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>"
+            result_display += f"""
+            <div style="margin-bottom: 10px; padding: 8px; border-left: 4px solid #4CAF50; background-color: #f9f9f9;">
+                <strong>單字：</strong> {word} <br>
+                <strong>音標：</strong> {phonetic or '無'} <br>
+                <strong>句子：</strong> {sentence} <br>
+                <strong>來源：</strong> {src} {f'({model})' if model else ''}
+            </div>
+            """
         if not sentence_records:
-            result_display = f"<p style='color:red;'>❌ 查無句子</p>"
+            result_display = f"<p style='color:red;'>❌ 查無例句：{word}</p>"
+
     else:
+        # 使用 AI 生成
+        status_log.append(f"🤖 使用 AI 模型 {model_name} 生成句子中...")
         try:
             sentence = generate_sentence(word, model_name)
-            result_display = f"<p>🤖 AI 生成: {sentence} (模型: {model_name})</p>"
+            result_display = f"""
+            <div style="margin-bottom: 10px; padding: 8px; border-left: 4px solid #2196F3; background-color: #f9f9f9;">
+                <strong>單字：</strong> {word} <br>
+                <strong>音標：</strong> {phonetic} <br>
+                <strong>句子：</strong> {sentence} <br>
+                <strong>來源：</strong> AI 生成 ({model_name})
+            </div>
+            """
+
+            # 若句庫沒這個單字，才存入資料庫
+            if not sentence_records:
+                save_sentence(word, phonetic, sentence, 'ai', model_name)
+                status_log.append(f"✅ AI 生成句子已儲存：{word} ({model_name})")
+            else:
+                status_log.append(f"🔄 AI 生成句子，但句庫已有紀錄，未覆蓋：{word}")
+
         except Exception as e:
-            result_display = f"<p style='color:red;'>❌ AI 生成失敗：{str(e)}</p>"
+            result_display = f"<p style='color:red;'>❌ AI 生成句子失敗：{str(e)}</p>"
+            status_log.append(f"❌ 生成失敗：{str(e)}")
+
+    return result_display, "\n".join(status_log)
+
+
+# 模式切換邏輯
+def toggle_mode(mode):
+    return {
+        "query": (gr.update(visible=True), gr.update(visible=False)),
+        "random": (gr.update(visible=False), gr.update(visible=True)),
+    }[mode]
 
-    return result_display, "完成"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 📖 VocabLine 單字例句查詢 & 生成平台")
-    gr.Markdown("開發者：余彦志 (大宇 ian) | Email: [email protected] | [GitHub](https://github.com/dayuian)")
+    gr.Markdown(
+        """
+        # 📖 VocabLine 單字例句查詢 & 生成平台
+
+        ## 📝 專案簡介
+        VocabLine 是一個單字例句工具，提供查詢句庫例句、AI 生成句子功能。  
+        可作為 LINE 單字推播、自學工具、英文教學輔助等用途。
 
-    mode = gr.Radio(["查詢模式", "隨機抽單字模式"], value="查詢模式")
+        ## ⚙️ 主要功能
+        - 支援從多個單字庫選擇。
+        - 可查詢單字句庫例句或使用 GPT AI 生成例句。
+        - 隨機抽取單字，生成例句。
 
+        ## 🗂️ 資料來源
+        - 單字庫：common3000 等
+        - 句庫來源：Tatoeba 例句、AI 生成句子（GPT）
+
+        ## 🛠️ 技術架構
+        - Gradio Blocks 前端介面
+        - Hugging Face Transformers 語言模型（GPT）
+        - SQLite 句庫
+
+        ## 👨‍💻 開發者資訊
+        - 開發者：余彦志 (大宇 ian)
+        - 信箱：[email protected]
+        - GitHub：[https://github.com/dayuian](https://github.com/dayuian)
+
+        ## 📲 LINE 推播（未來開發）
+        - 本工具將結合 LINE，透過 Make.com 每日推播單字和例句，敬請期待！
+        """
+    )
+
+    # 模式選擇
+    mode_radio = gr.Radio(
+        ["query", "random"],
+        label="選擇模式",
+        choices=["query", "random"],
+        value="query",
+        interactive=True,
+        info="query：單字查詢模式 | random：隨機抽單字模式"
+    )
+
+    # 查詢模式區塊
     with gr.Group(visible=True) as query_mode:
         source_dropdown_query = gr.Dropdown(choices=get_sources(), value="common3000", label="選擇單字庫")
         word_input = gr.Textbox(label="輸入單字")
@@ -41,14 +126,41 @@ with gr.Blocks() as demo:
         query_status = gr.Textbox()
         query_button = gr.Button("查詢例句")
 
-    def toggle_model(use_ai):
-        return gr.update(visible=use_ai)
+    # 隨機抽單字模式區塊
+    with gr.Group(visible=False) as random_mode:
+        source_dropdown_random = gr.Dropdown(choices=get_sources(), value="common3000", label="選擇單字庫")
+        num_input = gr.Number(value=5, label="抽幾個單字")
 
-    use_ai_checkbox.change(toggle_model, inputs=use_ai_checkbox, outputs=model_dropdown)
+        random_result_output = gr.HTML()
+        random_status_output = gr.Textbox()
+        submit_btn = gr.Button("生成例句")
+
+    # 查詢模式互動
     query_button.click(
         fetch_sentence,
         inputs=[word_input, source_dropdown_query, use_ai_checkbox, model_dropdown],
-        outputs=[query_result, query_status]
+        outputs=[query_result, query_status],
+    )
+
+    # 隨機抽單字模式互動
+    submit_btn.click(
+        fn=get_words_with_sentences,
+        inputs=[source_dropdown_random, num_input],
+        outputs=[random_result_output, random_status_output]
+    )
+
+    # 模式切換邏輯
+    mode_radio.change(
+        toggle_mode,
+        inputs=[mode_radio],
+        outputs=[query_mode, random_mode],
+    )
+
+    # 切換 AI 模型下拉選單可見性
+    use_ai_checkbox.change(
+        lambda use_ai: gr.update(visible=use_ai),
+        inputs=[use_ai_checkbox],
+        outputs=[model_dropdown]
     )
 
 demo.launch()