Update app.py

app.py

@@ -31,96 +31,102 @@ if uploaded_file:
 
         # Display patient data
         patient_data = df[df['Patient ID'] == patient_id]
-        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 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']}")
 
-        # 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)
+            # 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)
 
-        # 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)
+            # 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)
 
-                # 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!")
+                    # 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!")
 
-                # Reload data
-                df = load_data(io.StringIO(buffer.getvalue()))
-                patient_data = df[df['Patient ID'] == patient_id]
+                    # 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",
-            data=bytes_data,
-            file_name="updated_patient_data.csv",
-            mime="text/csv"
-        )
+            # 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"
+            )
 
-        # 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)]
-        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']}")
+            # 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.")
+        else:
+            st.write("No patient data available.")
 
     elif action == "Create New Patient":
         st.header("Create New Patient Account")