Update app.py

app.py

@@ -334,7 +334,7 @@ def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p
 
     except Exception as e:
         display_msgs[-1]["content"] = f"⚠️ Error: {str(e)}"
-        yield display_msgs
+        yield (display_msgs, "", selected_label)
 
 # ---------------------------
 # App UI
@@ -348,53 +348,21 @@ if not labels:
 default_selected = labels[0] if labels else None
 
 with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.themes.Soft()) as demo:
+    # Minimal CSS for compact, professional look
     gr.HTML("""
     <style>
-    /* General button style */
-    #session-action-row button,
-    #save-title-btn {
-        width: 36px !important;
-        height: 36px !important;
-        border-radius: 8px !important;
-        font-size: 18px !important;
-        padding: 0 !important;
-        margin: 2px !important;
-        background-color: #22c55e !important;
-        color: #fff !important;
-        border: none !important;
-        cursor: pointer;
-    }
-    
-    /* Hover & focus effects */
-    #session-action-row button:hover,
-    #save-title-btn:hover {
-        background-color: #16a34a !important;
-    }
-    
-    #session-action-row button:focus,
-    #save-title-btn:focus {
-        outline: 2px solid #166534 !important;
-    }
-    
-    /* Compact form/sliders keep */
-    .compact-sliders .gr-form {
-        margin-bottom: 0.5rem !important;
-    }
-    .compact-sliders .gr-slider {
-        margin-bottom: 0.5rem !important;
-    }
-    .compact-sliders .gr-number {
-        margin-bottom: 0.5rem !important;
-    }
-    .system-message-compact .gr-form {
-        margin-bottom: 0.5rem !important;
+    /* compact action row below chat list */
+    #session-actions-row .gr-button {
+        margin-right: 6px;
     }
+    /* keep settings compact */
+    .compact-sliders .gr-slider,
+    .compact-sliders .gr-number { margin-bottom: 0.5rem !important; }
     .system-message-compact .gr-textarea {
         min-height: 80px !important;
         max-height: 120px !important;
     }
     </style>
-
     """)
 
     gr.Markdown("## 🗄️ LLaMA & Mistral Backend-Focused CRUD Automation — with Persistent History")
@@ -402,7 +370,7 @@ 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")
-    
+
             # Main session list
             session_list = gr.Radio(
                 choices=labels,
@@ -410,23 +378,29 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
                 label="Your Chats",
                 interactive=True
             )
-    
-            # Row for actions just below the list (not at the very bottom)
+
+            # Row for actions (visible/hidden controlled via callbacks)
             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)
+                save_title_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 textbox (hidden until Rename is clicked)
             edit_title_box = gr.Textbox(
                 label="Edit Chat Name",
                 placeholder="Type new chat name here…",
                 visible=False
             )
-    
-            # Callbacks
+
+            # Callback to open the rename box (show edit box + save button)
+            def show_rename_ui():
+                return gr.update(visible=True), gr.update(visible=True)
+
+            rename_btn.click(fn=show_rename_ui, outputs=[edit_title_box, save_title_btn])
+
+            # Callback for renaming -> updates DB and refreshes list, clears edit box
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
                 if sid and new_title.strip():
@@ -435,18 +409,17 @@ 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 updated session list selection and clear the edit box value
                 return gr.update(choices=labels, value=new_selected), gr.update(value="")
-    
-            save_btn.click(
+
+            save_title_btn.click(
                 rename_session_cb,
                 inputs=[edit_title_box, session_list],
                 outputs=[session_list, edit_title_box]
             )
-    
+
             # -------- Model selection --------
             gr.Markdown("### 🤖 Model Selection")
             model_choice = gr.Dropdown(
@@ -455,7 +428,7 @@ 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(
@@ -464,7 +437,7 @@ 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"):
@@ -473,12 +446,12 @@ 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")
-    
+
         with gr.Column(scale=3):
             chatbot = gr.Chatbot(label="Assistant", height=500, type="messages")
             with gr.Row():
@@ -487,8 +460,6 @@ 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,