Update app.py

app.py

@@ -486,11 +486,11 @@ if st.button("Generate Audio", key="generate_audio_button"):
     else:
         st.info("Description field is required.")
 
-# Display the login interface if needed
-if st.session_state.get('login_needed'):
+# Display the login interface if login_needed and user is not logged in
+if st.session_state.get('login_needed') and st.session_state.get("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")
+    password = st.text_input("Password", type="password", key="password_input", value="")
 
     if st.button("Log in", key="login_button"):
         login_url = "https://songlabai.com/wp-json/jwt-auth/v1/token"
@@ -505,7 +505,7 @@ if st.session_state.get('login_needed'):
             st.session_state["session_id"] = str(uuid.uuid4())  # Generate new session ID
             st.success("Logged in successfully!")
             st.session_state['login_needed'] = False
-            # Retrieve the parameters and generate audio
+            # Retrieve the parameters and generate audio if they were stored
             params = st.session_state.get('generate_audio_params')
             if params:
                 generate_audio(**params)
@@ -515,7 +515,6 @@ if st.session_state.get('login_needed'):
             # Keep login_needed True
     # Add Register link
     st.markdown("Don't have an account? [Register here](https://songlabai.com/register/)")
-
 else:
     pass  # Do nothing if login is not needed