Update app.py

app.py

@@ -93,12 +93,12 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
     # ======================
     # Tabs: Login / Signup
     # ======================
-    with gr.Tabs():
+    with gr.Tabs() as tabs:
 
         # -------------------
         # LOGIN TAB
         # -------------------
-        with gr.Tab("Login"):
+        with gr.Tab("Login") as login_tab:
             log_user = gr.Textbox(label="Username")
             log_pw = gr.Textbox(label="Password", type="password")
             log_status = gr.Textbox(label="Status", interactive=False)
@@ -123,7 +123,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
         chatbot = gr.Chatbot(type="messages")
         msg = gr.Textbox(placeholder="Таны асуулт...", lines=1)
         clear = gr.Button("🗑️ Clear History")
-        msg.submit(chat_fn, inputs=[msg, chatbot], outputs=[chatbot, chatbot])
+        msg.submit(chat_fn, [msg, chatbot], [chatbot, chatbot])
         clear.click(lambda: [], None, chatbot)
 
     # ======================
@@ -141,13 +141,14 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
     def login_action(username, password):
         res = login(username, password)
         if res == "SUCCESS":
-            return "✅ Амжилттай нэвтэрлээ!", gr.update(visible=True)
-        return res, gr.update(visible=False)
+            # Hide login/signup tabs & show chat UI
+            return "✅ Амжилттай нэвтэрлээ!", gr.update(visible=False), gr.update(visible=True)
+        return res, gr.update(visible=True), gr.update(visible=False)
 
     log_btn.click(
         login_action,
         inputs=[log_user, log_pw],
-        outputs=[log_status, chat_ui]
+        outputs=[log_status, tabs, chat_ui]
     )
 
 # ======================