Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 from openai import OpenAI
 import gradio as gr
-from auth import register_user, login_user, logout_user, save_chats
+from auth import register_user, login_user, logout_user, save_chats, delete_account
 
 # ================== YOUR SETTINGS ==================
 API_KEY = os.environ.get("API_KEY")
@@ -11,71 +11,35 @@ USER_DATA_REPO = "qwendashcoder/fluade_1"
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN secret not set in HF Spaces settings!")
 
-client = OpenAI(
-    base_url="https://integrate.api.nvidia.com/v1",
-    api_key=API_KEY
-)
-
-auth_globals = {
-    "hf_token": HF_TOKEN,
-    "user_repo": USER_DATA_REPO
-}
-
-model_choices = [
-    "meta/llama-3.1-405b-instruct",
-    "meta/llama-3.1-70b-instruct",
-    "meta/llama-3.1-8b-instruct",
-    "mistralai/mistral-large-2-instruct",
-    "mistralai/mixtral-8x22b-instruct",
-    "mistralai/mistral-nemo-instruct-2407",
-    "qwen/qwen-2.5-72b-instruct",
-    "google/gemma-2-27b-it",
-    "google/gemma-2-9b-it",
-    "google/codegemma-7b-it",
-]
-
-deprecated_models = [
-    "qwen/qwen3-coder-480b-a35b-instruct",
-    "qwen/qwen2.5-coder-32b-instruct",
-]
+client = OpenAI(base_url="https://integrate.api.nvidia.com/v1", api_key=API_KEY)
 
+auth_globals = {"hf_token": HF_TOKEN, "user_repo": USER_DATA_REPO}
+
+model_choices = ["meta/llama-3.1-405b-instruct","meta/llama-3.1-70b-instruct","meta/llama-3.1-8b-instruct","mistralai/mistral-large-2-instruct","mistralai/mixtral-8x22b-instruct","mistralai/mistral-nemo-instruct-2407","qwen/qwen-2.5-72b-instruct","google/gemma-2-27b-it","google/gemma-2-9b-it","google/codegemma-7b-it"]
+deprecated_models = ["qwen/qwen3-coder-480b-a35b-instruct","qwen/qwen2.5-coder-32b-instruct"]
 model_choices = [m for m in model_choices if m not in deprecated_models]
 default_model = model_choices[0] if model_choices else None
 
 def is_model_available(model_id):
     try:
-        test_completion = client.chat.completions.create(
-            model=model_id,
-            messages=[{"role": "user", "content": "Say 'test'"}],
-            max_tokens=5,
-            stream=False
-        )
+        test_completion = client.chat.completions.create(model=model_id, messages=[{"role": "user", "content": "Say 'test'"}], max_tokens=5, stream=False)
         return test_completion.choices[0].message.content is not None
     except Exception:
         return False
 
-available_models = [model for model in model_choices if is_model_available(model)]
-if not available_models:
-    available_models = [default_model] if default_model else []
+available_models = [model for model in model_choices if is_model_available(model)] or [default_model]
 default_model = available_models[0] if available_models else None
 
 def load_model(selected_model):
     try:
-        test_completion = client.chat.completions.create(
-            model=selected_model,
-            messages=[{"role": "user", "content": "Say 'test'"}],
-            max_tokens=5,
-            stream=False
-        )
+        test_completion = client.chat.completions.create(model=selected_model, messages=[{"role": "user", "content": "Say 'test'"}], max_tokens=5, stream=False)
         if test_completion.choices[0].message.content:
             return selected_model, f"✅ Loaded & Ready: {selected_model}"
     except Exception as e:
         return None, f"❌ Unavailable: {str(e)}\nTry another model."
 
-# ================== MULTI-CHAT HELPERS ==================
 def create_new_chat(user_chats, logged_in_username):
-    if not logged_in_username:
-        return user_chats, None, [], gr.update(choices=[], value=None), "⚠️ Please login first!"
+    if not logged_in_username: return user_chats, None, [], gr.update(choices=[], value=None), "⚠️ Please login first!"
     new_num = len(user_chats) + 1
     new_title = f"Chat {new_num}"
     new_chat = {"title": new_title, "history": []}
@@ -84,16 +48,14 @@ def create_new_chat(user_chats, logged_in_username):
     return new_user_chats, new_title, [], gr.update(choices=choices, value=new_title), f"✅ New chat created: {new_title}"
 
 def switch_chat(selected_title, user_chats):
-    if not selected_title or not user_chats:
-        return [], None
+    if not selected_title or not user_chats: return [], None
     for chat in user_chats:
         if chat["title"] == selected_title:
             return chat["history"], selected_title
     return [], None
 
 def delete_current_chat(current_title, user_chats, logged_in_username):
-    if not logged_in_username or not current_title:
-        return user_chats, current_title, [], gr.update(), "⚠️ Nothing to delete!"
+    if not logged_in_username or not current_title: return user_chats, current_title, [], gr.update(), "⚠️ Nothing to delete!"
     new_user_chats = [chat for chat in user_chats if chat["title"] != current_title]
     choices = [chat["title"] for chat in new_user_chats]
     if new_user_chats:
@@ -104,83 +66,56 @@ def delete_current_chat(current_title, user_chats, logged_in_username):
         new_history = []
     return new_user_chats, new_title, new_history, gr.update(choices=choices, value=new_title), f"🗑️ Deleted {current_title}"
 
-# ================== UPDATE CHATS AFTER STREAMING (fixes persistence) ==================
 def update_chats_after_response(user_chats, current_chat_title, final_history, logged_in_username):
-    if not logged_in_username or not current_chat_title or not user_chats:
-        return user_chats
+    if not logged_in_username or not current_chat_title or not user_chats: return user_chats
     for chat in user_chats:
         if chat["title"] == current_chat_title:
-            chat["history"] = final_history[:]   # save the full updated history
+            chat["history"] = final_history[:]
             break
     save_chats(logged_in_username, user_chats, auth_globals)
     return user_chats
 
-# ================== CHAT STREAM (streaming only) ==================
 def chat_stream(message, history, current_model, logged_in_username, user_chats, current_chat_title):
     history = history or []
-
-    if not logged_in_username:
+    if not logged_in_username: 
         history.append({"role": "assistant", "content": "⚠️ Please login first!"})
-        yield history
-        return
-    if not current_model:
+        yield history; return
+    if not current_model: 
         history.append({"role": "assistant", "content": "⚠️ Load a model first!"})
-        yield history
-        return
-    if not current_chat_title:
+        yield history; return
+    if not current_chat_title: 
         history.append({"role": "assistant", "content": "⚠️ Select or create a chat first!"})
-        yield history
-        return
+        yield history; return
 
-    # Normal chat
     history.append({"role": "user", "content": message})
     yield history
-
     history.append({"role": "assistant", "content": ""})
     yield history
 
-    messages = []
-    for msg in history[:-1]:
-        content = msg["content"] if isinstance(msg["content"], str) else str(msg["content"])
-        messages.append({"role": msg["role"], "content": content})
+    messages = [{"role": msg["role"], "content": str(msg["content"])} for msg in history[:-1]]
 
     try:
-        stream = client.chat.completions.create(
-            model=current_model,
-            messages=messages,
-            temperature=0.7,
-            top_p=0.8,
-            max_tokens=4096,
-            stream=True
-        )
+        stream = client.chat.completions.create(model=current_model, messages=messages, temperature=0.7, top_p=0.8, max_tokens=4096, stream=True)
         for chunk in stream:
             if chunk.choices and chunk.choices[0].delta.content is not None:
                 history[-1]["content"] += chunk.choices[0].delta.content
                 yield history
     except Exception as e:
-        history[-1]["content"] = f"❌ Error: {str(e)}\nTry switching models."
+        history[-1]["content"] = f"❌ Error: {str(e)}"
         yield history
 
-# ================== GUI (dropdown now auto-loads) ==================
+# ================== GUI ==================
 with gr.Blocks(title="NVIDIA AI Chat") as demo:
     gr.Markdown("# 🟢 NVIDIA AI Chat\n**Multiple chats • Mobile optimized • Auto-save**")
     gr.Markdown("Powered By AI🤖 • Edited/Compiled By tr0ecolor")
 
     with gr.Accordion("📖 **How to Use the Interface (Step-by-Step)**", open=True):
         gr.Markdown("""
-**Step 1: Login / Register**  
-Enter username + password → click **Register** or **Login**. Fields auto-clear.
-
-**Step 2: Load Model**  
-Pick model → **Load & Test Model**.
-
-**Step 3: Manage Chats**  
-• **➕ New Chat**  
-• **Switch Chat** dropdown → click any chat = loads instantly  
-• **🗑️ Delete Current**
-
-**Step 4: Chat!**  
-Type → Enter or **➤**. Message box clears automatically. All chats saved forever.
+**Step 1:** Login / Register (fields auto-clear)  
+**Step 2:** Load a model  
+**Step 3:** Manage chats (New / Switch / Delete current)  
+**Step 4:** Chat! Everything auto-saves  
+**Danger Zone (bottom):** Delete your entire account permanently
         """)
 
     current_model = gr.State(None)
@@ -195,23 +130,11 @@ Type → Enter or **➤**. Message box clears automatically. All chats saved for
             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
-            )
+            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=available_models,
-                    value=default_model,
-                    label="🎯 Select Model",
-                    interactive=True,
-                    scale=4
-                )
+                model_dropdown = gr.Dropdown(choices=available_models, value=default_model, label="🎯 Select Model", interactive=True, scale=4)
                 load_btn = gr.Button("Load & Test Model", variant="primary", scale=1)
 
             chatbot = gr.Chatbot(height=480, buttons=["copy", "copy_all"])
@@ -227,67 +150,44 @@ Type → Enter or **➤**. Message box clears automatically. All chats saved for
         login_btn = gr.Button("Login", variant="primary")
         logout_btn = gr.Button("Logout", visible=False, variant="stop")
 
+    # Danger Zone
+    with gr.Accordion("⚠️ Danger Zone - Permanent Actions", 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 your exact username to confirm", placeholder="your_username")
+            confirm_delete_btn = gr.Button("YES, DELETE ACCOUNT FOREVER", variant="stop")
+
     auth_globals_state = gr.State(auth_globals)
 
-    # ================== EVENT HANDLERS ==================
-    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]
-    ).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]
-    ).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]
-    )
+    # Event Handlers
+    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: ("",""), 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: ("",""), 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])
 
     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]
-    )
-
-    # Auto-load when you click any chat in dropdown
-    chat_dropdown.change(
-        switch_chat,
-        [chat_dropdown, user_chats_state],
-        [chatbot, current_chat_title]
-    )
-
-    # Send message → stream + auto-save after finish
-    msg.submit(
-        chat_stream,
-        [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title],
-        chatbot
-    ).then(
-        update_chats_after_response,
-        [user_chats_state, current_chat_title, chatbot, logged_in_username],
-        user_chats_state
-    ).then(lambda: "", None, msg)
-
-    send_btn.click(
-        chat_stream,
-        [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title],
-        chatbot
-    ).then(
-        update_chats_after_response,
-        [user_chats_state, current_chat_title, chatbot, logged_in_username],
-        user_chats_state
-    ).then(lambda: "", None, msg)
+    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])
+
+    msg.submit(chat_stream, [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title], chatbot)\
+        .then(update_chats_after_response, [user_chats_state, current_chat_title, chatbot, logged_in_username], user_chats_state)\
+        .then(lambda: "", None, msg)
+
+    send_btn.click(chat_stream, [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title], chatbot)\
+        .then(update_chats_after_response, [user_chats_state, current_chat_title, chatbot, logged_in_username], user_chats_state)\
+        .then(lambda: "", None, msg)
 
 demo.launch(theme=gr.themes.Soft(), ssr_mode=False, show_error=True)