Update vocab.py

vocab.py

@@ -1,11 +1,12 @@
 import sqlite3
 import os
+import json
 
 DATA_DIR = "./data"
 DB_PATH = os.path.join(DATA_DIR, "sentences.db")
 
 
-# 初始化資料表（保險起見，但你本地應該已建過了）
+# 初始化資料表
 def init_db():
     conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
@@ -24,7 +25,7 @@ def init_db():
     conn.close()
 
 
-# 查詢某個單字的所有例句，包含來源 & 模型
+# 查某個單字的所有例句
 def get_sentences_by_word(word):
     conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
@@ -34,7 +35,7 @@ def get_sentences_by_word(word):
     return results
 
 
-# 儲存句子（GPT 生成 or 句庫）
+# 儲存句子到 SQLite
 def save_sentence(word, phonetic, sentence, source, model):
     conn = sqlite3.connect(DB_PATH)
     c = conn.cursor()
@@ -45,3 +46,18 @@ def save_sentence(word, phonetic, sentence, source, model):
     ''', (word, phonetic, sentence, source, model))
     conn.commit()
     conn.close()
+
+
+# 掃描 /data 內的 JSON 檔，回傳單字庫名稱清單
+def get_sources():
+    files = os.listdir(DATA_DIR)
+    sources = [f.split(".json")[0] for f in files if f.endswith(".json")]
+    return sources
+
+
+# 根據單字庫名稱，回傳該單字庫所有單字
+def get_words_from_source(source):
+    data_path = os.path.join(DATA_DIR, f"{source}.json")
+    with open(data_path, 'r', encoding='utf-8') as f:
+        words = json.load(f)
+    return [w['word'] for w in words]