Update app.py

app.py

@@ -3,6 +3,7 @@ from datasets import load_dataset, concatenate_datasets
 import json
 import os
 import base64
+from datetime import datetime
 
 def load_and_combine_datasets():
     python_codes_dataset = load_dataset('flytech/python-codes-25k', split='train')
@@ -13,10 +14,14 @@ def load_and_combine_datasets():
     
     return combined_dataset
 
+def datetime_serializer(o):
+    if isinstance(o, datetime):
+        return o.strftime('%Y-%m-%d %H:%M:%S')
+
 def save_combined_dataset_as_jsonl(combined_dataset, file_path):
     with open(file_path, 'w', encoding='utf-8') as f:
         for example in combined_dataset:
-            json.dump(example, f, ensure_ascii=False)
+            json.dump(example, f, ensure_ascii=False, default=datetime_serializer)
             f.write('\n')
 
 def main():
@@ -42,8 +47,8 @@ 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:
-            bytes_data = f.read().encode()
+        with open(file_path, "rb") as f:
+            bytes_data = f.read()
         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)
@@ -55,4 +60,4 @@ def main():
                            mime="application/jsonl")
 
 if __name__ == "__main__":
-    main()
+    main()