Update app.py

app.py

@@ -18,6 +18,10 @@ def load_data(file):
 uploaded_file = st.sidebar.file_uploader("Upload Patient Data CSV", type="csv")
 if uploaded_file:
     df = load_data(uploaded_file)
+    buffer = io.StringIO()
+    df.to_csv(buffer, index=False)
+    buffer.seek(0)
+    bytes_data = io.BytesIO(buffer.getvalue().encode('utf-8'))
 
     # Sidebar: Select or create patient
     action = st.sidebar.selectbox("Action", ["Select Patient", "Create New Patient"])
@@ -92,6 +96,10 @@ if uploaded_file:
                 bytes_data = io.BytesIO(buffer.getvalue().encode('utf-8'))
                 st.success("New visit entry added successfully!")
 
+                # Reload data
+                df = load_data(io.StringIO(buffer.getvalue()))
+                patient_data = df[df['Patient ID'] == patient_id]
+
         # Download button (moved outside the form)
         st.download_button(
             label="Download Updated Data",
@@ -100,10 +108,6 @@ if uploaded_file:
             mime="text/csv"
         )
 
-        # Reload data
-        df = load_data(io.StringIO(buffer.getvalue()))
-        patient_data = df[df['Patient ID'] == patient_id]
-
         # Dropdown menu of previous visits
         st.header("Previous Visits")
         visit_date = st.selectbox("Select Visit Date", patient_data['Visit Date'].dt.strftime('%Y-%m-%d').unique())