Update app.py

app.py

@@ -26,17 +26,17 @@ password = os.environ['PASSWORD']
 # Create a login form container
 login_container = st.container()
 
-with login_container:
-    password_input = st.text_input("Password", type="password")
-    st.button("authenticate", on_click=login, args=(password_input))
-
-def login(username, password_input):
+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")
 
+with login_container:
+    password_input = st.text_input("Password", type="password")
+    st.button("authenticate", on_click=login, args=(password_input))
+
 # Only allow access to the app if the user is authenticated
 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