Update app.py

app.py

@@ -156,7 +156,6 @@ st.title("Songlab AI")
 # Ensure session state keys are initialized
 if "jwt_token" not in st.session_state:
     st.session_state["jwt_token"] = None
-
 if "session_id" not in st.session_state:
     st.session_state["session_id"] = None
 
@@ -164,6 +163,7 @@ if st.session_state["jwt_token"] is None:
     st.header("Please log in to continue")
     username = st.text_input("Username", key="username_input")
     password = st.text_input("Password", type="password", key="password_input")
+
     if st.button("Log in", key="login_button"):
         login_url = "https://songlabai.com/wp-json/jwt-auth/v1/token"
         data = {
@@ -178,15 +178,9 @@ if st.session_state["jwt_token"] is None:
             st.success("Logged in successfully!")
             st.rerun()
         else:
-             st.error("Invalid username or password")
-             st.stop()
-
-    '''with col2:
-        if st.button("Don't have an account?", key="register_button"):
-            js = "window.location.href = 'https://songlabai.com/register/';"
-            html = '<script>{}</script>'.format(js)
-            st.components.v1.html(html)
-        st.stop()'''
+            st.error("Invalid username or password")
+            st.stop()
+    st.stop()
 else:
     # User is logged in, continue with the app
     if st.button("Log out", key="logout_button"):
@@ -196,6 +190,7 @@ else:
         st.rerun()
 
 
+
 # Function to get the auth headers using the current jwt_token
 def get_auth_headers():
     jwt_token = st.session_state.get("jwt_token")