removed reevaluation button

pages/Model_Evaluation.py

@@ -16,13 +16,16 @@ import streamlit as st
 import matplotlib.pyplot as plt
 from fpdf import FPDF
 
-# ---- Stop Evaluation button -----
+# ---- Streamlit State Initialization ----
 if 'stop_eval' not in st.session_state:
     st.session_state.stop_eval = False
 
 if 'evaluation_done' not in st.session_state:
     st.session_state.evaluation_done = False
 
+if 'trigger_eval' not in st.session_state:
+    st.session_state.trigger_eval = False
+
 # ---- Streamlit Title ----
 st.markdown("<h2 style='color: #2E86C1;'>📈 Model Evaluation</h2>", unsafe_allow_html=True)
 
@@ -106,7 +109,7 @@ def load_model():
     model.eval()
     return model
 
-# ---- Main Evaluation ----
+# ---- Main UI Buttons ----
 csv_path = r"D:\\DR_Classification\\splits\\test_labels.csv"
 model = load_model()
 test_loader = load_test_data(csv_path)
@@ -117,9 +120,7 @@ with col1:
     if st.button("🚀 Start Evaluation"):
         st.session_state.stop_eval = False
         st.session_state.evaluation_done = False
-        run_eval = True
-    else:
-        run_eval = False
+        st.session_state.trigger_eval = True
 
 with col2:
     if st.button("🚩 Stop Evaluation"):
@@ -128,7 +129,8 @@ with col2:
 if st.session_state.evaluation_done:
     reevaluate_col, download_col = st.columns([1, 1])
 
-if run_eval or st.session_state.evaluation_done:
+# ---- Evaluation Logic ----
+if st.session_state.trigger_eval:
     st.markdown("### ⏱️ Evaluation Results")
 
     start_time = time.time()
@@ -168,6 +170,7 @@ if run_eval or st.session_state.evaluation_done:
 
     if not st.session_state.stop_eval:
         st.session_state.evaluation_done = True
+        st.session_state.trigger_eval = False  # ✅ Reset the trigger
         st.success(f"✅ Evaluation completed in **{eval_time:.2f} seconds**")
 
         report = classification_report(y_true, y_pred, target_names=class_names, output_dict=True)
@@ -241,7 +244,3 @@ if run_eval or st.session_state.evaluation_done:
             reevaluate_col, download_col = st.columns([1, 1])
             with download_col:
                 st.download_button("📄 Download Full Evaluation PDF", f, file_name="evaluation_report.pdf")
-            with reevaluate_col:
-                if st.button("🔁 Re-evaluate"):
-                    st.session_state.stop_eval = False
-                    st.session_state.evaluation_done = False