Update app.py

app.py

@@ -42,6 +42,8 @@ with st.form("patient_form"):
         rxlocation_o = st.selectbox("RxLocation_O (1=Yes, 0=No)", [0, 1], help="Whether RxLocation is O")
         previous_meth_episodes = st.number_input("Previous Meth Episodes", min_value=0, value=0)
         timesince_last_episode = st.number_input("Time Since Last Episode (days)", min_value=0, value=0)
+        ctrxs_episode = st.number_input("CtRxsEpisode", min_value=0, value=0)
+        rpl_theme2 = st.number_input("RPL_THEME2", min_value=0.0, value=0.5, step=0.001)
 
     with col2:
         prior_meth_days_supply = st.number_input("Prior Meth Days Supply (days)", min_value=0, value=0, step=1)
@@ -49,6 +51,7 @@ with st.form("patient_form"):
         moudtype_nalt = st.selectbox("MOUDType_Nalt (1=Yes, 0=No)", [0, 1], help="Whether the patient is on Nalt MOUD")
         rpl_theme3 = st.number_input("RPL_THEME3", min_value=0.0, value=0.5, step=0.001)
         numberof_mhvisits = st.number_input("Number of MH Visits", min_value=0, value=0)
+        ct_days_covered_antidep_episode = st.number_input("CtDaysCoveredAntidepEpisode", min_value=0, value=0)
 
     st.subheader("Calculated Features *")
     col3, col4 = st.columns(2)
@@ -59,11 +62,6 @@ with st.form("patient_form"):
         number_of_previous_meth_episodes = st.number_input("Number of Previous Meth Episodes", min_value=0, value=0)
         number_of_failed_meth_episodes = st.number_input("Number of Failed Meth Episodes", min_value=0, value=0)
 
-    # Additional inputs for missing features
-    ctrxs_episode = st.number_input("CtRxsEpisode", min_value=0, value=0)
-    rpl_theme2 = st.number_input("RPL_THEME2", min_value=0.0, value=0.5, step=0.001)
-    ct_days_covered_antidep_episode = st.number_input("CtDaysCoveredAntidepEpisode", min_value=0, value=0)
-
     # Dynamic calculations
     failed_bup_rate = calculate_failed_rates(number_of_previous_bup_episodes, number_of_failed_bup_episodes)
     failed_meth_rate = calculate_failed_rates(number_of_previous_meth_episodes, number_of_failed_meth_episodes)
@@ -94,7 +92,7 @@ if submitted:
         st.write(f"**Predicted Class:** {'Retention (1)' if predicted_class == 1 else 'No Retention (0)'}")
         st.write(f"**Predicted Probability of Retention:** {predicted_probability:.2%}")
 
-        # Highlight thresholds
+        # Highlight results
         st.markdown("""
             - **Predicted Class:** Based on the highest probability.
             - **Probability:** Shows the likelihood of retention.