Update app.py

app.py

@@ -54,9 +54,9 @@ st.markdown(
 # Helpers
 # =========================
 
-# Modal dialog fallback for older Streamlit
+# Prefer real dialog (Streamlit ≥1.31). Fallback to expander if older.
 try:
-    dialog = st.dialog  # Streamlit ≥1.31
+    dialog = st.dialog
 except AttributeError:
     def dialog(title):
         def deco(fn):
@@ -216,18 +216,43 @@ def stats_table(df: pd.DataFrame, cols: list[str]) -> pd.DataFrame:
     return out
 
 @dialog("Preview data")
-def preview_modal(book: dict[str, pd.DataFrame], default_cols: list[str]):
+def preview_modal_dev(book: dict[str, pd.DataFrame], feature_cols: list[str]):
     if not book:
         st.info("No data loaded yet.")
         return
-    sheet = st.selectbox("Sheet", list(book.keys()), index=0)
-    df = book[sheet]
-    present = [c for c in default_cols if c in df.columns]
+    # Auto detect Train/Test-like sheets
+    sh_train = find_sheet(book, ["Train","Training","training2","train","training"])
+    sh_test  = find_sheet(book, ["Test","Testing","testing2","test","testing"])
+    tabs = []
+    data = []
+    if sh_train: tabs.append("Train"); data.append(book[sh_train])
+    if sh_test:  tabs.append("Test");  data.append(book[sh_test])
+    if not tabs:  # nothing matched, just show the first sheet
+        first_name = list(book.keys())[0]
+        tabs = [first_name]; data = [book[first_name]]
+
+    st.write("Use the tabs to switch between Train/Test views (if available).")
+    t_objs = st.tabs(tabs)
+    for t, df in zip(t_objs, data):
+        with t:
+            tab_tracks, tab_summary = st.tabs(["Tracks", "Summary"])
+            with tab_tracks:
+                st.pyplot(make_index_tracks(df, feature_cols), use_container_width=True)
+            with tab_summary:
+                st.dataframe(stats_table(df, feature_cols), use_container_width=True)
+
+@dialog("Preview data")
+def preview_modal_val(book: dict[str, pd.DataFrame], feature_cols: list[str]):
+    if not book:
+        st.info("No data loaded yet.")
+        return
+    vname = find_sheet(book, ["Validation","Validate","validation2","Val","val"]) or list(book.keys())[0]
+    df = book[vname]
     tab_tracks, tab_summary = st.tabs(["Tracks", "Summary"])
     with tab_tracks:
-        st.pyplot(make_index_tracks(df, present), use_container_width=True)
+        st.pyplot(make_index_tracks(df, feature_cols), use_container_width=True)
     with tab_summary:
-        st.dataframe(stats_table(df, present), use_container_width=True)
+        st.dataframe(stats_table(df, feature_cols), use_container_width=True)
 
 # =========================
 # Model presence (local or optional download)
@@ -287,7 +312,7 @@ if ("Train" in st.session_state.results) or ("Test" in st.session_state.results)
     st.session_state.dev_ready = True
 
 # =========================
-# Hero header (logo + title)
+# Hero header (logo + title) — website link moved to footer
 # =========================
 st.markdown(
     f"""
@@ -332,17 +357,27 @@ if st.session_state.app_step == "intro":
 # =========================
 if st.session_state.app_step == "dev":
     st.sidebar.header("Model Development Data")
-    train_test_file = st.sidebar.file_uploader("Upload Data (Excel)", type=["xlsx","xls"], key="dev_upload")
-
-    # Preview button (uses modal)
+    dev_label = "Upload Data (Excel)"
+    if st.session_state.get("dev_file_name"):
+        dev_label = "Replace data (Excel)"
+    train_test_file = st.sidebar.file_uploader(dev_label, type=["xlsx","xls"], key="dev_upload")
+    if train_test_file is not None:
+        st.session_state.dev_file_name = train_test_file.name
+        # small status line in sidebar
+        _book_tmp = read_book(train_test_file)
+        if _book_tmp:
+            first_df = next(iter(_book_tmp.values()))
+            st.sidebar.caption(f"**Data loaded:** {train_test_file.name} • {first_df.shape[0]} rows × {first_df.shape[1]} cols")
+
+    # Preview button (modal)
     preview_btn = st.sidebar.button("Preview data", use_container_width=True, disabled=(train_test_file is None))
     if preview_btn and train_test_file is not None:
         _book = read_book(train_test_file)
-        preview_modal(_book, FEATURES)
+        preview_modal_dev(_book, FEATURES)
 
     run_btn = st.sidebar.button("Run Model", type="primary", use_container_width=True)
 
-    # Proceed button BELOW run, always visible; enables immediately after first successful run
+    # Proceed button BELOW run, always visible; enables after first successful run
     st.sidebar.button(
         "Proceed to Prediction ▶",
         use_container_width=True,
@@ -352,7 +387,7 @@ if st.session_state.app_step == "dev":
 
     # Header + helper sentence under the header
     st.subheader("Model Development")
-    st.write("Upload your data to train the model and review the development performance.")
+    st.write("**Upload your data to build a case, run the model, and review the development performance.**")
 
     if run_btn and train_test_file is not None:
         with st.status("Processing…", expanded=False) as status:
@@ -440,13 +475,22 @@ if st.session_state.app_step == "dev":
 # =========================
 if st.session_state.app_step == "predict":
     st.sidebar.header("Prediction (Validation)")
-    validation_file = st.sidebar.file_uploader("Upload Validation Excel", type=["xlsx","xls"], key="val_upload")
+    val_label = "Upload Validation Excel"
+    if st.session_state.get("val_file_name"):
+        val_label = "Replace data (Excel)"
+    validation_file = st.sidebar.file_uploader(val_label, type=["xlsx","xls"], key="val_upload")
+    if validation_file is not None:
+        st.session_state.val_file_name = validation_file.name
+        _book_tmp = read_book(validation_file)
+        if _book_tmp:
+            first_df = next(iter(_book_tmp.values()))
+            st.sidebar.caption(f"**Data loaded:** {validation_file.name} • {first_df.shape[0]} rows × {first_df.shape[1]} cols")
 
     # Preview button for validation data
     preview_val_btn = st.sidebar.button("Preview data", use_container_width=True, disabled=(validation_file is None))
     if preview_val_btn and validation_file is not None:
         _book = read_book(validation_file)
-        preview_modal(_book, FEATURES)
+        preview_modal_val(_book, FEATURES)
 
     predict_btn = st.sidebar.button("Predict", type="primary", use_container_width=True)
     st.sidebar.button("⬅ Back", on_click=lambda: st.session_state.update(app_step="dev"), use_container_width=True)
@@ -546,7 +590,13 @@ if st.session_state.app_step == "predict":
             st.warning(str(e))
 
 # =========================
-# Footer
+# Footer (with website link)
 # =========================
 st.markdown("---")
-st.markdown("<div style='text-align:center; color:#6b7280;'>ST_GeoMech_UCS • © Smart Thinking</div>", unsafe_allow_html=True)
+st.markdown(
+    "<div style='text-align:center; color:#6b7280;'>"
+    "ST_GeoMech_UCS • © Smart Thinking • "
+    "<a href='https://www.smartthinking.com.sa' target='_blank'>smartthinking.com.sa</a>"
+    "</div>",
+    unsafe_allow_html=True
+)