Update app.py

app.py

@@ -4,18 +4,16 @@ import gradio as gr
 from auth import register_user, login_user, logout_user, save_chats, delete_account
 from response_modes import chat_stream
 
-# ================== YOUR SETTINGS ==================
 API_KEY = os.environ.get("API_KEY")
 HF_TOKEN = os.environ.get("HF_TOKEN")
-username= os.environ.get("username")
+username = os.environ.get("username")
 password = os.environ.get("password")
 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)  # kept for model loading only
-
+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"]
@@ -70,17 +68,17 @@ 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}"
 
-# ================== GUI ==================
 with gr.Blocks(title="NVIDIA AI Chat") as demo:
-    gr.Markdown("# 🟢 NVIDIA AI Chat\n**Multiple chats • Fast Search • Deep Think**")
+    gr.Markdown("# 🟢 NVIDIA AI Chat\n**Multiple chats • Fast Search • Deep Think • File Upload**")
     gr.Markdown("Powered By AI🤖 • Edited/Compiled By tr0ecolor")
 
     with gr.Accordion("📖 **How to Use**", open=True):
         gr.Markdown("""
 **Step 1:** Login / Register  
 **Step 2:** Load Model  
-**Step 3:** Choose **Response Mode**  
-**Step 4:** Chat!
+**Step 3:** Choose Response Mode  
+**Step 4:** (Optional) Attach .txt / .pdf / .docx  
+**Step 5:** Type message → Send
         """)
 
     current_model = gr.State(None)
@@ -108,7 +106,8 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
             chatbot = gr.Chatbot(height=480, buttons=["copy", "copy_all"])
 
             with gr.Row():
-                msg = gr.Textbox(placeholder="Ask anything... (Enter or ➤)", show_label=False, scale=9)
+                msg = gr.Textbox(placeholder="Ask anything... (Enter or ➤)", show_label=False, scale=8)
+                file_upload = gr.File(label="Attach (.txt/.pdf/.docx)", file_types=[".txt", ".pdf", ".docx"], scale=2)
                 send_btn = gr.Button("➤", variant="primary", scale=1)
 
     with gr.Row():
@@ -124,7 +123,6 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
             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")
 
-    # ================== EVENTS ==================
     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)\
@@ -151,9 +149,12 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
 
     chat_dropdown.change(switch_chat, [chat_dropdown, user_chats_state], [chatbot, current_chat_title])
 
-    # FIXED: no client_state, no raw client passed
-    msg.submit(chat_stream, [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title, mode_radio, auth_globals_state], chatbot).then(lambda: "", None, msg)
+    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)\
+        .then(lambda: None, None, file_upload)
 
-    send_btn.click(chat_stream, [msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title, mode_radio, auth_globals_state], 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)\
+        .then(lambda: None, None, file_upload)
 
-demo.launch(auth=(username, password),theme=gr.themes.Soft(), ssr_mode=False, show_error=True)
+demo.launch(theme=gr.themes.Soft(), ssr_mode=False, show_error=True)