Update app.py

app.py

@@ -23,20 +23,25 @@ st.subheader("Chatbot", divider="rainbow", anchor=False)
 # LOGIN
 password = os.environ['PASSWORD']
 
+# Create a login form container
+login_container = st.container()
+
 def login(password_input):
     if password_input == password:
         st.session_state.authenticated = True
-        return  # Exit the function after successful login
-
-    st.error("Invalid username or password")
+        login_container.empty()  # Clear the login form container
+    else:
+        st.error("Invalid password")
 
-login_container = st.container()
+with login_container:
+    password_input = st.text_input("Password", type="password")
+    st.button("authenticate", on_click=login(password_input))
 
-# Password input and button inside the conditional block
+# Only allow access to the app if the user is authenticated
 if "authenticated" not in st.session_state or not st.session_state.authenticated:
-    with login_container:
-        password_input = st.text_input("Password", type="password")
-        st.button("authenticate", on_click=login, args=password_input)
+    st.stop()  # Stop the app from running if the user is not authenticated
+else:
+    login_container.empty()
 
 client = Groq(
     api_key=os.environ['GROQ_API_KEY'],