Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from datasets import load_dataset, concatenate_datasets
 import json
 import os
-import base64  # Add base64 for encoding
+import base64
 
 def load_and_combine_datasets():
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
@@ -42,11 +42,17 @@ def main():
         file_path = os.path.join(os.getcwd(), f"{output_dataset_name}.jsonl")
         save_combined_dataset_as_jsonl(combined_dataset, file_path)
         st.write("Download the combined dataset as JSONL file:")
-        with open(file_path, "r") as f:  # Open file in read mode
-            bytes_data = f.read().encode()  # Encode the data as bytes
+        with open(file_path, "r") as f:
+            bytes_data = f.read().encode()
         b64 = base64.b64encode(bytes_data).decode()
         href = f'<a href="data:file/jsonl;base64,{b64}" download="{output_dataset_name}.jsonl">Download JSONL File</a>'
         st.markdown(href, unsafe_allow_html=True)
+        
+        # Provide download button
+        st.download_button(label="Click to Download",
+                           data=bytes_data,
+                           file_name=f"{output_dataset_name}.jsonl",
+                           mime="application/jsonl")
 
 if __name__ == "__main__":
     main()