Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import csv
+import io
 
 # Initialize an empty list for storing the daily routine
 daily_routine = []
@@ -33,18 +34,21 @@ def display_todo_list():
         st.write(f"**Description**: {task['description']}")
         st.markdown("---")
 
-# Function to save the daily routine to a CSV file
-def save_to_csv(filename):
-    try:
-        with open(filename, mode="w", newline="") as file:
-            writer = csv.writer(file)
-            writer.writerow(["Time", "Task", "Description"])  # Write headers
-            for task in daily_routine:
-                writer.writerow([task['time'], task['task'], task['description']])
-
-        st.success(f"To-do list saved as '{filename}'")
-    except Exception as e:
-        st.error(f"Error: {e}")
+# Function to save the daily routine to a CSV file in memory (using StringIO)
+def save_to_csv():
+    # Create a StringIO object to hold the CSV data in memory
+    csv_buffer = io.StringIO()
+    
+    # Write to the buffer
+    writer = csv.writer(csv_buffer)
+    writer.writerow(["Time", "Task", "Description"])  # Write headers
+    for task in daily_routine:
+        writer.writerow([task['time'], task['task'], task['description']])
+    
+    # Move the buffer's cursor to the beginning of the file
+    csv_buffer.seek(0)
+    
+    return csv_buffer.getvalue()
 
 # Streamlit UI
 def main():
@@ -59,9 +63,15 @@ def main():
         if st.button("Display To-Do List"):
             display_todo_list()
 
-    # Save the list to a CSV file
-    if st.button("Save to CSV"):
-        save_to_csv('/app/daily_routine.csv')  # Path within Hugging Face Spaces
+    # Save the list to CSV and provide a download button
+    if len(daily_routine) > 0:
+        csv_data = save_to_csv()
+        st.download_button(
+            label="Download Daily Routine as CSV",
+            data=csv_data,
+            file_name="daily_routine.csv",
+            mime="text/csv"
+        )
 
 if __name__ == "__main__":
     main()