Update app.py

app.py

@@ -15,27 +15,16 @@ os.environ["OPENAI_API_KEY"] = os.environ['SECRET_CODE']
 DATASET_REPO_URL = "https://huggingface.co/datasets/peterpull/MediatorBot"
 DATA_FILENAME = "data.csv"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
+
+# I am guessing we need a write access token.
 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
-)
+    local_dir="data", 
+    clone_from=DATASET_REPO_URL, 
+    use_auth_token=HF_TOKEN)
 
 def generate_html() -> str:
     with open(DATA_FILE) as csvfile:
@@ -56,12 +45,12 @@ def generate_html() -> str:
             html += "</div>"
             return html
 
-def store_message(name: str, message: str):
+def store_message(chatinput: str, chatresponse: 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())}
+                {"User": chatinput, "Chatbot": chatresponse, "time": str(datetime.now())}
             )
         commit_url = repo.push_to_hub()
         print(commit_url)
@@ -82,6 +71,8 @@ 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")
+
+    store_message(input_text,response)
     
     # return the response
     return response.response