Update app.py

app.py

@@ -47,37 +47,55 @@ st.markdown(
     """,
     unsafe_allow_html=True
 )
-# ---------- Password Gate (add this near the top, before any model loading) ----------
-import os, streamlit as st
-
-def _require_password():
-    """Stop the app until the correct password is entered.
-       Password is read from environment variable APP_PASSWORD.
-       If APP_PASSWORD is not set, the gate is disabled (app is open)."""
-    expected = os.environ.get("APP_PASSWORD", "").strip()
-    if not expected:
-        return  # gate disabled (no password configured)
-
-    # already authenticated this session?
-    if st.session_state.get("_authed", False):
-        return
-
-    st.title("🔒 Protected")
-    st.write("This app requires a password to continue.")
-
-    with st.form("pw_form", clear_on_submit=False):
-        pw = st.text_input("Password", type="password")
-        ok = st.form_submit_button("Enter")
-    if ok:
-        if pw == expected:
-            st.session_state["_authed"] = True
-            st.rerun()
-        else:
-            st.error("Incorrect password. Please try again.")
-    st.stop()  # block the rest of the script until authenticated
+# --- Password gate (branded) ---
+def add_password_gate() -> bool:
+    """
+    Shows a branded access screen until the correct password is entered.
+    Returns True when access is granted (or no password is configured).
+    """
+    # Where the password is stored
+    try:
+        required = st.secrets.get("APP_PASSWORD", "")
+    except Exception:
+        required = os.environ.get("APP_PASSWORD", "")
+
+    # If no password configured, don't block access
+    if not required:
+        return True
 
-_require_password()
-# -------------------------------------------------------------------------------
+    # Already authenticated?
+    if st.session_state.get("auth_ok", False):
+        return True
+
+    # --- Branded header -------------------------------------------------------
+    st.markdown(
+        f"""
+        <div style="display:flex;align-items:center;gap:14px;margin:8px 0 6px 0;">
+          <img src="{inline_logo()}" style="width:56px;height:56px;object-fit:contain"/>
+          <div>
+            <div style="font-size:1.9rem;font-weight:800;">ST_GeoMech_UCS</div>
+            <div style="color:#667085;">Smart Thinking • Secure Access</div>
+          </div>
+        </div>
+        <div style="font-size:1.25rem;font-weight:700;margin:8px 0 4px 0;">Protected Area</div>
+        <div style="color:#6b7280;margin-bottom:14px;">
+          Please enter your access key to continue.
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+    # Input + action
+    pwd = st.text_input("Access key", type="password", placeholder="••••••••")
+    col1, _ = st.columns([1, 3])
+    with col1:
+        if st.button("Unlock", type="primary", use_container_width=True):
+            if pwd == required:
+                st.session_state.auth_ok = True
+                st.rerun()
+            else:
+                st.error("Incorrect key. Please try again.")
+    st.stop()
 
 # =========================
 # Helpers
@@ -404,7 +422,7 @@ if st.session_state.app_step == "intro":
     st.markdown(
         "This software is developed by *Smart Thinking AI-Solutions Team* to estimate UCS from drilling data."
     )
-    st.subheader("Required Input Columns")
+    st.subheader("Expected Input Features (in Order)")
     st.markdown(
         "- Q, gpm — Flow rate (gallons per minute)  \n"
         "- SPP(psi) — Stand pipe pressure  \n"