Update app.py

app.py

@@ -4,174 +4,149 @@ import datetime as dt
 import matplotlib.pyplot as plt
 import io
 
-# Function to load data
+# Function to load data from the local CSV file
 @st.cache_data
-def load_data(file):
-    file.seek(0)  # Move to the start of the file
-    df = pd.read_csv(file)
+def load_data():
+    df = pd.read_csv('patient_data.csv')
     df['Date of Birth'] = pd.to_datetime(df['Date of Birth'], format='%Y-%m-%d', errors='coerce')
     df['Visit Date'] = pd.to_datetime(df['Visit Date'], errors='coerce')
     df['Age'] = df['Date of Birth'].apply(lambda dob: (dt.datetime.now() - dob).days // 365)
     return df
 
-# Sidebar: Upload CSV 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'))
+# Load data from the CSV file
+df = load_data()
 
-    # Sidebar: Select or create patient
-    action = st.sidebar.selectbox("Action", ["Select Patient", "Create New Patient"])
+# Function to save data to the local CSV file
+def save_data(df):
+    df.to_csv('patient_data.csv', index=False)
 
-    if action == "Select Patient":
-        patient_id = st.sidebar.selectbox("Patient ID", df['Patient ID'].unique())
+# Sidebar: Select or create patient
+action = st.sidebar.selectbox("Action", ["Select Patient", "Create New Patient"])
 
-        # Display patient data
-        patient_data = df[df['Patient ID'] == patient_id]
-        if not patient_data.empty:
-            st.header("General Information")
-            st.write(f"**Patient ID:** {patient_id}")
-            st.write(f"**Name:** {patient_data.iloc[0]['Patient Name']}")
-            st.write(f"**Date of Birth:** {patient_data.iloc[0]['Date of Birth'].strftime('%Y-%m-%d')}")
-            st.write(f"**Age:** {patient_data.iloc[0]['Age']}")
-            st.write(f"**Gender:** {patient_data.iloc[0]['Gender']}")
-            st.write(f"**Medical History:** {patient_data.iloc[0]['Medical History']}")
-            st.write(f"**Allergies:** {patient_data.iloc[0]['Allergies']}")
+if action == "Select Patient":
+    patient_id = st.sidebar.selectbox("Patient ID", df['Patient ID'].unique())
 
-            # Graphs of medical data
-            st.header("Medical Data Over Time")
-            fig, ax = plt.subplots(2, 2, figsize=(12, 8))
-            ax[0, 0].plot(patient_data['Visit Date'], patient_data['Systolic BP'])
-            ax[0, 0].set_title("Systolic Blood Pressure")
-            ax[0, 1].plot(patient_data['Visit Date'], patient_data['Glucose'])
-            ax[0, 1].set_title("Glucose")
-            ax[1, 0].plot(patient_data['Visit Date'], patient_data['Cholesterol'])
-            ax[1, 0].set_title("Cholesterol")
-            ax[1, 1].plot(patient_data['Visit Date'], patient_data['Hemoglobin'])
-            ax[1, 1].set_title("Hemoglobin")
-            st.pyplot(fig)
+    # Display patient data
+    patient_data = df[df['Patient ID'] == patient_id]
+    if not patient_data.empty:
+        st.header("General Information")
+        st.write(f"**Patient ID:** {patient_id}")
+        st.write(f"**Name:** {patient_data.iloc[0]['Patient Name']}")
+        st.write(f"**Date of Birth:** {patient_data.iloc[0]['Date of Birth'].strftime('%Y-%m-%d')}")
+        st.write(f"**Age:** {patient_data.iloc[0]['Age']}")
+        st.write(f"**Gender:** {patient_data.iloc[0]['Gender']}")
+        st.write(f"**Medical History:** {patient_data.iloc[0]['Medical History']}")
+        st.write(f"**Allergies:** {patient_data.iloc[0]['Allergies']}")
 
-            # Current visit input
-            st.header("Current Visit")
-            with st.form("current_visit_form"):
-                new_visit_date = st.date_input("Visit Date", dt.date.today())
-                complaint = st.text_area("Complaint")
-                physical_exam = st.text_area("Physical Examination")
-                systolic_bp = st.number_input("Systolic Blood Pressure", min_value=0)
-                diastolic_bp = st.number_input("Diastolic Blood Pressure", min_value=0)
-                temperature = st.number_input("Temperature", min_value=0.0, format="%.1f")
-                glucose = st.number_input("Glucose", min_value=0)
-                cholesterol = st.number_input("Cholesterol", min_value=0)
-                hemoglobin = st.number_input("Hemoglobin", min_value=0)
-                other_notes = st.text_area("Other Notes")
-                submitted = st.form_submit_button("Add Entry")
-                if submitted:
-                    new_entry = {
-                        "Patient ID": patient_id,
-                        "Patient Name": patient_data.iloc[0]['Patient Name'],
-                        "Date of Birth": patient_data.iloc[0]['Date of Birth'],
-                        "Age": patient_data.iloc[0]['Age'],
-                        "Gender": patient_data.iloc[0]['Gender'],
-                        "Medical History": patient_data.iloc[0]['Medical History'],
-                        "Allergies": patient_data.iloc[0]['Allergies'],
-                        "Visit Date": new_visit_date,
-                        "Complaint": complaint,
-                        "Physical Examination": physical_exam,
-                        "Systolic BP": systolic_bp,
-                        "Diastolic BP": diastolic_bp,
-                        "Temperature": temperature,
-                        "Glucose": glucose,
-                        "Cholesterol": cholesterol,
-                        "Hemoglobin": hemoglobin,
-                        "Other Notes": other_notes,
-                    }
-                    df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
+        # Graphs of medical data
+        st.header("Medical Data Over Time")
+        fig, ax = plt.subplots(2, 2, figsize=(12, 8))
+        ax[0, 0].plot(patient_data['Visit Date'], patient_data['Systolic BP'])
+        ax[0, 0].set_title("Systolic Blood Pressure")
+        ax[0, 1].plot(patient_data['Visit Date'], patient_data['Glucose'])
+        ax[0, 1].set_title("Glucose")
+        ax[1, 0].plot(patient_data['Visit Date'], patient_data['Cholesterol'])
+        ax[1, 0].set_title("Cholesterol")
+        ax[1, 1].plot(patient_data['Visit Date'], patient_data['Hemoglobin'])
+        ax[1, 1].set_title("Hemoglobin")
+        st.pyplot(fig)
 
-                    # Save the updated data to CSV
-                    buffer = io.StringIO()
-                    df.to_csv(buffer, index=False)
-                    buffer.seek(0)
-                    bytes_data = io.BytesIO(buffer.getvalue().encode('utf-8'))
-                    st.success("New visit entry added successfully!")
+        # Current visit input
+        st.header("Current Visit")
+        with st.form("current_visit_form"):
+            new_visit_date = st.date_input("Visit Date", dt.date.today())
+            complaint = st.text_area("Complaint")
+            physical_exam = st.text_area("Physical Examination")
+            systolic_bp = st.number_input("Systolic Blood Pressure", min_value=0)
+            diastolic_bp = st.number_input("Diastolic Blood Pressure", min_value=0)
+            temperature = st.number_input("Temperature", min_value=0.0, format="%.1f")
+            glucose = st.number_input("Glucose", min_value=0)
+            cholesterol = st.number_input("Cholesterol", min_value=0)
+            hemoglobin = st.number_input("Hemoglobin", min_value=0)
+            other_notes = st.text_area("Other Notes")
+            submitted = st.form_submit_button("Add Entry")
+            if submitted:
+                new_entry = {
+                    "Patient ID": patient_id,
+                    "Patient Name": patient_data.iloc[0]['Patient Name'],
+                    "Date of Birth": patient_data.iloc[0]['Date of Birth'],
+                    "Age": patient_data.iloc[0]['Age'],
+                    "Gender": patient_data.iloc[0]['Gender'],
+                    "Medical History": patient_data.iloc[0]['Medical History'],
+                    "Allergies": patient_data.iloc[0]['Allergies'],
+                    "Visit Date": new_visit_date,
+                    "Complaint": complaint,
+                    "Physical Examination": physical_exam,
+                    "Systolic BP": systolic_bp,
+                    "Diastolic BP": diastolic_bp,
+                    "Temperature": temperature,
+                    "Glucose": glucose,
+                    "Cholesterol": cholesterol,
+                    "Hemoglobin": hemoglobin,
+                    "Other Notes": other_notes,
+                }
+                df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
 
-                    # Reload data
-                    df = load_data(io.StringIO(buffer.getvalue()))
-                    patient_data = df[df['Patient ID'] == patient_id]
+                # Save the updated data to CSV
+                save_data(df)
+                st.success("New visit entry added successfully!")
 
-            # Download button (moved outside the form)
-            st.download_button(
-                label="Download Updated Data",
-                data=bytes_data,
-                file_name="updated_patient_data.csv",
-                mime="text/csv"
-            )
+                # Reload data
+                df = load_data()
+                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())
-            visit_data = patient_data[patient_data['Visit Date'] == pd.to_datetime(visit_date)]
-            if not visit_data.empty:
-                st.write(f"**Complaint:** {visit_data.iloc[0]['Complaint']}")
-                st.write(f"**Physical Examination:** {visit_data.iloc[0]['Physical Examination']}")
-                st.write(f"**Systolic BP:** {visit_data.iloc[0]['Systolic BP']}")
-                st.write(f"**Diastolic BP:** {visit_data.iloc[0]['Diastolic BP']}")
-                st.write(f"**Temperature:** {visit_data.iloc[0]['Temperature']}")
-                st.write(f"**Glucose:** {visit_data.iloc[0]['Glucose']}")
-                st.write(f"**Cholesterol:** {visit_data.iloc[0]['Cholesterol']}")
-                st.write(f"**Hemoglobin:** {visit_data.iloc[0]['Hemoglobin']}")
-                st.write(f"**Other Notes:** {visit_data.iloc[0]['Other Notes']}")
-            else:
-                st.write("No visit data available for the selected date.")
+        # 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())
+        visit_data = patient_data[patient_data['Visit Date'] == pd.to_datetime(visit_date)]
+        if not visit_data.empty:
+            st.write(f"**Complaint:** {visit_data.iloc[0]['Complaint']}")
+            st.write(f"**Physical Examination:** {visit_data.iloc[0]['Physical Examination']}")
+            st.write(f"**Systolic BP:** {visit_data.iloc[0]['Systolic BP']}")
+            st.write(f"**Diastolic BP:** {visit_data.iloc[0]['Diastolic BP']}")
+            st.write(f"**Temperature:** {visit_data.iloc[0]['Temperature']}")
+            st.write(f"**Glucose:** {visit_data.iloc[0]['Glucose']}")
+            st.write(f"**Cholesterol:** {visit_data.iloc[0]['Cholesterol']}")
+            st.write(f"**Hemoglobin:** {visit_data.iloc[0]['Hemoglobin']}")
+            st.write(f"**Other Notes:** {visit_data.iloc[0]['Other Notes']}")
         else:
-            st.write("No patient data available.")
+            st.write("No visit data available for the selected date.")
+    else:
+        st.write("No patient data available.")
 
-    elif action == "Create New Patient":
-        st.header("Create New Patient Account")
-        with st.form("new_patient_form"):
-            new_patient_id = st.text_input("Patient ID")
-            new_patient_name = st.text_input("Patient Name")
-            new_dob = st.date_input("Date of Birth")
-            new_age = (dt.datetime.now() - pd.to_datetime(new_dob)).days // 365
-            new_gender = st.selectbox("Gender", ["Male", "Female", "Other"])
-            new_med_history = st.text_area("Medical History")
-            new_allergies = st.text_area("Allergies")
-            new_submitted = st.form_submit_button("Add New Patient")
-            if new_submitted:
-                new_patient = {
-                    "Patient ID": new_patient_id,
-                    "Patient Name": new_patient_name,
-                    "Date of Birth": new_dob,
-                    "Age": new_age,
-                    "Gender": new_gender,
-                    "Medical History": new_med_history,
-                    "Allergies": new_allergies,
-                    "Visit Date": None,
-                    "Complaint": None,
-                    "Physical Examination": None,
-                    "Systolic BP": None,
-                    "Diastolic BP": None,
-                    "Temperature": None,
-                    "Glucose": None,
-                    "Cholesterol": None,
-                    "Hemoglobin": None,
-                    "Other Notes": None,
-                }
-                df = pd.concat([df, pd.DataFrame([new_patient])], ignore_index=True)
-
-                # Save the updated data to CSV
-                buffer = io.StringIO()
-                df.to_csv(buffer, index=False)
-                buffer.seek(0)
-                bytes_data = io.BytesIO(buffer.getvalue().encode('utf-8'))
-                st.success("New patient account created successfully!")
+elif action == "Create New Patient":
+    st.header("Create New Patient Account")
+    with st.form("new_patient_form"):
+        new_patient_id = st.text_input("Patient ID")
+        new_patient_name = st.text_input("Patient Name")
+        new_dob = st.date_input("Date of Birth")
+        new_age = (dt.datetime.now() - pd.to_datetime(new_dob)).days // 365
+        new_gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+        new_med_history = st.text_area("Medical History")
+        new_allergies = st.text_area("Allergies")
+        new_submitted = st.form_submit_button("Add New Patient")
+        if new_submitted:
+            new_patient = {
+                "Patient ID": new_patient_id,
+                "Patient Name": new_patient_name,
+                "Date of Birth": new_dob,
+                "Age": new_age,
+                "Gender": new_gender,
+                "Medical History": new_med_history,
+                "Allergies": new_allergies,
+                "Visit Date": None,
+                "Complaint": None,
+                "Physical Examination": None,
+                "Systolic BP": None,
+                "Diastolic BP": None,
+                "Temperature": None,
+                "Glucose": None,
+                "Cholesterol": None,
+                "Hemoglobin": None,
+                "Other Notes": None,
+            }
+            df = pd.concat([df, pd.DataFrame([new_patient])], ignore_index=True)
 
-        # Download button (moved outside the form)
-        st.download_button(
-            label="Download Updated Data",
-            data=bytes_data,
-            file_name="updated_patient_data.csv",
-            mime="text/csv"
-        )
+            # Save the updated data to CSV
+            save_data(df)
+            st.success("New patient account created successfully!")