chat history import/export

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import base64
 import os
 from openai import OpenAI
+import json
 
 dump_controls = False
 log_to_console = False
@@ -159,6 +160,20 @@ def bot(message, history, oai_key, system_prompt, seed, temperature, max_tokens,
 
     return "", history
 
+def import_history(history, file):
+    with open(file.name, mode="rb") as f:
+        content = f.read()
+
+        if isinstance(content, bytes):
+            content = content.decode('utf-8', 'replace')
+        else:
+            content = str(content)
+
+    # Deserialize the JSON content to history
+    history = json.loads(content)
+    # The history is returned and will be set to the chatbot component
+    return history
+
 with gr.Blocks() as demo:
     gr.Markdown("# OAI Chat (Nils' Version™️)")
 
@@ -238,4 +253,26 @@ with gr.Blocks() as demo:
     file_msg = btn.upload(add_file, [chatbot, btn], [chatbot], queue=False, postprocess=False)
     img_msg = img_btn.upload(add_img, [chatbot, img_btn], [chatbot], queue=False, postprocess=False)
 
+    with gr.Accordion("Import/Export", open = False):
+        import_button = gr.UploadButton("Import")
+        export_button = gr.Button("Export")
+        export_button.click(lambda: None, [chatbot], js="""
+            (chat_history) => {
+                // Convert the chat history to a JSON string
+                const history_json = JSON.stringify(chat_history);
+                // Create a Blob from the JSON string
+                const blob = new Blob([history_json], {type: 'application/json'});
+                // Create a download link
+                const url = URL.createObjectURL(blob);
+                const a = document.createElement('a');
+                a.href = url;
+                a.download = 'chat_history.json';
+                document.body.appendChild(a);
+                a.click();
+                document.body.removeChild(a);
+                URL.revokeObjectURL(url);
+            }
+            """)
+        import_button.upload(import_history, inputs=[chatbot, import_button], outputs=[chatbot])
+
 demo.queue().launch()