Update app.py

app.py

@@ -3,7 +3,10 @@ import time
 import random
 import json
 import mysql.connector
-import json
+import os
+import csv
+from datetime import datetime
+from huggingface_hub import Repository, hf_hub_download
 
 session_index = None
 
@@ -23,6 +26,38 @@ get_window_session_index = """
         }
     """
 
+DATASET_REPO_URL = "https://huggingface.co/datasets/botsi/trust-game-llama-2-7b-chat"
+DATASET_REPO_ID = "botsi/trust-game-llama-2-7b-chat"
+DATA_FILENAME = "your_data.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+# Initialize Hugging Face Dataset Repository
+try:
+    hf_hub_download(
+        repo_id=DATASET_REPO_ID,
+        filename=DATA_FILENAME,
+        force_filename=DATA_FILENAME
+    )
+except:
+    print("File not found")
+
+repo = Repository(
+    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
+
+# Store Message Function
+def store_message(name: str, message: str):
+    if name and message:
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
+            writer.writerow(
+                {"name": name, "message": message, "time": str(datetime.now())}
+            )
+        commit_url = repo.push_to_hub()
+    return generate_html()
+
+
 # get url params out of JavaScript function as variables and pass the session_index to fetch_personalized_data() function
 
 with gr.Blocks() as demo:
@@ -37,8 +72,7 @@ type `read query`
     def user(user_message, url_params, history):
         return "", history + [[user_message, None]]
 
-
-    def bot(history, url_params):
+        def bot(history, url_params):
         global session_index
         if "read query" in history[-1][0]:
             session_index = url_params.get('session_index')
@@ -142,51 +176,13 @@ print(personalized_data)
 demo.launch()
 
 
-def generate_html() -> str:
-    with open(DATA_FILE) as csvfile:
-        reader = csv.DictReader(csvfile)
-        rows = []
-        for row in reader:
-            rows.append(row)
-        rows.reverse()
-        if len(rows) == 0:
-            return "no messages yet"
-        else:
-            html = "<div class='chatbot'>"
-            for row in rows:
-                html += "<div>"
-                html += f"<span>{row['name']}</span>"
-                html += f"<span class='message'>{row['message']}</span>"
-                html += "</div>"
-            html += "</div>"
-            return html
 
-def store_message(name: str, message: str):
-    if name and message:
-        with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
-            writer.writerow(
-                {"name": name, "message": message, "time": str(datetime.now())}
-            )
-        commit_url = repo.push_to_hub()
-    return generate_html()
 
-iface = gr.Interface(
-    store_message,
-    [
-        inputs.Textbox(placeholder="Your name"),
-        inputs.Textbox(placeholder="Your message", lines=2),
-    ],
-    "html",
-    css="""
-    .message {background-color:cornflowerblue;color:white; padding:4px;margin:4px;border-radius:4px; }
-    """,
-    title="Reading/writing to a HuggingFace dataset repo from Spaces",
-    description=f"This is a demo of how to do simple *shared data persistence* in a Gradio Space, backed by a dataset repo.",
-    article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL})",
-)
 
-iface.launch()
+
+
+
+
 
 
 '''