Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from datasets import load_dataset, concatenate_datasets
+import os
 
 def load_and_combine_datasets():
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
@@ -12,6 +13,8 @@ def load_and_combine_datasets():
 
 def save_combined_dataset(combined_dataset, dataset_name):
     combined_dataset.save_to_disk(dataset_name)
+    # Print the location where the dataset is saved
+    st.write(f"Dataset saved at: {dataset_name}")
 
 def main():
     st.title("Combined Dataset Viewer")
@@ -29,5 +32,15 @@ def main():
             save_combined_dataset(combined_dataset, dataset_name)
             st.success(f"Combined dataset saved as '{dataset_name}'!")
 
+    # Add option to download the combined dataset
+    if st.button("Download Combined Dataset"):
+        dataset_name = st.text_input("Enter a name for the combined dataset:")
+        if dataset_name:
+            save_combined_dataset(combined_dataset, dataset_name)
+            filepath = os.path.join(os.getcwd(), dataset_name)
+            filesize = os.path.getsize(filepath) / (1024 * 1024)  # Size in MB
+            st.write(f"Download the combined dataset: [{dataset_name}]({filepath})")
+            st.write(f"File Size: {filesize:.2f} MB")
+
 if __name__ == "__main__":
     main()