With Password

app.py

@@ -4,6 +4,7 @@ import requests
 from bs4 import BeautifulSoup
 import openai
 import os
+import hmac
 
 load_dotenv()
 
@@ -91,6 +92,33 @@ if "article_links" not in st.session_state:
 if "final_articles" not in st.session_state:
     st.session_state["final_articles"] = []
 
+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"], os.environ.get("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.
+
 col1, col2 = st.columns([2, 1])
 
 col1.title("AI journalist")