Update app.py

app.py

@@ -421,7 +421,7 @@ def get_dropdown_choices(info_dict):
 
 def login(username, password):
     if username == USERNAME and password == PASSWORD:
-        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)
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(value="Login successful! You can now access the Caption Captain tab.", visible=True)
     else:
         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)
 
@@ -449,27 +449,28 @@ with gr.Blocks(theme="Hev832/Applio", css=css, fill_width=True, fill_height=True
                     """
                 )
 
-                with gr.Column(scale=1):
-                    with gr.Row():
-                        gr.Markdown(
-                            """
-                            Login below using the internal<br>
-                            username and password to access the full app.<br>
-                            
-                            Once logged in, a new tab will appear named<br>
-                            "QR Code Art Generator" allowing you to access.
-                            """
-                        )
-
-                    with gr.Row():
-                        username = gr.Textbox(label="Username", placeholder="Enter your username", value="ugd")
-                    with gr.Row():
-                        password = gr.Textbox(label="Password", type="password", placeholder="Enter your password", value="ugd!")
-                    with gr.Row():
-                        login_button = gr.Button("Login", size="sm")
-                    login_message = gr.Markdown(visible=False)
-
-    with gr.Tab("Caption Captain") as app_container:
+            with gr.Column(scale=1):
+                with gr.Row():
+                    gr.Markdown(
+                        """
+                        Login below using the internal<br>
+                        username and password to access the full app.<br>
+                        
+                        Once logged in, a new tab will appear named<br>
+                        "Caption Captain" allowing you to access the app.
+                        """
+                    )
+
+                with gr.Row():
+                    username = gr.Textbox(label="Username", placeholder="Enter your username")
+                with gr.Row():
+                    password = gr.Textbox(label="Password", type="password", placeholder="Enter your password")
+                with gr.Row():
+                    login_button = gr.Button("Login", size="sm")
+                login_message = gr.Markdown(visible=False)
+
+    caption_captain_tab = gr.Tab("Caption Captain", visible=False)
+    with caption_captain_tab:
         with gr.Accordion("How to Use Caption Captain", open=False):
             gr.Markdown("""
             # How to Use Caption Captain
@@ -596,6 +597,11 @@ with gr.Blocks(theme="Hev832/Applio", css=css, fill_width=True, fill_height=True
         outputs=[error_message, output_caption]
     )
 
+    login_button.click(
+        login,
+        inputs=[username, password],
+        outputs=[caption_captain_tab, username, password, login_message]
+    )
 
 if __name__ == "__main__":
     demo.launch()