Update app/streamlit_app.py

app/streamlit_app.py

@@ -109,7 +109,7 @@ if page == "Prediction":
         if submit_btn:
             payload = {"Time": st.session_state.t_time, "Amount": st.session_state.t_amount, **v_data}
             
-            # ✅ Save payload for SHAP page
+            # store for SHAP page
             st.session_state["last_payload"] = payload
             
             try:
@@ -136,8 +136,7 @@ if page == "Prediction":
             except Exception as e:
                 st.error(f"⏳ Error connecting to API: {e}")
 
-
-# 🔥 NEW SHAP PAGE
+# --- NEW SHAP PAGE ---
 elif page == "Explainability (SHAP)":
     st.title("🧠 Explainable AI (SHAP)")
 
@@ -164,7 +163,7 @@ elif page == "Explainability (SHAP)":
             except Exception as e:
                 st.error(f"❌ Error: {e}")
 
-
+# --- DRIFT MONITORING (UNCHANGED EXACTLY) ---
 elif page == "Drift Monitoring":
     st.title("📉 Data Drift Monitoring")
     st.markdown("### Monitor model health & trigger verified retraining")
@@ -178,7 +177,7 @@ elif page == "Drift Monitoring":
                     for i in range(1, 29):
                         skewed_data[f"V{i}"] = random.uniform(-15.0, 15.0)
                     save_to_db(skewed_data, pred=1, prob=0.99)
-                st.success("✅ 50 Skewed rows injected!")
+                st.success("✅ 50 Skewed rows injected! Now click 'Run Drift Detection' below.")
 
     st.markdown("---")
 
@@ -197,7 +196,7 @@ elif page == "Drift Monitoring":
                     st.success("✅ Drift report generated!")
                     st.session_state["drift_done"] = True
                 else:
-                    st.warning("⚠️ Not enough data in live DB")
+                    st.warning("⚠️ Not enough data in live DB (Needs 50 rows). Use the Demo Injector above!")
             except Exception as e:
                 st.error(f"⚠️ Error running drift: {e}")
 
@@ -207,10 +206,17 @@ elif page == "Drift Monitoring":
             html = f.read()
         components.html(html, height=800, scrolling=True)
 
+    st.markdown("---")
+    st.subheader("🔁 Human-in-the-Loop Retraining")
+    st.write("Ensure database contains human-verified `Actual_Class` labels before retraining.")
+
     if st.session_state.get("drift_done", False):
-        if st.button("⚡ Retrain Model"):
-            try:
-                retrain()
-                st.success("✅ Model retrained successfully!")
-            except Exception as e:
-                st.error(f"❌ Error: {e}")
+        if st.button("⚡ Retrain Model (Requires Verified Data)"):
+            with st.spinner("Retraining model..."):
+                try:
+                    retrain()
+                    st.success("✅ Model retrained successfully with verified data!")
+                except ValueError as ve:
+                    st.error(f"❌ {ve}")
+                except Exception as e:
+                    st.error(f"❌ Error: {e}")