passwords

.gitignore

@@ -0,0 +1 @@
+.streamlit

app.py

@@ -1,3 +1,4 @@
+import hmac
 import os
 import tempfile
 
@@ -15,6 +16,34 @@ from torch.utils.data import DataLoader
 from transformers import AutoProcessor
 
 
+def check_password():
+    """Returns `True` if the user had the correct password."""
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if hmac.compare_digest(st.session_state["password"], st.secrets["password"]):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # Don't store the password.
+        else:
+            st.session_state["password_correct"] = False
+
+    # Return True if the password is validated.
+    if st.session_state.get("password_correct", False):
+        return True
+
+    # Show input for password.
+    st.text_input(
+        "Password", type="password", on_change=password_entered, key="password"
+    )
+    if "password_correct" in st.session_state:
+        st.error("😕 Password incorrect")
+    return False
+
+
+if not check_password():
+    st.stop()  # Do not continue if check_password is not True.
+
+
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 SS = st.session_state