allow dynamic tgt selection

src/pages/2_Bulk_Predictions.py

@@ -25,6 +25,18 @@ if uploaded_file:
     if "bulk_input_df" in st.session_state:
         st.session_state.bulk_input_df = st.data_editor(st.session_state.bulk_input_df, num_rows="dynamic")
 
+    # TODO: Define the target column (customize this based on your use case)
+    # target_col = st.session_state.target_col # "GVHD"  # or "Acute GVHD(<100 days)", etc.
+    st.session_state.target_col = st.selectbox(
+        "Select target column to predict:",
+        options=[
+            "GVHD",
+            "Acute GVHD(<100 days)",
+            "Chronic GVHD>100 days",
+        ],
+        index=0
+    )
+    
     if st.button("Predict"):
         if "bulk_input_df" not in st.session_state:
             st.warning("Please preprocess data first.")
@@ -49,17 +61,7 @@ if uploaded_file:
 
             df = st.session_state.bulk_input_df
 
-            # TODO: Define the target column (customize this based on your use case)
-            # target_col = st.session_state.target_col # "GVHD"  # or "Acute GVHD(<100 days)", etc.
-            target_col = st.selectbox(
-                "Select target column to predict:",
-                options=[
-                    "GVHD",
-                    "Acute GVHD(<100 days)",
-                    "Chronic GVHD>100 days",
-                ],
-                index=0
-            )
+            target_col = st.session_state.target_col
             
             # Optional filtering depending on target choice
             if target_col in ["Acute GVHD(<100 days)", "Chronic GVHD>100 days"]: