Update app.py

app.py

@@ -61,7 +61,7 @@ def load_model(selected_model):
         return None, f"**❌ Unavailable: {error_msg}**"
 
 # ================== MULTI-CHAT HELPERS ==================
-def create_new_chat(user_chats, logged_in_username):
+def create_new_chat(user_chats, logged_in_username, coding_files):
     if not logged_in_username:
         return user_chats, None, [], gr.update(choices=[], value=None), "⚠️ Please login first!", ""
     new_num = len(user_chats) + 1
@@ -72,7 +72,7 @@ def create_new_chat(user_chats, logged_in_username):
     save_chats(logged_in_username, new_user_chats, auth_globals)  # Persist changes
     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):
+def switch_chat(selected_title, user_chats, coding_files):
     if not selected_title or not user_chats:
         return [], None, ""
     for chat in user_chats:
@@ -80,7 +80,7 @@ def switch_chat(selected_title, user_chats):
             return chat["history"], selected_title, chat.get("context", "")
     return [], None, ""
 
-def delete_current_chat(current_title, user_chats, logged_in_username):
+def delete_current_chat(current_title, user_chats, logged_in_username, coding_files):
     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]
@@ -113,6 +113,64 @@ def save_chat_context(new_context, user_chats, current_chat_title, logged_in_use
 def format_history_for_chatbot(history):
     """Return history as-is (already dict format)"""
     return history or []
+
+
+def refresh_file_list(user_chats, current_title):
+    if not current_title or not user_chats:
+        return gr.update(choices=[], value=None)
+    for chat in user_chats:
+        if chat["title"] == current_title:
+            files = list(chat.get("coding_files", {}).keys())
+            return gr.update(choices=files, value=files[0] if files else None)
+    return gr.update(choices=[], value=None)
+
+def load_file_to_editor(selected_file, user_chats, current_title):
+    if not selected_file or not current_title:
+        return "", "python"
+    for chat in user_chats:
+        if chat["title"] == current_title:
+            f = chat.get("coding_files", {}).get(selected_file, {})
+            return f.get("content", ""), f.get("language", "python")
+    return "", "python"
+
+def save_file_content(selected_file, new_content, new_language, user_chats, current_title, logged_in_username, auth_globals):
+    if not selected_file or not current_title or not logged_in_username:
+        return user_chats, "❌ No file selected"
+    for chat in user_chats:
+        if chat["title"] == current_title:
+            if "coding_files" not in chat:
+                chat["coding_files"] = {}
+            chat["coding_files"][selected_file] = {"content": new_content, "language": new_language}
+            break
+    save_chats(logged_in_username, user_chats, auth_globals)
+    return user_chats, f"✅ Saved {selected_file}"
+
+def add_uploaded_files(uploaded_files, user_chats, current_title, logged_in_username, auth_globals):
+    if not uploaded_files or not current_title:
+        return user_chats, gr.update(), "❌ No chat selected"
+    for chat in user_chats:
+        if chat["title"] == current_title:
+            if "coding_files" not in chat:
+                chat["coding_files"] = {}
+            for path in uploaded_files if isinstance(uploaded_files, list) else [uploaded_files]:
+                name = os.path.basename(path)
+                with open(path, "r", encoding="utf-8") as f:
+                    content = f.read()
+                lang = "python" if name.endswith(".py") else "javascript" if name.endswith(".js") else "html" if name.endswith(".html") else "text"
+                chat["coding_files"][name] = {"content": content, "language": lang}
+            break
+    save_chats(logged_in_username, user_chats, auth_globals)
+    return user_chats, refresh_file_list(user_chats, current_title), "✅ Files added to persistent storage"
+
+def delete_selected_file(selected_file, user_chats, current_title, logged_in_username, auth_globals):
+    if not selected_file or not current_title:
+        return user_chats, gr.update(), "❌ Nothing to delete"
+    for chat in user_chats:
+        if chat["title"] == current_title and "coding_files" in chat:
+            chat["coding_files"].pop(selected_file, None)
+            break
+    save_chats(logged_in_username, user_chats, auth_globals)
+    return user_chats, refresh_file_list(user_chats, current_title), f"🗑️ Deleted {selected_file}"
     
 # ================== CHAT HANDLER WRAPPER (with file limit) ==================
 def handle_chat(msg, chatbot, current_model, logged_in_username, user_chats_state, current_chat_title, mode_radio, auth_globals_state, file_upload, current_chat_context):
@@ -140,6 +198,8 @@ def handle_chat(msg, chatbot, current_model, logged_in_username, user_chats_stat
     except Exception as e:
         gr.Warning(f"⚠️ An error occurred in the chat processing: {str(e)}")
         return chatbot  # Gracefully return the existing chatbot state
+
+
 # ================== GUI ==================
 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**")
@@ -154,8 +214,12 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
     current_chat_title = gr.State(None)
     current_chat_context = gr.State("")  # Added: State for current chat's context
     auth_globals_state = gr.State(auth_globals)
+    coding_files_state = gr.State({})      # { "filename.py": {"content": "...", "language": "python"} }
+    current_file_name = gr.State(None)
     status_label = gr.Markdown("⚠️ No model loaded. Login + create/select a chat to begin.", elem_id="status-bold")  # Made more prominent
 
+    
+    
     with gr.Row():
         with gr.Column(scale=1, min_width=200):
             gr.Markdown("### 📋 Your Chats")
@@ -182,6 +246,39 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
                              interactive=True, visible=False)
         save_context_btn = gr.Button("💾 Save Context", variant="secondary", visible=False)
 
+    # ================== CODING MODE FILE MANAGER (visible only in Coding Mode) ==================
+    with gr.Row(visible=False) as coding_manager_row:
+        with gr.Column(scale=1, min_width=280):
+            gr.Markdown("### 📁 Persistent Project Files\n(saved per chat • only in Coding Mode)")
+            file_upload_coding = gr.File(
+                label="Upload files",
+                file_types=[".py",".js",".html",".css",".txt",".md",".json",".ts"],
+                file_count="multiple",
+                type="filepath"
+            )
+            add_files_btn = gr.Button("➕ Add / Replace files", variant="primary")
+            
+            file_list_dropdown = gr.Dropdown(choices=[], label="Select file to edit", interactive=True)
+            delete_file_btn = gr.Button("🗑️ Delete selected file", variant="stop", size="sm")
+
+        with gr.Column(scale=4):
+            file_editor = gr.Code(
+                label="Editor",
+                value="",
+                language="python",
+                lines=24,
+                interactive=True,
+                show_label=True
+            )
+            with gr.Row():
+                language_selector = gr.Dropdown(
+                    choices=["python", "javascript", "typescript", "html", "css", "markdown", "json", "text"],
+                    value="python",
+                    label="Language",
+                    scale=1
+                )
+                save_file_btn = gr.Button("💾 Save file", variant="primary", scale=2)
+
     with gr.Row():
         msg = gr.Textbox(placeholder="Ask anything...", show_label=False, scale=9)
         file_upload = gr.File(
@@ -211,7 +308,7 @@ 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])
+    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])
     
@@ -249,4 +346,46 @@ with gr.Blocks(title="NVIDIA AI Chat") as demo:
         chatbot
     ).then(lambda: "", None, msg)
 
+# File manager events
+    add_files_btn.click(
+        add_uploaded_files,
+        [file_upload_coding, user_chats_state, current_chat_title, logged_in_username, auth_globals_state],
+        [user_chats_state, file_list_dropdown, status_label]
+    )
+
+    file_list_dropdown.change(
+        load_file_to_editor,
+        [file_list_dropdown, user_chats_state, current_chat_title],
+        [file_editor, language_selector]
+    )
+
+    save_file_btn.click(
+        save_file_content,
+        [file_list_dropdown, file_editor, language_selector, user_chats_state, current_chat_title, logged_in_username, auth_globals_state],
+        [user_chats_state, status_label]
+    ).then(refresh_file_list, [user_chats_state, current_chat_title], file_list_dropdown)
+
+    delete_file_btn.click(
+        delete_selected_file,
+        [file_list_dropdown, user_chats_state, current_chat_title, logged_in_username, auth_globals_state],
+        [user_chats_state, file_list_dropdown, status_label]
+    )
+
+    language_selector.change(
+        lambda lang: gr.update(language=lang),
+        language_selector,
+        file_editor
+    )
+
+    # Show/hide file manager when mode changes
+    mode_radio.change(
+        lambda m: gr.update(visible=(m == "💻 Coding Mode")),
+        mode_radio,
+        coding_manager_row
+    ).then(
+        refresh_file_list,
+        [user_chats_state, current_chat_title],
+        file_list_dropdown
+    )
+
 demo.launch(auth=(username, password), theme=gr.themes.Soft(), server_name="0.0.0.0", debug=True, show_error=True)