Update app.py

app.py

@@ -64,6 +64,108 @@ def delete_current_chat(current_title, user_chats, logged_in_username):
     return new_user_chats, new_title, new_history, gr.update(choices=choices, value=new_title), f"🗑️ Deleted {current_title}"
 
 # ================== GUI ==================
-gui_blocks.general()
+with gr.Blocks(title="NVIDIA AI Chat") as demo:
+    gr.Markdown("# 🟢 NVIDIA AI Chat\n**School Project • Pre-approved usernames only • Max 20 Accounts globally**")
+    gr.Markdown(f"{mymsg.welcome_caution()}")
+    
+    with gr.Accordion("📖 How to Use", open=True):
+        gr.Markdown(f"mymsg.welcome_message[0]")
+    
+    current_model = gr.State(None)
+    logged_in_username = gr.State(None)
+    user_chats_state = gr.State([])
+    current_chat_title = gr.State(None)
+    auth_globals_state = gr.State(auth_globals)
+    
+    status_label = gr.Markdown("⚠️ No model loaded. Login + create/select a chat to begin.")
+    
+    with gr.Row():
+        with gr.Column(scale=1, min_width=200):
+            gr.Markdown("### 📋 Your Chats")
+            new_chat_btn = gr.Button("➕ New Chat", variant="primary", size="large")
+            delete_chat_btn = gr.Button("🗑️ Delete Current", variant="stop", size="large")
+            chat_dropdown = gr.Dropdown(choices=[], label="Switch Chat", interactive=True, value=None)
+    
+        with gr.Column(scale=4):
+            with gr.Row():
+                model_dropdown = gr.Dropdown(choices=model_choices, value=model_choices[0], label="🎯 Select Model", interactive=True, scale=4)
+                load_btn = gr.Button("Load & Test Model", variant="primary", scale=1)
+    
+            mode_radio = gr.Radio(
+                choices=["Default Chat", "Fast Search", "Deep Think", "💻 Coding Mode"], 
+                value="Default Chat", 
+                label="🧠 Response Mode", 
+                interactive=True
+                )
+    
+            chatbot = gr.Chatbot(height=480, buttons=["copy", "copy_all"])
+    
+        with gr.Row():
+            msg = gr.Textbox(placeholder="Ask anything...", show_label=False, scale=9)
+            file_upload = gr.File(
+                label="Attach up to 5 files (.txt, .pdf, .docx)",
+                file_types=[".txt", ".pdf", ".docx"],
+                file_count="multiple",
+                type="filepath"
+                )
+        send_btn = gr.Button("➤", variant="primary", scale=1)
+    
+    # Authentication (clean version - no phone/OTP)
+    with gr.Row():
+        username_input = gr.Textbox(label="Username (will be forced lowercase)", scale=2, placeholder="admin")
+        password_input = gr.Textbox(label="Password", type="password", scale=2)
+    
+    with gr.Row():
+        register_btn = gr.Button("Register", variant="secondary")
+        login_btn = gr.Button("Login", variant="primary")
+        logout_btn = gr.Button("Logout", visible=False, variant="stop")
+    
+    with gr.Accordion("⚠️ Danger Zone", open=False):
+        delete_acc_btn = gr.Button("🗑️ Delete My Account", variant="stop", size="large", visible=False)
+        with gr.Row(visible=False) as confirm_row:
+            confirm_username_input = gr.Textbox(label="Type exact username to confirm", placeholder="yourusername")
+            confirm_delete_btn = gr.Button("YES, DELETE FOREVER", variant="stop")
+    
+        # ================== 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])
+    
+        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])
+    
+        chat_dropdown.change(switch_chat, [chat_dropdown, user_chats_state], [chatbot, current_chat_title])
+    
+        register_btn.click(
+            register_user,
+            [username_input, password_input, logged_in_username, auth_globals_state],
+            [user_chats_state, current_chat_title, chatbot, chat_dropdown, logged_in_username, status_label, logout_btn, delete_acc_btn, confirm_row]
+        ).then(lambda: gr.Info("✅ Registration successful!", duration=4), None, None)\
+         .then(lambda: ("",""), None, [username_input, password_input])
+    
+        login_btn.click(
+            login_user,
+            [username_input, password_input, logged_in_username, auth_globals_state],
+            [user_chats_state, current_chat_title, chatbot, chat_dropdown, logged_in_username, status_label, logout_btn, delete_acc_btn, confirm_row]
+        ).then(lambda: gr.Info("✅ Login successful!", duration=4), None, None)\
+         .then(lambda: ("",""), None, [username_input, password_input])
+    
+        logout_btn.click(logout_user, [], [user_chats_state, current_chat_title, chatbot, chat_dropdown, logged_in_username, status_label, logout_btn, delete_acc_btn, confirm_row])
+    
+        delete_acc_btn.click(lambda: gr.update(visible=True), None, confirm_row)
+        confirm_delete_btn.click(delete_account, [confirm_username_input, logged_in_username, auth_globals_state],
+            [status_label, logged_in_username, logout_btn, delete_acc_btn, confirm_row, user_chats_state, current_chat_title, chatbot, chat_dropdown])
+    
+        msg.submit(
+            chat_stream,
+            [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title, mode_radio, auth_globals_state, file_upload],
+            chatbot
+        ).then(lambda: "", None, msg)
+        
+        send_btn.click(
+            chat_stream,
+            [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title, mode_radio, auth_globals_state, file_upload],
+            chatbot
+        ).then(lambda: "", None, msg)
 
 demo.launch(auth=(username,password),theme=gr.themes.Soft(), ssr_mode=False, show_error=True)