Update app.py

app.py

@@ -1,9 +1,10 @@
 import streamlit as st
+from fpdf import FPDF
 
 # Set page config
 st.set_page_config(page_title="BMI Calculator", page_icon="⚖️", layout="centered")
 
-# Apply custom CSS for fonts and style
+# Apply custom CSS
 st.markdown("""
     <style>
     .main {
@@ -30,8 +31,17 @@ st.markdown("---")
 with st.container():
     st.markdown("<div class='card'>", unsafe_allow_html=True)
 
-    # Inputs
-    st.subheader("Enter your details:")
+    # Patient Information
+    st.subheader("Patient Information")
+    patient_name = st.text_input("Name of Patient:")
+    patient_age = st.number_input("Age:", min_value=0, max_value=120, step=1)
+    patient_gender = st.selectbox("Gender:", ["Select", "Male", "Female", "Other"])
+    patient_notes = st.text_area("Additional Notes:")
+
+    st.markdown("---")
+    
+    # BMI Calculation Section
+    st.subheader("Body Measurements")
 
     weight = st.number_input("Weight (in kilograms):", min_value=1.0, format="%.2f")
 
@@ -42,7 +52,6 @@ with st.container():
     with col2:
         inches = st.number_input("Inches:", min_value=0, format="%i")
 
-    # Calculate BMI live
     def calculate_bmi(weight, feet, inches):
         total_inches = (feet * 12) + inches
         height_meters = total_inches * 0.0254
@@ -57,17 +66,57 @@ with st.container():
         st.markdown("---")
         st.subheader("Result:")
 
-        # Show result with different colors
+        bmi_status = ""
         if bmi < 18.5:
             st.success(f"Your BMI is {bmi} — Underweight.")
+            bmi_status = "Underweight"
         elif 18.5 <= bmi < 24.9:
             st.success(f"Your BMI is {bmi} — Normal weight. 👍")
+            bmi_status = "Normal weight"
         elif 25 <= bmi < 29.9:
             st.warning(f"Your BMI is {bmi} — Overweight.")
+            bmi_status = "Overweight"
         else:
             st.error(f"Your BMI is {bmi} — Obese.")
-
+            bmi_status = "Obese"
     else:
         st.info("Please enter your height and weight.")
 
+    st.markdown("---")
+    
+    # Function to generate PDF
+    def create_pdf(name, age, gender, notes, bmi_value, bmi_status):
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", 'B', 16)
+        pdf.cell(0, 10, "BMI Report", ln=True, align="C")
+        pdf.ln(10)
+
+        pdf.set_font("Arial", '', 12)
+        pdf.cell(0, 10, f"Patient Name: {name}", ln=True)
+        pdf.cell(0, 10, f"Age: {age}", ln=True)
+        pdf.cell(0, 10, f"Gender: {gender}", ln=True)
+        pdf.cell(0, 10, f"BMI: {bmi_value} ({bmi_status})", ln=True)
+        pdf.ln(5)
+        pdf.multi_cell(0, 10, f"Notes: {notes}")
+
+        return pdf
+
+    # Only show download if BMI is calculated
+    if bmi and patient_name:
+        pdf = create_pdf(patient_name, patient_age, patient_gender, patient_notes, bmi, bmi_status)
+        
+        # Save PDF temporarily in memory
+        import io
+        pdf_buffer = io.BytesIO()
+        pdf.output(pdf_buffer)
+        pdf_buffer.seek(0)
+
+        st.download_button(
+            label="📄 Download BMI Report as PDF",
+            data=pdf_buffer,
+            file_name=f"{patient_name}_BMI_Report.pdf",
+            mime="application/pdf"
+        )
+
     st.markdown("</div>", unsafe_allow_html=True)