Update app.py

app.py

@@ -268,6 +268,16 @@ def main():
     # Display chat history with response selection
     display_chat_history()
 
+    # Add a button to download training data
+    training_data = load_training_data()
+    if training_data:
+        st.download_button(
+            label="Download Training Data",
+            data=json.dumps(training_data, indent=2),
+            file_name="pte_assistant_training_data.json",
+            mime="application/json"
+        )
+