feat: record everything in data.json

app.py

@@ -3,6 +3,7 @@ import os
 import shutil
 import requests
 import threading
+import time
 
 import gradio as gr
 from huggingface_hub import Repository, InferenceClient
@@ -51,12 +52,6 @@ def asynchronous_push(f_stop):
 f_stop = threading.Event()
 asynchronous_push(f_stop)
 
-def vote(data: gr.LikeData):
-    if data.liked:
-        print("You upvoted this response: " + data.value)
-    else:
-        print("You downvoted this response: " + data.value)
-
 def format_prompt(message, history, system_prompt):
   prompt = ""
   if system_prompt:
@@ -186,6 +181,34 @@ def evolve_text(x):
 
     return possible_generation
 
+def vote(data: gr.LikeData):
+    if data.liked:
+        print("You upvoted this response: " + data.value)
+    else:
+        print("You downvoted this response: " + data.value)
+
+    print(data)
+
+    with open(DATA_FILE, "a") as jsonlfile:
+        json_data = [
+            json.dumps(
+                    {
+                        "index": data.index,
+                        "time_stamp": time.time(),
+                        "prompt": "",
+                        "history": "",
+                        "system prompt": additional_inputs[0].value,
+                        "temperature": additional_inputs[1].value,
+                        "max_new_tokens": additional_inputs[2].value,
+                        "top_p": additional_inputs[3].value,
+                        "repetition_penalty": additional_inputs[4].value,
+                        "response": data.value,
+                        "label": data.liked,
+                    }
+            )
+        ]
+        jsonlfile.write("\n".join(json_data) + "\n")
+
 with gr.Blocks(css=CSS) as demo:
     with gr.Row():
         with gr.Column():