Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,7 @@ import requests, time, regex, re
 from datetime import datetime
 from dotenv import load_dotenv
 import certifi
+import os
 
 load_dotenv()
 
@@ -18,12 +19,12 @@ from src.parameters import (
     params,
     display_chat_history,
     stream_outputs,
+    app_password
     wx_api_key,
     wx_project_id,
     wx_url,
 )
 from src.helper_functions import (
-    check_password,
     setup_watsonxai_client,
     initialize_session_state,
     create_pdf_from_chat,
@@ -40,13 +41,48 @@ st.set_page_config(
 )
 initialize_session_state()
 
+
+def check_password():
+    def password_entered():
+        if st.session_state["password"] == app_password:
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]
+        else:
+            st.session_state["password_correct"] = False
+
+    if "password_correct" not in st.session_state:
+        st.markdown("\n\n")
+        st.text_input(
+            "Enter the password",
+            type="password",
+            on_change=password_entered,
+            key="password",
+        )
+        st.divider()
+        st.info("Designed and developed by Milan Mrdenovic © IBM Norway 2025")
+        return False
+    elif not st.session_state["password_correct"]:
+        st.markdown("\n\n")
+        st.text_input(
+            "Enter the password",
+            type="password",
+            on_change=password_entered,
+            key="password",
+        )
+        st.divider()
+        st.info("Designed and developed by Milan Mrdenovic © IBM Norway 2025")
+        st.error("😕 Password incorrect")
+        return False
+    else:
+        return True
+
+
 if not check_password():
     st.stop()
 
 if "current_page" not in st.session_state:
     st.session_state.current_page = 0
 
-
 wx_client = setup_watsonxai_client(
     api_key=wx_api_key, project_id=wx_project_id, url=wx_url
 )