From 687b99f9abe04b2f6af33a1663f63ea059a61d2b Mon Sep 17 00:00:00 2001 From: jeffser Date: Mon, 2 Sep 2024 13:36:17 -0600 Subject: [PATCH] Update welcome screen when a model is pulled / created / deleted --- src/custom_widgets/chat_widget.py | 6 ++++++ src/custom_widgets/model_widget.py | 7 ++++--- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/src/custom_widgets/chat_widget.py b/src/custom_widgets/chat_widget.py index 39324ed..892a408 100644 --- a/src/custom_widgets/chat_widget.py +++ b/src/custom_widgets/chat_widget.py @@ -262,6 +262,12 @@ class chat_list(Gtk.ListBox): self.connect("row-selected", lambda listbox, row: self.chat_changed(row)) self.tab_list = [] + def update_welcome_screens(self, show_prompts:bool): + print('SHOW_PROMPTS: ', show_prompts) + for tab in self.tab_list: + if tab.chat_window.welcome_screen: + tab.chat_window.show_welcome_screen(show_prompts) + def get_tab_by_name(self, chat_name:str) -> chat_tab: for tab in self.tab_list: if tab.chat_window.get_name() == chat_name: diff --git a/src/custom_widgets/model_widget.py b/src/custom_widgets/model_widget.py index 04b7ebf..82071fc 100644 --- a/src/custom_widgets/model_widget.py +++ b/src/custom_widgets/model_widget.py @@ -436,9 +436,10 @@ class model_manager_container(Gtk.Box): if response.status_code == 200: self.local_list.remove_model(model_name) - if len(list(self.local_list)) == 0: - self.local_list.set_visible(False) self.model_selector.remove_model(model_name) + if len(self.get_model_list()) == 0: + self.local_list.set_visible(False) + window.chat_list_box.update_welcome_screens(False) window.show_toast(_("Model deleted successfully"), window.manage_models_overlay) else: window.manage_models_dialog.close() @@ -526,7 +527,7 @@ class model_manager_container(Gtk.Box): GLib.idle_add(window.connection_error) self.pulling_list.remove(model) - + GLib.idle_add(window.chat_list_box.update_welcome_screens, len(self.get_model_list()) > 0) if len(list(self.pulling_list)) == 0: GLib.idle_add(self.pulling_list.set_visible, False)