Update app.py

app.py

@@ -4,10 +4,72 @@ import gradio as gr
 import sys
 import os
 import datetime
+import huggingface_hub
+from huggingface_hub import Repository
+from datetime import datetime
+import csv
 
 os.environ["OPENAI_API_KEY"] = os.environ['SECRET_CODE']
 
+# Need to write to persistent dataset because cannot store temp data on spaces
+DATASET_REPO_URL = "https://huggingface.co/datasets/peterpull/MediatorBot"
+DATA_FILENAME = "data.csv"
+DATA_FILE = os.path.join("data", DATA_FILENAME)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+print("HF TOKEN is none?", HF_TOKEN is None)
+print("HF hub ver", huggingface_hub.__version__)
 
+# overriding/appending to the gradio template
+SCRIPT = """
+<script>
+if (!window.hasBeenRun) {
+    window.hasBeenRun = true;
+    console.log("should only happen once");
+    document.querySelector("button.submit").click();
+}
+</script>
+"""
+with open(os.path.join(gr.networking.STATIC_TEMPLATE_LIB, "frontend", "index.html"), "a") as f:
+    f.write(SCRIPT)
+
+
+repo = Repository(
+    local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
+
+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['User input']}</span>"
+                html += f"<span class='message'>{row['Chatbot Response']}</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=["User", "Chatbot", "time"])
+            writer.writerow(
+                {"User": {input_text}, "Chatbot": {response.response}, "time": str(datetime.now())}
+            )
+        commit_url = repo.push_to_hub()
+        print(commit_url)
+
+    return generate_html()
+
+            
+#gets the index file which is the context data
 def get_index(index_file_path):
     if os.path.exists(index_file_path):
         return GPTSimpleVectorIndex.load_from_disk(index_file_path)
@@ -15,25 +77,11 @@ def get_index(index_file_path):
         print(f"Error: '{index_file_path}' does not exist.")
         sys.exit()
 
-
+# passes the prompt to the chatbot
 def chatbot(input_text, mentioned_person='Mediator John Haynes'):
     index = get_index('./index/indexsmall.json')
     prompt = f"You are {mentioned_person}: {input_text}\n\n At the end of your answer  ask a provocative question."
     response = index.query(prompt, response_mode="compact")
-
-
-
-   # code to save chat log to file
-    current_time = datetime.datetime.now()
-    current_time_str = current_time.strftime("%Y-%m-%d_%H-%M-%S")
-    chat_log_filename = "chat_history.txt"
-    chat_log_dir = os.path.dirname(os.path.abspath(__file__))
-    chat_log_filepath = os.path.join(chat_log_dir, chat_log_filename)
-    with open(chat_log_filepath, "a") as f:
-        f.write(f"Chat at {current_time_str}\n")
-        f.write(f"User: {input_text}\n")
-        f.write(f"Chatbot: {response.response}\n\n")
-        print(f"Chat log written to {chat_log_filepath}")
     
     # return the response
     return response.response