adds feedback-dataset

app.py

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
 import csv
 from datetime import datetime
 import json
+import huggingface_hub
 
 # Load environment variables from .env file if it exists
 load_dotenv()
@@ -35,8 +36,9 @@ FEEDBACK_FILE = "feedback.csv"
 FEEDBACK_HEADERS = ["timestamp", "session_id", "message", "response", "rating", "comment"]
 
 def save_feedback(session_id, message, response, rating, comment):
-    """Save feedback to CSV file."""
+    """Save feedback to CSV file and upload to Hugging Face."""
     try:
+        # Save to local CSV file
         file_exists = os.path.isfile(FEEDBACK_FILE)
         with open(FEEDBACK_FILE, 'a', newline='') as f:
             writer = csv.DictWriter(f, fieldnames=FEEDBACK_HEADERS)
@@ -50,6 +52,20 @@ def save_feedback(session_id, message, response, rating, comment):
                 "rating": rating,
                 "comment": comment
             })
+        
+        # Upload to Hugging Face if token is available
+        hf_token = os.getenv("HF_TOKEN")
+        if hf_token:
+            try:
+                api = huggingface_hub.HfApi(token=hf_token)
+                api.upload_file(
+                    path_or_fileobj=FEEDBACK_FILE,
+                    path_in_repo=FEEDBACK_FILE,
+                    repo_id=os.getenv("HF_REPO_ID", "your-username/your-repo"),
+                    repo_type="dataset"
+                )
+            except Exception as e:
+                print(f"Warning: Failed to upload feedback to Hugging Face: {e}")
     except Exception as e:
         print(f"Error saving feedback: {e}")