Update app.py

app.py

@@ -6,11 +6,11 @@ from ai_sentence import MODEL_LIST
 
 def process_sentence(mode, word, source, num, use_ai, model_name):
     try:
-        if mode == 'query':
+        if mode == '查詢單字':
             if not word:
                 return "<p style='color:red;'>❌ 請輸入單字</p>", "未輸入單字"
             words = [word.strip()]
-        elif mode == 'random':
+        elif mode == '隨機抽單字':
             num = int(num)
             if num <= 0:
                 return "<p style='color:red;'>❌ 抽取數量須大於0</p>", "數量錯誤"
@@ -55,17 +55,16 @@ def project_description():
 with gr.Blocks(css="""
     #card-group { padding: 15px; border-radius: 12px; background-color: rgba(255, 255, 255, 0.05); box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1); margin-bottom: 15px; }
     .gradio-container { max-width: 800px; margin: auto; }
-    #generate-button.ai-active { background-color: #ff9800 !important; color: white !important; }
 """) as demo:
     gr.Markdown(project_description())
 
     with gr.Group():
         with gr.Row():
             mode_radio = gr.Radio(
-                ["query", "random"],
+                ["查詢單字", "隨機抽單字"],
                 label="選擇模式",
-                choices=["query", "random"],
-                value="query",
+                choices=["查詢單字", "隨機抽單字"],
+                value="查詢單字",
                 interactive=True
             )
 
@@ -79,14 +78,14 @@ with gr.Blocks(css="""
             )
 
         with gr.Group(elem_id="card-group"):
-            use_ai_checkbox = gr.Checkbox(label="使用 AI 生成句子", elem_id="use-ai-checkbox")
+            use_ai_checkbox = gr.Checkbox(label="使用 AI 生成句子（較慢，約 30 秒）", elem_id="use-ai-checkbox")
             model_dropdown = gr.Dropdown(
                 choices=MODEL_LIST,
                 value=MODEL_LIST[0],
                 label="選擇 AI 模型",
                 visible=False
             )
-            ai_warning = gr.Markdown("⚠️ 若使用 AI 生成句子，生成過程可能需要 30 秒以上，請耐心等待。", visible=False)
+            ai_warning = gr.Textbox(value="⚠️ 若使用 AI 生成句子，生成過程可能需要 30 秒以上，請耐心等待。", visible=False, interactive=False, label="")
 
         generate_button = gr.Button("✨ 生成句子", elem_id="generate-button")
 
@@ -94,7 +93,7 @@ with gr.Blocks(css="""
         status_output = gr.Textbox(label="處理狀態", interactive=False)
 
     def switch_mode(mode):
-        if mode == 'query':
+        if mode == '查詢單字':
             return gr.update(visible=True), gr.update(visible=False)
         else:
             return gr.update(visible=False), gr.update(visible=True)
@@ -108,23 +107,12 @@ with gr.Blocks(css="""
     def toggle_ai_button(use_ai):
         model_visibility = gr.update(visible=use_ai)
         warning_visibility = gr.update(visible=use_ai)
-        js_toggle_class = f"""
-        function toggleButtonClass() {{
-            const btn = document.getElementById('generate-button');
-            if ({'true' if use_ai else 'false'}) {{
-                btn.classList.add('ai-active');
-            }} else {{
-                btn.classList.remove('ai-active');
-            }}
-        }}
-        toggleButtonClass();
-        """
-        return model_visibility, gr.update(_js=js_toggle_class), warning_visibility
+        return model_visibility, warning_visibility
 
     use_ai_checkbox.change(
         toggle_ai_button,
         inputs=[use_ai_checkbox],
-        outputs=[model_dropdown, generate_button, ai_warning]
+        outputs=[model_dropdown, ai_warning]
     )
 
     generate_button.click(
@@ -133,4 +121,4 @@ with gr.Blocks(css="""
         outputs=[result_output, status_output]
     )
 
-demo.launch()
+demo.launch()