Merge branch 'restore' of https://github.com/intuit/Trapheus into res…

…tore
intuit · Oct 30, 2023 · fdd9c59 · fdd9c59
2 parents 4800d9d + 022db9d
commit fdd9c59
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/labs/TrapheusAI/trapheusai_app.py b/labs/TrapheusAI/trapheusai_app.py
@@ -47,6 +47,9 @@ def handle_dataset_search():
                 repository_url = data["fullurl"]
                 df = pd.read_csv(repository_url)
                 streamlit.data_editor(df)
+                streamlit.download_button(
+                    "Download as CSV", data = df.to_csv(index=False), file_name=f"{choice}.csv", 
+                    mime = "text/csv", on_click = showDownloadMessage)
                 question = streamlit.chat_input("Ask any question related to the dataset")
                 if question:
 
@@ -98,6 +101,9 @@ def style_page():
             """
     streamlit.markdown(style, unsafe_allow_html=True)
 
+def showDownloadMessage():
+    streamlit.success("Your dataset has been saved")
+
 
 if __name__ == "__main__":
     initialize_page()