Update app.py

app.py

@@ -382,7 +382,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
             # editable title input (kept but small)
             edit_title_box = gr.Textbox(label="Rename selected chat", placeholder="Type new title...", lines=1)
 
-            # rename callback
+            # rename callback (kept unchanged)
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
                 if sid and new_title and new_title.strip():
@@ -395,15 +395,6 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 new_selected = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
                 return gr.update(choices=labels, value=new_selected)
 
-            # wire rename button
-            rename_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=[session_list])
-
-            # wire other small buttons
-            # <-- FIXED: outputs must be component variables, not gr.update(...) dicts
-            new_btn.click(new_chat_cb, outputs=[session_list, chatbot, user_box])
-            del_btn.click(delete_chat_cb, inputs=[session_list], outputs=[session_list, chatbot])
-            refresh_btn.click(refresh_sessions_cb, outputs=[session_list])
-
             gr.Markdown("### 🤖 Model")
             model_choice = gr.Dropdown(
                 choices=list(MODELS.keys()),
@@ -439,12 +430,19 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 send_btn = gr.Button("Send ▶️", variant="primary")
                 regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
 
-    # Hook up callbacks (preserve your existing wires)
+    # Hook up callbacks (moved here so chatbot & user_box exist)
+    # wire rename button (rename callback returns gr.update)
+    rename_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=[session_list])
+
+    # wire other small buttons - use components in outputs (not gr.update)
     refresh_btn.click(refresh_sessions_cb, outputs=[session_list])
     new_btn.click(new_chat_cb, outputs=[session_list, chatbot, user_box])
     del_btn.click(delete_chat_cb, inputs=[session_list], outputs=[session_list, chatbot])
+
+    # session load/change
     session_list.change(load_session_cb, inputs=[session_list], outputs=[chatbot])
 
+    # main conversation wiring
     send_btn.click(
         send_cb,
         inputs=[user_box, session_list, chatbot, system_box, max_tokens, temperature, top_p, model_choice, dataset_choice],