Update sentiment.py

sentiment.py

@@ -1,25 +1,26 @@
 import requests
 import logging
-from config import HEADERS
+from config import HEADERS, MODEL_OPTIONS, DEFAULT_MODEL
 
 # 設定 logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
-CURRENT_MODEL = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
+CURRENT_MODEL = DEFAULT_MODEL
 API_URL = f"https://api-inference.huggingface.co/models/{CURRENT_MODEL}"
 
-def analyze_sentiment(text, model_id=None):
+# 📌 呼叫 Hugging Face API 進行情緒分析
+def analyze_sentiment(text, model_name=None):
     global CURRENT_MODEL, API_URL
 
-    if model_id and model_id != CURRENT_MODEL:
-        CURRENT_MODEL = model_id
+    if model_name and MODEL_OPTIONS[model_name] != CURRENT_MODEL:
+        CURRENT_MODEL = MODEL_OPTIONS[model_name]
         API_URL = f"https://api-inference.huggingface.co/models/{CURRENT_MODEL}"
-        logging.info(f" 重新載入模型: {CURRENT_MODEL}")
+        logging.info(f"🔄 切換模型: {CURRENT_MODEL}")
 
     try:
-        logging.info(" 發送 API 請求...")
+        logging.info("🚀 發送 API 請求...")
         response = requests.post(API_URL, headers=HEADERS, json={"inputs": text})
-        response.raise_for_status()  # 檢查 HTTP 狀態碼
+        response.raise_for_status()
         result = response.json()
         logging.info(f"✅ API 回應: {result}")
 
@@ -41,4 +42,4 @@ def analyze_sentiment(text, model_id=None):
         return f"❌ **字典鍵錯誤**: {str(e)}", 0.0
     except Exception as e:
         logging.error(f"❌ 未知錯誤: {e}")
-        return f"❌ **未知錯誤**: {str(e)}", 0.0
+        return f"❌ **未知錯誤**: {str(e)}", 0.0