Update app.py

app.py

@@ -338,7 +338,7 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
     with gr.Row():
         delete_last_btn = gr.Button("🗑️ Delete Last Message", variant="stop", size="sm")
         edit_last_btn = gr.Button("✏️ Edit & Re-run Last Prompt", variant="secondary", size="sm")
-        top_btn = gr.Button("⏹️ Stop Generation", variant="stop", visible=False)
+        stop_btn = gr.Button("⏹️ Stop Generation", variant="stop", visible=False)
     
         # Token display (optional but useful)
     token_display = gr.Markdown("**Tokens used this chat:** 0", visible=False)
@@ -362,11 +362,26 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
 # ================== EVENTS ==================
     load_btn.click(load_model, model_dropdown, [current_model, status_label])
     
-    new_chat_btn.click(create_new_chat, [user_chats_state, logged_in_username], [user_chats_state, current_chat_title, chatbot, chat_dropdown, status_label, current_chat_context])
-    
-    delete_chat_btn.click(delete_current_chat, [current_chat_title, user_chats_state, logged_in_username], [user_chats_state, current_chat_title, chatbot, chat_dropdown, status_label, current_chat_context])
-    
-    chat_dropdown.change(switch_chat, [chat_dropdown, user_chats_state], [chatbot, current_chat_title, current_chat_context])
+# New Chat
+    new_chat_btn.click(
+        create_new_chat, 
+        [user_chats_state, logged_in_username], 
+        [user_chats_state, current_chat_title, chatbot, chat_dropdown, status_label, current_chat_context]
+    )
+
+    # Delete Current Chat
+    delete_chat_btn.click(
+        delete_current_chat, 
+        [current_chat_title, user_chats_state, logged_in_username], 
+        [user_chats_state, current_chat_title, chatbot, chat_dropdown, status_label, current_chat_context]
+    )
+
+    # Switch Chat
+    chat_dropdown.change(
+        switch_chat, 
+        [chat_dropdown, user_chats_state], 
+        [chatbot, current_chat_title, current_chat_context]
+    )
     
     register_btn.click(
         register_user,