Update app.py

app.py

@@ -6,21 +6,21 @@ import json
 from datasets import Dataset, DatasetDict, load_dataset, load_from_disk
 from huggingface_hub import HfApi, HfFolder
 
-# 从环境变量中获取 Hugging Face API 令牌和其他配置
+# 從環境變量中獲取 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"
 DATASET_NAME = os.environ.get("DATASET_NAME")
 
-# 确保令牌不为空
+# 確保令牌不為空
 if HF_API_TOKEN is None:
-    raise ValueError("HF_API_TOKEN 环境变量未设置。请在 Hugging Face Space 的设置中添加该环境变量。")
+    raise ValueError("HF_API_TOKEN 環境變量未設置。請在 Hugging Face Space 的設置中添加該環境變量。")
 
-# 设置 Hugging Face API 令牌
+# 設置 Hugging Face API 令牌
 HfFolder.save_token(HF_API_TOKEN)
 
-# 加载或创建数据集
+# 加載或創建數據集
 try:
     dataset = load_dataset(DATASET_NAME)
 except:
@@ -96,14 +96,18 @@ def save_feedback(user_input, response, feedback_type, improvement):
     dataset.push_to_hub(DATASET_NAME)
 
 def handle_feedback(response, feedback_type, improvement):
-    save_feedback(response, feedback_type, improvement)
-    return "Thank you for your feedback!"
+    # 獲取最新的用戶輸入（假設用戶輸入保存在全局變量中）
+    global last_user_input
+    save_feedback(last_user_input, response, feedback_type, improvement)
+    return "感謝您的反饋！"
 
 def handle_user_input(user_input):
     print(f"User input: {user_input}")
+    global last_user_input
+    last_user_input = user_input  # 保存最新的用戶輸入
     return run_sync(user_input)
 
-# 读取并显示反馈内容的函数
+# 讀取並顯示反饋內容的函數
 def show_feedback():
     try:
         feedbacks = dataset["feedback"].to_pandas().to_dict(orient="records")
@@ -154,4 +158,10 @@ with iface:
 
     show_feedback_button.click(fn=show_feedback, outputs=feedback_display)
 
-iface.launch()
+# 添加示例
+examples = [
+    ["AlCoCrFeNi HEA coating 可用怎樣的實驗方法做到 ?"],
+    ["請問high entropy nitride coatings的形成，主要可透過那些元素來熱這個材料形成熱穩定?"]
+]
+
+iface.launch(examples=examples)