add download button

src/perfguru.py

@@ -78,8 +78,8 @@ def chat_with_llms(prompt, code_files, profile_file, profile_type):
     vote_buttons = gr.Button(interactive=True), gr.Button(interactive=True), gr.Button(interactive=True), gr.Button(interactive=True)
     yield c1 or gr.Textbox(), source1, formatted1, c2 or gr.Textbox(), source2, formatted2, *vote_buttons
 
-def log_interaction(prompt, vote, response1, model1, formatter1, full_prompt1, response2, model2, formatter2, full_prompt2):
-    interaction = {
+def get_interaction_log(prompt, vote, response1, model1, formatter1, full_prompt1, response2, model2, formatter2, full_prompt2):
+    log = {
         "prompt": prompt,
         "full_prompt1": full_prompt1,
         "full_prompt2": full_prompt2,
@@ -92,21 +92,10 @@ def log_interaction(prompt, vote, response1, model1, formatter1, full_prompt1, r
         "formatter2": formatter2,
         "timestamp": datetime.datetime.now().isoformat()
     }
-
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S-%f")[:-3]
-
-    log_file_path = f"perf_guru_logs/log_{timestamp}.json" 
-    if os.path.exists(log_file_path):
-        with open(log_file_path, "r") as log_file:
-            logs = json.load(log_file)
-    else:
-        logs = []
-    
-    logs.append(interaction)
-    
-    # Write updated logs to file
-    with open(log_file_path, "w") as log_file:
-        json.dump(logs, log_file, indent=4)
+    fpath = f"interaction_log_{datetime.datetime.now().isoformat()}.json"
+    with open(fpath, "w") as f:
+        json.dump(log, f, indent=2)
+    return fpath
     
 
 def handle_vote(prompt, vote, response1, source1, full_prompt1, response2, source2, full_prompt2):
@@ -115,9 +104,11 @@ def handle_vote(prompt, vote, response1, source1, full_prompt1, response2, sourc
 
     label1_class = ["voted"] if vote == "Vote for Response 1" else []
     label2_class = ["voted"] if vote == "Vote for Response 2" else []
+    did_vote = vote != "Skip"
 
-    log_interaction(prompt, vote, response1, model1, formatter1, full_prompt1, response2, model2, formatter2, full_prompt2)
+    log_fpath = get_interaction_log(prompt, vote, response1, model1, formatter1, full_prompt1, response2, model2, formatter2, full_prompt2)
     return gr.Markdown(visible=True, elem_classes=label1_class), gr.Markdown(visible=True, elem_classes=label2_class), \
+        gr.Button(interactive=did_vote, value=log_fpath), \
         gr.Button(interactive=False), gr.Button(interactive=False), gr.Button(interactive=False), gr.Button(interactive=False)
 
 # Define the Gradio interface
@@ -175,9 +166,11 @@ with gr.Blocks(css=".not-voted p { color: black; } .voted p { color: green; } .r
         tie_button = gr.Button("Vote for Tie", interactive=False)
         skip_button = gr.Button("Skip", interactive=False)
     
+    download_btn = gr.DownloadButton("Download Log", interactive=False)
+
     vote_btns = [vote1_button, vote2_button, tie_button, skip_button]
     for btn in vote_btns:
-        btn.click(handle_vote, inputs=[prompt, btn, response1, source1, full_prompt1, response2, source2, full_prompt2], outputs=[source1, source2, *vote_btns])
+        btn.click(handle_vote, inputs=[prompt, btn, response1, source1, full_prompt1, response2, source2, full_prompt2], outputs=[source1, source2, download_btn, *vote_btns])
 
     # final chat button
     chat_button.click(