Update app.py

app.py

@@ -401,20 +401,31 @@ 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("### 📁 Sessions")
+            gr.Markdown("### 📁 Chats")
+    
+            # --- Chat list ---
             session_list = gr.Radio(
                 choices=labels,
                 value=default_selected,
-                label="Your chats",
+                label="Your Chats",
                 interactive=True
             )
-            # -----------------------
-            # Editable title input
-            # -----------------------
-            # Hidden initially — will be shown when a chat is created/selected
-            edit_title_box = gr.Textbox(label="✏️ Rename Chat", placeholder="Edit selected chat title...", visible=False)
-            save_title_btn = gr.Button("💾", elem_id="save-title-btn", visible=False)
-
+    
+            # --- 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)
+    
+            # --- Rename box & save button below the action buttons ---
+            edit_title_box = gr.Textbox(
+                label="✏️ Rename Chat",
+                placeholder="Edit selected chat title...",
+                visible=False
+            )
+            save_title_btn = gr.Button("💾 Save Title", elem_id="save-title-btn", visible=False)
+    
+            # Callback for renaming session
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
                 if sid and new_title.strip():
@@ -423,21 +434,16 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                     cur.execute("UPDATE sessions SET title=? WHERE id=?", (new_title.strip(), sid))
                     conn.commit()
                     conn.close()
-            
+    
                 # 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)
-            
-            # Connect button to callback
+    
+            # Connect rename button
             save_title_btn.click(rename_session_cb, inputs=[edit_title_box, session_list], outputs=session_list)
-
-            # Action buttons: New Chat always visible, others hidden initially
-            with gr.Row(elem_id="session-action-row"):
-                new_btn = gr.Button("➕", elem_id="new-chat-btn")
-                del_btn = gr.Button("🗑️", elem_id="delete-chat-btn", visible=False)
-                refresh_btn = gr.Button("🔄", elem_id="refresh-btn", visible=False)
-
+    
+            # --- Model Selection ---
             gr.Markdown("### 🤖 Model Selection")
             model_choice = gr.Dropdown(
                 choices=list(MODELS.keys()),
@@ -445,7 +451,8 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 label="Choose a model",
                 interactive=True
             )
-
+    
+            # --- Dataset Selection ---
             gr.Markdown("### 📚 Dataset Selection")
             dataset_choice = gr.Dropdown(
                 choices=DATASETS,
@@ -453,7 +460,8 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 label="Select a dataset",
                 interactive=True
             )
-
+    
+            # --- Generation Settings ---
             gr.Markdown("### ⚙️ Generation Settings")
             with gr.Group(elem_classes="system-message-compact"):
                 system_box = gr.Textbox(
@@ -461,12 +469,13 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                     label="System message",
                     lines=3
                 )
-            
+    
             with gr.Group(elem_classes="compact-sliders"):
                 max_tokens = gr.Slider(256, 4096, value=1200, step=16, label="Max tokens")
                 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():
@@ -475,6 +484,7 @@ 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")
 
+
     # Wire callbacks — updated outputs so visibility changes flow correctly
     refresh_btn.click(
         refresh_sessions_cb,