Update app.py

app.py

@@ -10,14 +10,14 @@ if "page" not in st.session_state:
     st.session_state.page = "heart"
 
 st.title("🩺 Disease Prediction App")
-st.markdown("Follow the steps to assess your **Heart**, **Chest**, and **Combined Health Risk**.")
+st.markdown("Follow the steps to assess your *Heart, **Chest, and **Combined Health Risk*.")
 
 # ───────────────────────────── Step 1: Heart Prediction ───────────────────────────── #
 if st.session_state.page == "heart":
+    st.header("❤ Step 1: Heart Disease Prediction")
+
     st.session_state.age = st.slider("Age", 1, 100, 30)
     sex = st.selectbox("Sex", ["Male", "Female"])
-    st.header("❤️ Step 1: Heart Disease Prediction")
-    
     st.session_state.sex = 1 if sex == "Male" else 0
 
     st.session_state.cp = st.selectbox("Chest Pain Type", [0, 1, 2, 3])
@@ -46,6 +46,7 @@ if st.session_state.page == "heart":
 elif st.session_state.page == "chest":
     st.header("🫁 Step 2: Chest Disease Symptoms")
 
+    st.session_state.gender = st.selectbox("Gender", ["Male", "Female"])
     st.session_state.position = st.selectbox("View Position", ["PA", "AP"])
 
     uploaded_file = st.file_uploader("Upload Chest X-ray", type=["jpg", "jpeg", "png"])
@@ -59,47 +60,37 @@ elif st.session_state.page == "chest":
         st.session_state.conditions.append(1 if st.checkbox(question) else 0)
 
     if st.button("Next → Combined Prediction"):
-        if not uploaded_file:
-            st.warning("Please upload a chest X-ray to continue.")
-        elif not any(st.session_state.conditions):
-            st.warning("Please answer at least one clinical question.")
-        else:
+        if uploaded_file:
             st.session_state.page = "combined"
             st.rerun()
+        else:
+            st.warning("Please upload a chest X-ray to continue.")
 
-# ───────────────────────────── Step 3: Combined Prediction ───────────────────────────── #
 # ───────────────────────────── Step 3: Combined Prediction ───────────────────────────── #
 elif st.session_state.page == "combined":
     st.header("🔁 Step 3: Combined Prediction Results")
 
-    # Check required session state values
-    if "position" in st.session_state and "conditions" in st.session_state:
-        chest_result = predict_chest(
-            st.session_state.position,
-            st.session_state.conditions
-        )
-    else:
-        st.error("Missing chest input data. Please complete the chest step first.")
-        if st.button("⬅️ Go Back to Chest Symptoms"):
-            st.session_state.page = "chest"
-            st.rerun()
-        st.stop()
-
-    heart_result = st.session_state.get("heart_result", "Unknown")
+    chest_result = predict_chest(
+        st.session_state.age,
+        st.session_state.gender,
+        st.session_state.position,
+        st.session_state.conditions
+    )
+    heart_result = st.session_state.heart_result
 
     st.image(st.session_state.image, caption="Chest X-ray", use_container_width=True)
 
     if chest_result == "Chest Disease Detected":
-        st.error(f"❌ Chest Disease Prediction: **{chest_result}**")
+        st.error(f"❌ Chest Disease Prediction: *{chest_result}*")
     else:
-        st.success(f"✅ Chest Disease Prediction: **{chest_result}**")
+        st.success(f"✅ Chest Disease Prediction: *{chest_result}*")
 
-    st.success(f"❤️ Heart Disease Prediction: **{heart_result}**")
+    st.success(f"❤ Heart Disease Prediction: *{heart_result}*")
 
     if "Detected" in chest_result or "Detected" in heart_result:
-        st.error("⚠️ Combined Risk: **High** — Immediate attention recommended.")
+        st.error("⚠ Combined Risk: *High* — Immediate attention recommended.")
     else:
-        st.success("✅ Combined Risk: **Low** — You seem to be in good health.")
+        st.success("✅ Combined Risk: *Low* — You seem to be in good health.")
 
     if st.button("🔁 Start Over"):
         st.session_state.clear()