add auto prediciton

app.py

@@ -61,6 +61,14 @@ def append_suggestion(current, choice):
     # 模擬輸入法候選選中
     return current + choice
 
+def suggest_on_change(text, model_name, k, m, auto_flag):
+    """
+    當輸入框內容改變時，如果自動預測開啟，則觸發建議生成；否則清空候選。
+    """
+    if auto_flag:
+        return suggest_next(text, model_name, k, m)
+    return update(choices=[], value=None)
+
 # 自訂 CSS：模擬經典中文輸入法候選欄樣式
 custom_css = """
 #suggestions-bar .candidate-list {
@@ -91,7 +99,8 @@ custom_css = """
 with gr.Blocks(css=custom_css) as demo:
     # 標題和說明
     gr.Markdown(
-        "## 🇹🇼 繁體中文輸入法加速器  \n"        
+        "## 🇹🇼 繁體中文輸入法加速器  
+"
         "結合小型語言模型與 ZeroGPU，即時 IME 風格候選條。"
     )
 
@@ -101,12 +110,13 @@ with gr.Blocks(css=custom_css) as demo:
         elem_id="suggestions-bar", elem_classes="candidate-list"
     )
 
-    # 輸入區與按鈕：單行輸入框 + 小按鈕
+    # 輸入區與按鈕：單行輸入框 + 小按鈕 + 自動預測開關
     with gr.Row():
         input_text = gr.Textbox(
             label="", placeholder="請輸入拼音或文字…", lines=1, max_lines=1
         )
         gpu_button = gr.Button("建議")
+        auto_predict = gr.Checkbox(value=True, label="自動預測 (輸入框內容變更時觸發)")
 
     # 進階參數設定（可折疊）
     with gr.Accordion("進階設定", open=False):
@@ -126,10 +136,15 @@ with gr.Blocks(css=custom_css) as demo:
         inputs=[input_text, model_selector, k_slider, m_slider],
         outputs=suggestions,
     )
+    input_text.change(
+        fn=suggest_on_change,
+        inputs=[input_text, model_selector, k_slider, m_slider, auto_predict],
+        outputs=suggestions,
+    )
     suggestions.change(
         fn=append_suggestion,
         inputs=[input_text, suggestions],
         outputs=input_text,
     )
 
-    demo.launch()
+    demo.launch()