Update app.py

app.py

@@ -23,23 +23,20 @@ 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
-        login_container.empty()  # Clear the login form container
-    else:
-        st.error("Invalid username or password")
+        return  # Exit the function after successful login
 
-with login_container:
-    password_input = st.text_input("Password", type="password")
-    st.button("authenticate", on_click=login(password_input))
+    st.error("Invalid username or password")
+
+login_container = st.container()
 
-# Only allow access to the app if the user is authenticated
+# Password input and button inside the conditional block
 if "authenticated" not in st.session_state or not st.session_state.authenticated:
-    st.stop()  # Stop the app from running if the user is not authenticated
+    with login_container:
+        password_input = st.text_input("Password", type="password")
+        st.button("authenticate", on_click=login, args=password_input)
 
 client = Groq(
     api_key=os.environ['GROQ_API_KEY'],