Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import re
 from pathlib import Path
 from statistics import median
-
+import json
 import pandas as pd
 import streamlit as st
 from bs4 import BeautifulSoup
@@ -213,6 +213,7 @@ def boot():
         st.session_state.costing = []
     if "messages" not in st.session_state:
         st.session_state.messages = []
+
     for message in st.session_state.messages:
         st.chat_message("human").write(message[0])
         st.chat_message("ai").write(message[1])
@@ -222,6 +223,7 @@ def boot():
         sorted_references = sorted([ref.metadata["chunk_id"] for ref in references])
         references_str = " ".join([f"[{ref}]" for ref in sorted_references])
         st.chat_message("ai").write(response + "\n\n---\nReferences:" + references_str)
+
         st.session_state.costing.append(
             {
                 "prompt tokens": stats.prompt_tokens,
@@ -232,6 +234,16 @@ def boot():
         stats_df = pd.DataFrame(st.session_state.costing)
         stats_df.loc["total"] = stats_df.sum()
         st.sidebar.write(stats_df)
+    st.sidebar.download_button(
+        "Download Conversation",
+        json.dumps(
+            [
+                {"human": message[0], "ai": message[1]}
+                for message in st.session_state.messages
+            ]
+        ),
+        "conversation.json",
+    )
 
 
 if __name__ == "__main__":