Update app.py

app.py

@@ -153,19 +153,43 @@ def load_dataset_by_name(name: str):
 def refresh_sessions_cb():
     labels, _ = list_sessions()
     selected = labels[0] if labels else None
-    return gr.update(choices=labels, value=selected)
+    visible = True if selected else False
+    return (
+        gr.update(choices=labels, value=selected),
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+    )
 
 def new_chat_cb():
     sid = create_session("New chat")
     labels, _ = list_sessions()
     selected = next((lbl for lbl in labels if lbl.startswith(f"{sid} ")), None)
-    return (gr.update(choices=labels, value=selected), [], "")
+    # Return: session_list, chatbot (empty), user_box cleared, then make edit/save/delete/refresh visible
+    return (
+        gr.update(choices=labels, value=selected),
+        [],  # chatbot messages
+        "",  # user_box cleared
+        gr.update(visible=True),  # edit_title_box visible
+        gr.update(visible=True),  # save_title_btn visible
+        gr.update(visible=True),  # del_btn visible
+        gr.update(visible=True),  # refresh_btn visible
+    )
 
 def load_session_cb(selected_label):
     sid = label_to_id(selected_label)
     if not sid:
-        return []
-    return get_messages(sid)
+        # no valid selection -> hide controls
+        return [], gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    msgs = get_messages(sid)
+    return (
+        msgs,
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=True),
+    )
 
 def delete_chat_cb(selected_label):
     sid = label_to_id(selected_label)
@@ -173,7 +197,15 @@ def delete_chat_cb(selected_label):
         delete_session(sid)
     labels, _ = list_sessions()
     selected = labels[0] if labels else None
-    return gr.update(choices=labels, value=selected), []
+    visible = True if selected else False
+    return (
+        gr.update(choices=labels, value=selected),
+        [],  # chatbot cleared
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+        gr.update(visible=visible),
+    )
 
 FRONTEND_KEYWORDS = [
     "react", "vue", "angular", "html", "css", "javascript", "tailwind", "recharts", "typescript"
@@ -379,14 +411,10 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
             # -----------------------
             # Editable title input
             # -----------------------
-            # -----------------------
-            # Editable title input
-            # -----------------------
-            edit_title_box = gr.Textbox(label="✏️ Rename Chat", placeholder="Edit selected chat title...")
-            save_title_btn = gr.Button("💾", elem_id="save-title-btn")
-
+            # 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)
 
-            
             def rename_session_cb(new_title, selected_label):
                 sid = label_to_id(selected_label)
                 if sid and new_title.strip():
@@ -404,13 +432,11 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
             # Connect button to callback
             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")
-                refresh_btn = gr.Button("🔄", elem_id="refresh-btn")
-
-
+                del_btn = gr.Button("🗑️", elem_id="delete-chat-btn", visible=False)
+                refresh_btn = gr.Button("🔄", elem_id="refresh-btn", visible=False)
 
             gr.Markdown("### 🤖 Model Selection")
             model_choice = gr.Dropdown(
@@ -449,10 +475,25 @@ 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")
 
-    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_list.change(load_session_cb, inputs=session_list, outputs=chatbot)
+    # Wire callbacks — updated outputs so visibility changes flow correctly
+    refresh_btn.click(
+        refresh_sessions_cb,
+        outputs=[session_list, edit_title_box, save_title_btn, del_btn, refresh_btn],
+    )
+    new_btn.click(
+        new_chat_cb,
+        outputs=[session_list, chatbot, user_box, edit_title_box, save_title_btn, del_btn, refresh_btn],
+    )
+    del_btn.click(
+        delete_chat_cb,
+        inputs=session_list,
+        outputs=[session_list, chatbot, edit_title_box, save_title_btn, del_btn, refresh_btn],
+    )
+    session_list.change(
+        load_session_cb,
+        inputs=session_list,
+        outputs=[chatbot, edit_title_box, save_title_btn, del_btn, refresh_btn],
+    )
 
     send_btn.click(
         send_cb,
@@ -473,4 +514,4 @@ with gr.Blocks(title="Backend-Focused LLaMA/Mistral CRUD Assistant", theme=gr.th
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()