Update app.py

app.py

@@ -6,13 +6,21 @@ import json
 from datasets import Dataset, DatasetDict, load_dataset, load_from_disk
 from huggingface_hub import HfApi, HfFolder
 
-HEA_API_TOKEN = os.environ.get("HF_API_TOKEN")
+# 从环境变量中获取 Hugging Face API 令牌和其他配置
+HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
 LLM_API = os.environ.get("LLM_API")
 LLM_URL = os.environ.get("LLM_URL")
 USER_ID = "HuggingFace Space"
-HfFolder.save_token(HEA_API_TOKEN)
 DATASET_NAME = os.environ.get("DATASET_NAME")
 
+# 确保令牌不为空
+if HF_API_TOKEN is None:
+    raise ValueError("HF_API_TOKEN 环境变量未设置。请在 Hugging Face Space 的设置中添加该环境变量。")
+
+# 设置 Hugging Face API 令牌
+HfFolder.save_token(HF_API_TOKEN)
+
+# 加载或创建数据集
 try:
     dataset = load_dataset(DATASET_NAME)
 except:
@@ -88,11 +96,6 @@ def save_feedback(user_input, response, feedback_type, improvement):
     dataset.push_to_hub(DATASET_NAME)
 
 def handle_feedback(response, feedback_type, improvement):
-    feedback = {
-        "response": response,
-        "feedback_type": feedback_type,
-        "improvement": improvement
-    }
     save_feedback(response, feedback_type, improvement)
     return "Thank you for your feedback!"