Update app.py

app.py

@@ -20,11 +20,26 @@ if HF_API_TOKEN is None:
 # 設置 Hugging Face API 令牌
 HfFolder.save_token(HF_API_TOKEN)
 
+# 定義數據集特徵
+features = {
+    "user_input": "string",
+    "response": "string",
+    "feedback_type": "string",
+    "improvement": "string"
+}
+
 # 加載或創建數據集
 try:
     dataset = load_dataset(DATASET_NAME)
 except:
-    dataset = DatasetDict({"feedback": Dataset.from_dict({"user_input": [], "response": [], "feedback_type": [], "improvement": []})})
+    dataset = DatasetDict({
+        "feedback": Dataset.from_dict({
+            "user_input": [],
+            "response": [],
+            "feedback_type": [],
+            "improvement": []
+        })
+    })
 
 async def send_chat_message(user_input):
     payload = {
@@ -90,9 +105,14 @@ def save_feedback(user_input, response, feedback_type, improvement):
     }
     print(f"Saving feedback: {feedback}")
     # Append to the dataset
-    new_data = {"user_input": user_input, "response": response, "feedback_type": feedback_type, "improvement": improvement}
+    new_data = {
+        "user_input": [user_input],
+        "response": [response],
+        "feedback_type": [feedback_type],
+        "improvement": [improvement]
+    }
     global dataset
-    dataset = dataset["feedback"].add_item(new_data)
+    dataset["feedback"] = dataset["feedback"].add_item(new_data)
     dataset.push_to_hub(DATASET_NAME)
 
 def handle_feedback(response, feedback_type, improvement):
@@ -165,5 +185,4 @@ with gr.Blocks() as iface:
 
     show_feedback_button.click(fn=show_feedback, outputs=feedback_display)
 
-
 iface.launch()