Update app.py

app.py

@@ -1,21 +1,33 @@
 import streamlit as st
-from datasets import load_dataset, concatenate_datasets
+from datasets import load_dataset, concatenate_datasets, save_dataset
 
 def load_and_combine_datasets():
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
     streamlit_issues_dataset = load_dataset("andfanilo/streamlit-issues", split='train')
     streamlit_docs_dataset = load_dataset("sai-lohith/streamlit_docs", split='train')
+    
     combined_dataset = concatenate_datasets([python_codes_dataset, streamlit_issues_dataset, streamlit_docs_dataset])
+    
     return combined_dataset
 
+def save_combined_dataset(combined_dataset, dataset_name):
+    save_dataset(combined_dataset, dataset_name)
+
 def main():
     st.title("Combined Dataset Viewer")
 
     # Load and combine datasets
     combined_dataset = load_and_combine_datasets()
 
-    # Display combined dataset
-    st.write("Combined Dataset:", combined_dataset)
+    # Display a subset of the combined dataset
+    st.write("Subset of Combined Dataset:", combined_dataset[:10])
+
+    # Add option to save the combined dataset
+    if st.button("Save Combined Dataset"):
+        dataset_name = st.text_input("Enter a name for the combined dataset:")
+        if dataset_name:
+            save_combined_dataset(combined_dataset, dataset_name)
+            st.success(f"Combined dataset saved as '{dataset_name}'!")
 
 if __name__ == "__main__":
     main()