Copy logged QA pairs from persistent storage to repo after each downvote

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import csv
+import shutil
 from datetime import datetime
 import openai
 import gradio as gr
@@ -34,7 +35,8 @@ chat = ChatOpenAI(model_name="gpt-4",temperature=0)
 # Make sure we don't exceed estimation of token limit:
 TOKEN_LIMIT = 4096  # GPT-3.5 Turbo token limit
 BUFFER = 100  # Extra tokens to consider for incoming messages
-LOG_PATH = "/data/downvoted_responses.csv" # File in which to log downvoted responses
+PERSISTENT_LOG_PATH = "/data/downvoted_responses.csv" # File in which to log downvoted responses
+LOCAL_LOG_PATH = "./data/downvoted_responses.csv"
 
 def estimate_tokens(texts):
     return sum([len(t.split()) for t in texts])
@@ -92,20 +94,23 @@ def bot(chatbot_history):
         time.sleep(0.05)
         yield chatbot_history
 
-def log_to_csv(question, answer, log_path=LOG_PATH):
+def log_to_csv(question, answer):
     """Append a line to a CSV. Create a new file if needed."""
     now = datetime.today().strftime("%Y%m%d_%H:%M:%S")
-    if not os.path.isfile(log_path):
+    if not os.path.isfile(PERSISTENT_LOG_PATH):
         # Add the column names to the CSV
-        with open(log_path, "a") as csv_file:
+        with open(PERSISTENT_LOG_PATH, "w+") as csv_file:
             writer = csv.writer(csv_file)
             writer.writerow(["datetime", "user_question", "bot_response"])
 
     # Write the disliked message to the CSV
-    with open(log_path, "a") as csv_file:
+    with open(PERSISTENT_LOG_PATH, "a") as csv_file:
         writer = csv.writer(csv_file)
         writer.writerow([now, question, answer])
 
+    # Copy file from persistent storage to local repo
+    shutil.copyfile(PERSISTENT_LOG_PATH, LOCAL_LOG_PATH)
+
 def get_voted_qa_pair(history, voted_answer):
     """Return the question-answer pair from the chat history, given a 
     particular bot answer. Note: This is required because the 'vote'
@@ -115,7 +120,7 @@ def get_voted_qa_pair(history, voted_answer):
         if answer == voted_answer:
             return question, answer
 
-def vote(data: gr.LikeData, history, log_path=LOG_PATH):
+def vote(data: gr.LikeData, history):
     """This is a function to do something with the voted information"""
     print(history)
     if data.liked:
@@ -124,7 +129,7 @@ def vote(data: gr.LikeData, history, log_path=LOG_PATH):
         print("You downvoted this response: " + data.value)
         # Find Q/A pair that was disliked
         question, answer = get_voted_qa_pair(history, data.value)
-        log_to_csv(question, answer, log_path)
+        log_to_csv(question, answer)
 
 # The Gradio App interface
 with gr.Blocks() as demo: