Update ui.py

ui.py

@@ -1,12 +1,6 @@
 import gradio as gr
 from sentiment import analyze_sentiment
-
-# 📌 可選擇的模型
-MODEL_OPTIONS = {
-    "🌎 多語言推特情緒分析 (XLM-RoBERTa)": "cardiffnlp/twitter-xlm-roberta-base-sentiment",
-    "📖 多語言情緒分析 (BERT)": "nlptown/bert-base-multilingual-uncased-sentiment",
-    "🇬🇧 英語情緒分析 (DistilBERT)": "distilbert-base-uncased-finetuned-sst-2-english"
-}
+from config import MODEL_OPTIONS
 
 # 📌 Gradio 介面說明
 intro_text = """
@@ -34,37 +28,39 @@ def create_ui():
     with gr.Blocks(theme=gr.themes.Soft()) as iface:
         gr.Markdown(intro_text)
 
-        text_input = gr.Textbox(lines=3, placeholder="請輸入文本...", label="輸入文本")
+        text_input = gr.Textbox(lines=3, placeholder="請輸入文本...", label="輸入文本", max_lines=5)
 
         model_selector = gr.Dropdown(choices=list(MODEL_OPTIONS.keys()), value="🌎 多語言推特情緒分析 (XLM-RoBERTa)", label="選擇 AI 模型")
 
         analyze_button = gr.Button("分析情緒")
+        clear_button = gr.Button("清除")
         progress_bar = gr.Textbox(value="", visible=False, label="模型載入進度")
 
         result_output = gr.Markdown(label="分析結果")
 
         # 📌 綁定按鈕功能
         def process_analysis(text, model_name):
-            model_id = MODEL_OPTIONS[model_name]
-
-            # **當使用者切換模型時，顯示進度條**
-            if model_id != analyze_sentiment.CURRENT_MODEL:
-                progress_bar.update("🔄 AI 模型載入中，請稍後...", visible=True)
-
-            # 📌 調用 API 進行分析
-            result, _ = analyze_sentiment(text, model_id)
-
-            # **確保進度條消失**
-            progress_bar.update("", visible=False)  
+            progress_bar.update(value="🔄 AI 模型載入中，請稍後...", visible=True)
+            result, _ = analyze_sentiment(text, model_name)
+            progress_bar.update(value="", visible=False)
             return result
 
-        # 📌 **修正 Gradio 事件綁定，確保所有輸入 & 輸出正確**
         analyze_button.click(
             fn=process_analysis,
             inputs=[text_input, model_selector],
-            outputs=[result_output, progress_bar]
+            outputs=[result_output]
+        )
+
+        clear_button.click(
+            lambda: ("", ""), 
+            inputs=[], 
+            outputs=[text_input, result_output]
         )
 
         gr.Markdown(developer_info)
 
     return iface
+
+if __name__ == "__main__":
+    ui = create_ui()
+    ui.launch()