Update app.py

app.py

@@ -392,12 +392,11 @@ ul, ol {
 }
 """
 
-# Add login function
 def login(username, password):
     if username == USERNAME and password == PASSWORD:
-        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(value="Login successful! You can now access the QR Code Art Generator tab.", visible=True)
     else:
-        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True), gr.update(value="Invalid username or password. Please try again.", visible=True)
 
 # Add login elements to the Gradio interface
 with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True) as blocks:
@@ -438,7 +437,7 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
                     password = gr.Textbox(label="Password", type="password", placeholder="Enter your password")
                 with gr.Row():
                     login_button = gr.Button("Login", size="sm")
-                login_error = gr.Markdown("Invalid username or password. Please try again.", visible=False)
+                login_message = gr.Markdown(visible=False)
 
 
     with gr.Tab("QR Code and Prompt Input", visible=False) as app_container:
@@ -800,19 +799,18 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
     login_button.click(
         login,
         inputs=[username, password],
-        outputs=[app_container, login_error, login_button]
+        outputs=[app_container, login_message, login_button, login_message]
     )
 
     # Define password textbox submit behavior
     password.submit(
         login,
         inputs=[username, password],
-        outputs=[app_container, login_error, login_button]
+        outputs=[app_container, login_message, login_button, login_message]
     )
 
-
 # Load models on launch
 load_models_on_launch()
 
 blocks.queue(max_size=20)
-blocks.launch(share=True, show_api=True)
+blocks.launch(share=False, show_api=True)