Update app.py

app.py

@@ -116,11 +116,9 @@ def add_password_gate() -> None:
     if st.session_state.get("auth_ok", False):
         return
 
-    # st.sidebar.image("logo.png", use_column_width=False)
-    # st.sidebar.markdown("### ST_GeoMech_UCS\nSmart Thinking • Secure Access")
     st.sidebar.markdown(f"""
         <div class="centered-container">
-            <img src="{inline_logo('logo.png')}" style="width: 16px; height: auto; object-fit: contain;">
+            <img src="{inline_logo('logo.png')}" style="width: 50px; height: auto; object-fit: contain;">
             <div style='font-weight:800;font-size:1.2rem; margin-top: 10px;'>ST_GeoMech_UCS</div>
             <div style='color:#667085;'>Smart Thinking • Secure Access</div>
         </div>
@@ -409,21 +407,40 @@ st.session_state.setdefault("dev_preview",False)
 # =========================
 # Branding in Sidebar
 # =========================
-# st.sidebar.image("logo.png", use_column_width=False)
-# st.sidebar.markdown(
-#     "<div style='font-weight:800;font-size:1.2rem;'>ST_GeoMech_UCS</div>"
-#     "<div style='color:#667085;'>Real-Time UCS Tracking While Drilling</div>",
-#     unsafe_allow_html=True
-# )
 st.sidebar.markdown(f"""
     <div class="centered-container">
-        <img src="{inline_logo('logo.png')}" style="width: 16px; height: auto; object-fit: contain;">
+        <img src="{inline_logo('logo.png')}" style="width: 64px; height: auto; object-fit: contain;">
         <div style='font-weight:800;font-size:1.2rem;'>ST_GeoMech_UCS</div>
         <div style='color:#667085;'>Real-Time UCS Tracking While Drilling</div>
     </div>
     """, unsafe_allow_html=True
 )
 
+# =========================
+# Reusable Sticky Header Function
+# =========================
+def sticky_header(title, message):
+    st.markdown(
+        f"""
+        <style>
+        .sticky-container {{
+            position: sticky;
+            top: 0;
+            background-color: white;
+            z-index: 100;
+            padding-top: 10px;
+            padding-bottom: 10px;
+            border-bottom: 1px solid #eee;
+        }}
+        </style>
+        <div class="sticky-container">
+            <h3>{title}</h3>
+            <p>{message}</p>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
 # =========================
 # INTRO
 # =========================
@@ -464,15 +481,13 @@ if st.session_state.app_step == "dev":
     if st.sidebar.button("Proceed to Validation ▶", use_container_width=True): st.session_state.app_step="validate"; st.rerun()
     if st.sidebar.button("Proceed to Prediction ▶", use_container_width=True): st.session_state.app_step="predict"; st.rerun()
 
-    helper_top = st.container()
-    with helper_top:
-        st.subheader("Case Building")
-        if st.session_state.dev_file_loaded and st.session_state.dev_preview:
-            st.info("Previewed ✓ — now click **Run Model**.")
-        elif st.session_state.dev_file_loaded:
-            st.info("📄 **Preview uploaded data** using the sidebar button, then click **Run Model**.")
-        else:
-            st.write("**Upload your data to build a case, then run the model to review development performance.**")
+    # Apply sticky header
+    if st.session_state.dev_file_loaded and st.session_state.dev_preview:
+        sticky_header("Case Building", "Previewed ✓ — now click **Run Model**.")
+    elif st.session_state.dev_file_loaded:
+        sticky_header("Case Building", "📄 **Preview uploaded data** using the sidebar button, then click **Run Model**.")
+    else:
+        sticky_header("Case Building", "**Upload your data to build a case, then run the model to review development performance.**")
 
     if run and st.session_state.dev_file_bytes:
         book = read_book_bytes(st.session_state.dev_file_bytes)
@@ -541,8 +556,7 @@ if st.session_state.app_step == "validate":
     if st.sidebar.button("⬅ Back to Case Building", use_container_width=True): st.session_state.app_step="dev"; st.rerun()
     if st.sidebar.button("Proceed to Prediction ▶", use_container_width=True): st.session_state.app_step="predict"; st.rerun()
 
-    st.subheader("Validate the Model")
-    st.write("Upload a dataset with the same **features** and **UCS** to evaluate performance.")
+    sticky_header("Validate the Model", "Upload a dataset with the same **features** and **UCS** to evaluate performance.")
 
     if go_btn and up is not None:
         book = read_book_bytes(up.getvalue())
@@ -609,8 +623,7 @@ if st.session_state.app_step == "predict":
     go_btn = st.sidebar.button("Predict", type="primary", use_container_width=True)
     if st.sidebar.button("⬅ Back to Case Building", use_container_width=True): st.session_state.app_step="dev"; st.rerun()
 
-    st.subheader("Prediction")
-    st.write("Upload a dataset with the feature columns (no **UCS**).")
+    sticky_header("Prediction", "Upload a dataset with the feature columns (no **UCS**).")
 
     if go_btn and up is not None:
         book = read_book_bytes(up.getvalue()); name = list(book.keys())[0]