add system prompt to history export

app.py

@@ -112,10 +112,13 @@ def import_history(history, file):
         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
+    # Deserialize the JSON content
+    import_data = json.loads(content)
+
+    new_history = import_data.get('history', history)  # Use existing history if not provided
+    new_system_prompt = import_data.get('system_prompt', '')  # Default to empty if not provided
+
+    return new_history, new_system_prompt
 
 with gr.Blocks() as demo:
     gr.Markdown("# Amazon™️ Bedrock™️ Chat™️ (Nils' Version™️) feat. Mistral™️ AI & Anthropic™️ Claude™️")
@@ -218,13 +221,14 @@ with gr.Blocks() as demo:
     with gr.Accordion("Import/Export", open = False):
         import_button = gr.UploadButton("History Import")
         export_button = gr.Button("History 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
+        export_button.click(lambda: None, [chatbot, system_prompt], js="""
+            (chat_history, system_prompt) => {
+                const export_data = {
+                    history: chat_history,
+                    system_prompt: system_prompt
+                };
+                const history_json = JSON.stringify(export_data);
                 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;
@@ -264,7 +268,7 @@ with gr.Blocks() as demo:
                 }
             }
         """)
-        import_button.upload(import_history, inputs=[chatbot, import_button], outputs=[chatbot])
+        import_button.upload(import_history, inputs=[chatbot, import_button], outputs=[chatbot, system_prompt])
 
     txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
         bot, [txt, chatbot, aws_access, aws_secret, aws_token, system_prompt, temp, max_tokens, model, region], [txt, chatbot],