Update app.py

app.py

@@ -20,23 +20,45 @@ import mysql.connector
 import urllib.parse
 import urllib.request
 
-# For Prompt Engineering 
-# import requests
-# from huggingface_hub import AsyncInferenceClient
 
 # Save chat history as JSON
 import atexit
+import os
+from huggingface_hub import HfApi, HfFolder
+
+# Define dataset repository URL and ID
+DATASET_REPO_URL = "https://huggingface.co/datasets/botsi/trust-game-llama-2-7b-chat"
+DATASET_REPO_ID = "botsi/trust-game-llama-2-7b-chat"
+
+# Define data file information
+DATA_FILENAME = "history_trust-game-llama-2-7b-chat.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+
+# Get Hugging Face token from environment variable
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+# Check if the data file exists
+if os.path.exists(DATA_FILE):
+    # Initialize Hugging Face API
+    api = HfApi()
+
+    # Upload file to dataset repository
+    with open(DATA_FILE, "rb") as f:
+        dataset_files = HfFolder.upload(
+            folder_or_file=f,
+            path_in_repo=DATA_FILENAME,
+            repo_id=DATASET_REPO_ID,
+            token=HF_TOKEN
+        )
+
+    # Print uploaded file information
+    print("Uploaded file to dataset repository:")
+    for file_info in dataset_files:
+        print(f"File path in repo: {file_info.path}")
+        print(f"File ID: {file_info.id}")
+else:
+    print("Data file does not exist.")
 
-# Add this global variable to store the chat history
-# global_chat_history = []
-# Add this function to store the chat history
-#def save_chat_history():
-#    """Save the chat history to a JSON file."""
-#    with open("chat_history.json", "w") as json_file:
-#        json.dump(global_chat_history, json_file)
-
-#from huggingface_hub import login
-#HF_TOKEN = os.getenv('HF_TOKEN')
 
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024