hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Better progressbar for models

Browse Source
This commit is contained in:
jeffser 2024-05-27 20:47:59 -06:00
parent 05461fb03a
commit 1e287dde6f
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/window.py
View File

@ -546,7 +546,6 @@ class AlpacaWindow(Adw.ApplicationWindow):
else: else:
if len(list(self.pulling_models.keys())) == 0: if len(list(self.pulling_models.keys())) == 0:
GLib.idle_add(self.pulling_model_list_box.set_visible, False) GLib.idle_add(self.pulling_model_list_box.set_visible, False)
sys.exit()
def pull_model_process(self, model): def pull_model_process(self, model):
data = {"name":model} data = {"name":model}
@ -556,11 +555,11 @@ class AlpacaWindow(Adw.ApplicationWindow):
if response['status'] == 'ok': if response['status'] == 'ok':
GLib.idle_add(self.show_notification, _("Task Complete"), _("Model '{}' pulled successfully.").format(model), True, Gio.ThemedIcon.new("emblem-ok-symbolic")) GLib.idle_add(self.show_notification, _("Task Complete"), _("Model '{}' pulled successfully.").format(model), True, Gio.ThemedIcon.new("emblem-ok-symbolic"))
GLib.idle_add(self.show_toast, "good", 1, self.manage_models_overlay) GLib.idle_add(self.show_toast, "good", 1, self.manage_models_overlay)
GLib.idle_add(self.pulling_models[model].get_parent().remove, self.pulling_models[model]) GLib.idle_add(self.pulling_models[model]['overlay'].get_parent().get_parent().remove, self.pulling_models[model]['overlay'].get_parent())
del self.pulling_models[model] del self.pulling_models[model]
else: else:
GLib.idle_add(self.show_notification, _("Pull Model Error"), _("Failed to pull model '{}' due to network error.").format(model), True, Gio.ThemedIcon.new("dialog-error-symbolic")) GLib.idle_add(self.show_notification, _("Pull Model Error"), _("Failed to pull model '{}' due to network error.").format(model), True, Gio.ThemedIcon.new("dialog-error-symbolic"))
GLib.idle_add(self.pulling_models[model].get_parent().remove, self.pulling_models[model]) GLib.idle_add(self.pulling_models[model]['overlay'].get_parent().get_parent().remove, self.pulling_models[model]['overlay'].get_parent())
del self.pulling_models[model] del self.pulling_models[model]
GLib.idle_add(self.manage_models_dialog.close) GLib.idle_add(self.manage_models_dialog.close)
GLib.idle_add(self.connection_error) GLib.idle_add(self.connection_error)
@ -579,9 +578,13 @@ class AlpacaWindow(Adw.ApplicationWindow):
title = model title = model
) )
thread = threading.Thread(target=self.pull_model_process, kwargs={"model": model}) thread = threading.Thread(target=self.pull_model_process, kwargs={"model": model})
overlay = Gtk.Overlay()
progress_bar = Gtk.ProgressBar( progress_bar = Gtk.ProgressBar(
valign = 3, valign = 2,
show_text = True show_text = False,
margin_start = 10,
margin_end = 10,
css_classes = ["osd", "horizontal", "bottom"]
) )
button = Gtk.Button( button = Gtk.Button(
icon_name = "media-playback-stop-symbolic", icon_name = "media-playback-stop-symbolic",
@ -590,10 +593,12 @@ class AlpacaWindow(Adw.ApplicationWindow):
css_classes = ["error"] css_classes = ["error"]
) )
button.connect("clicked", lambda button, model_name=model : dialogs.stop_pull_model(self, model_name)) button.connect("clicked", lambda button, model_name=model : dialogs.stop_pull_model(self, model_name))
model_row.add_suffix(progress_bar) #model_row.add_suffix(progress_bar)
model_row.add_suffix(button) model_row.add_suffix(button)
self.pulling_models[model] = {"row": model_row, "progress_bar": progress_bar} self.pulling_models[model] = {"row": model_row, "progress_bar": progress_bar, "overlay": overlay}
self.pulling_model_list_box.append(model_row) overlay.set_child(model_row)
overlay.add_overlay(progress_bar)
self.pulling_model_list_box.append(overlay)
thread.start() thread.start()
def update_list_available_models(self): def update_list_available_models(self):
@ -710,7 +715,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
self.new_chat_element(chat_name, True) self.new_chat_element(chat_name, True)
def stop_pull_model(self, model_name): def stop_pull_model(self, model_name):
self.pulling_models[model_name]['row'].get_parent().remove(self.pulling_models[model_name]['row']) self.pulling_models[model_name]['overlay'].get_parent().get_parent().remove(self.pulling_models[model_name]['overlay'].get_parent())
del self.pulling_models[model_name] del self.pulling_models[model_name]
def delete_model(self, model_name): def delete_model(self, model_name):