Log question and answer for each downvote to CSV

app.py

@@ -1,5 +1,7 @@
 import os
 import time
+import csv
+from datetime import datetime
 import openai
 import gradio as gr
 
@@ -32,7 +34,7 @@ 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/logs.txt" # File in which to log downvoted responses
+LOG_PATH = "/data/downvoted_responses.csv" # File in which to log downvoted responses
 
 def estimate_tokens(texts):
     return sum([len(t.split()) for t in texts])
@@ -90,14 +92,39 @@ def bot(chatbot_history):
         time.sleep(0.05)
         yield chatbot_history
 
-# This is a function to do something with the voted information (TODO: Save this info somewhere?)
-def vote(data: gr.LikeData, log_path=LOG_PATH):
+def log_to_csv(question, answer, log_path=LOG_PATH):
+    """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):
+        # Add the column names to the CSV
+        with open(log_path, "a") 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:
+        writer = csv.writer(csv_file)
+        writer.writerow([now, question, answer])
+
+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'
+    event handler only has access to the answer that was liked/disliked.
+    """
+    for question, answer in history:
+        if answer == voted_answer:
+            return question, answer
+
+def vote(data: gr.LikeData, history, log_path=LOG_PATH):
+    """This is a function to do something with the voted information"""
+    print(history)
     if data.liked:
         print("You upvoted this response: " + data.value)
     else:
         print("You downvoted this response: " + data.value)
-        with open(log_path, "a") as text_file:
-          print(f"Disliked content: {data.value}", file=text_file)
+        # Find Q/A pair that was disliked
+        question, answer = get_voted_qa_pair(history, data.value)
+        log_to_csv(question, answer, log_path)
 
 # The Gradio App interface
 with gr.Blocks() as demo:
@@ -112,7 +139,7 @@ with gr.Blocks() as demo:
         bot, chatbot, chatbot,
     )
     clear.click(lambda: None, None, chatbot, queue=False)
-    chatbot.like(vote, None, None)
+    chatbot.like(vote, chatbot, None)
 
 # Enable queuing
 demo.queue()