Update app.py

app.py

@@ -401,9 +401,9 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
 
     with gr.Row(equal_height=True):
         with gr.Column(scale=1, min_width=260):
-            gr.Markdown("### 📁 Chats")
+            gr.Markdown("### 📁 Sessions")
     
-            # --- Chat list ---
+            # Main session list
             session_list = gr.Radio(
                 choices=labels,
                 value=default_selected,
@@ -411,21 +411,22 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 interactive=True
             )
     
-            # --- Action buttons appear just below the chat list ---
-            with gr.Row():
-                new_btn = gr.Button("➕ New", elem_id="new-chat-btn")
-                del_btn = gr.Button("🗑️ Delete", elem_id="delete-chat-btn", visible=False)
-                refresh_btn = gr.Button("🔄 Refresh", elem_id="refresh-btn", visible=False)
+            # Row for actions just below the list (not at the very bottom)
+            with gr.Row(elem_id="session-actions-row"):
+                new_btn = gr.Button("New Chat", elem_id="new-chat-btn")
+                rename_btn = gr.Button("Rename", elem_id="rename-chat-btn", visible=False)
+                save_btn = gr.Button("Save", elem_id="save-title-btn", visible=False)
+                del_btn = gr.Button("Delete", elem_id="delete-chat-btn", visible=False)
+                refresh_btn = gr.Button("Refresh", elem_id="refresh-btn", visible=False)
     
-            # --- Rename box & save button below the action buttons ---
+            # Rename textbox (hidden until Rename is clicked)
             edit_title_box = gr.Textbox(
-                label="✏️ Rename Chat",
-                placeholder="Edit selected chat title...",
+                label="Edit Chat Name",
+                placeholder="Type new chat name here…",
                 visible=False
             )
-            save_title_btn = gr.Button("💾 Save Title", elem_id="save-title-btn", visible=False)
     
-            # Callback for renaming session
+            # Callbacks
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
                 if sid and new_title.strip():
@@ -438,12 +439,15 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 # Refresh the session list and keep the same one selected
                 labels, _ = list_sessions()
                 new_selected = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
-                return gr.update(choices=labels, value=new_selected)
+                return gr.update(choices=labels, value=new_selected), gr.update(value="")
     
-            # Connect rename button
-            save_title_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=session_list)
+            save_btn.click(
+                rename_session_cb,
+                inputs=[edit_title_box, session_list],
+                outputs=[session_list, edit_title_box]
+            )
     
-            # --- Model Selection ---
+            # -------- Model selection --------
             gr.Markdown("### 🤖 Model Selection")
             model_choice = gr.Dropdown(
                 choices=list(MODELS.keys()),
@@ -452,7 +456,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 interactive=True
             )
     
-            # --- Dataset Selection ---
+            # -------- Dataset selection --------
             gr.Markdown("### 📚 Dataset Selection")
             dataset_choice = gr.Dropdown(
                 choices=DATASETS,
@@ -461,7 +465,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 interactive=True
             )
     
-            # --- Generation Settings ---
+            # -------- Generation settings --------
             gr.Markdown("### ⚙️ Generation Settings")
             with gr.Group(elem_classes="system-message-compact"):
                 system_box = gr.Textbox(
@@ -475,7 +479,6 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 temperature = gr.Slider(0.0, 2.0, value=0.25, step=0.05, label="Temperature")
                 top_p = gr.Slider(0.1, 1.0, value=0.9, step=0.05, label="Top-p")
     
-        # --- Chatbot & Input Column ---
         with gr.Column(scale=3):
             chatbot = gr.Chatbot(label="Assistant", height=500, type="messages")
             with gr.Row():
@@ -485,6 +488,7 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 regen_btn = gr.Button("Regenerate 🔁", variant="secondary")
 
 
+
     # Wire callbacks — updated outputs so visibility changes flow correctly
     refresh_btn.click(
         refresh_sessions_cb,