Update app.py

app.py

@@ -2,10 +2,12 @@ import streamlit as st
 import pandas as pd
 import datetime as dt
 import matplotlib.pyplot as plt
+import io
 
 # Function to load data
 @st.cache_data
 def load_data(file):
+    file.seek(0)  # Move to the start of the file
     df = pd.read_csv(file)
     df['Date of Birth'] = pd.to_datetime(df['Date of Birth'])
     df['Visit Date'] = pd.to_datetime(df['Visit Date'])
@@ -82,11 +84,22 @@ if uploaded_file:
                     "Other Notes": other_notes,
                 }
                 df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
-                df.to_csv(uploaded_file, index=False)
+
+                # Save the updated data to CSV
+                buffer = io.StringIO()
+                df.to_csv(buffer, index=False)
+                buffer.seek(0)
+
+                st.download_button(
+                    label="Download Updated Data",
+                    data=buffer,
+                    file_name="updated_patient_data.csv",
+                    mime="text/csv"
+                )
                 st.success("New visit entry added successfully!")
 
                 # Reload data
-                df = load_data(uploaded_file)
+                df = load_data(buffer)
                 patient_data = df[df['Patient ID'] == patient_id]
 
         # Dropdown menu of previous visits
@@ -135,5 +148,16 @@ if uploaded_file:
                     "Other Notes": None,
                 }
                 df = pd.concat([df, pd.DataFrame([new_patient])], ignore_index=True)
-                df.to_csv(uploaded_file, index=False)
+
+                # Save the updated data to CSV
+                buffer = io.StringIO()
+                df.to_csv(buffer, index=False)
+                buffer.seek(0)
+
+                st.download_button(
+                    label="Download Updated Data",
+                    data=buffer,
+                    file_name="updated_patient_data.csv",
+                    mime="text/csv"
+                )
                 st.success("New patient account created successfully!")