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

UI tweaks

Browse Source
This commit is contained in:
jeffser 2024-06-26 16:38:00 -06:00
parent d457a1cfb8
commit 52449243ed
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/window.py
View File

@ -831,7 +831,6 @@ class AlpacaWindow(Adw.ApplicationWindow):
def pull_model(self, model): def pull_model(self, model):
if model in list(self.pulling_models.keys()): if model in list(self.pulling_models.keys()):
self.show_toast("info", 2, self.manage_models_overlay)
return return
if model in self.local_models: if model in self.local_models:
self.show_toast("info", 3, self.manage_models_overlay) self.show_toast("info", 3, self.manage_models_overlay)
@ -865,6 +864,7 @@ class AlpacaWindow(Adw.ApplicationWindow):
thread.start() thread.start()
def confirm_pull_model(self, model_name): def confirm_pull_model(self, model_name):
self.model_tag_list_box.unselect_all()
self.manage_models_title.set_title(_("Manage Models")) self.manage_models_title.set_title(_("Manage Models"))
self.create_model_button.set_visible(True) self.create_model_button.set_visible(True)
self.manage_models_back_button.set_visible(False) self.manage_models_back_button.set_visible(False)
@ -872,16 +872,19 @@ class AlpacaWindow(Adw.ApplicationWindow):
self.pull_model(model_name) self.pull_model(model_name)
def list_available_model_tags(self, model_name): def list_available_model_tags(self, model_name):
self.available_model_list_box.unselect_all()
self.manage_models_title.set_title(model_name.capitalize()) self.manage_models_title.set_title(model_name.capitalize())
self.create_model_button.set_visible(False) self.create_model_button.set_visible(False)
self.manage_models_back_button.set_visible(True) self.manage_models_back_button.set_visible(True)
self.model_tag_list_box.connect('row_selected', lambda list_box, row: self.confirm_pull_model(row.get_name()) if row else None)
self.model_tag_list_box.remove_all() self.model_tag_list_box.remove_all()
tags = self.available_models[model_name]['tags'] tags = self.available_models[model_name]['tags']
for tag_data in tags: for tag_data in tags:
if f"{model_name}:{tag_data[0]}" not in self.local_models: if f"{model_name}:{tag_data[0]}" not in self.local_models:
tag_row = Adw.ActionRow( tag_row = Adw.ActionRow(
title = tag_data[0], title = tag_data[0],
subtitle = tag_data[1] subtitle = tag_data[1],
name = f"{model_name}:{tag_data[0]}"
) )
pull_button = Gtk.Button( pull_button = Gtk.Button(
icon_name = "folder-download-symbolic", icon_name = "folder-download-symbolic",
@ -895,11 +898,13 @@ class AlpacaWindow(Adw.ApplicationWindow):
self.manage_models_carousel.scroll_to(self.manage_models_carousel.get_nth_page(1), True) self.manage_models_carousel.scroll_to(self.manage_models_carousel.get_nth_page(1), True)
def update_list_available_models(self): def update_list_available_models(self):
self.available_model_list_box.connect('row_selected', lambda list_box, row: self.list_available_model_tags(row.get_name()) if row else None)
self.available_model_list_box.remove_all() self.available_model_list_box.remove_all()
for name, model_info in self.available_models.items(): for name, model_info in self.available_models.items():
model = Adw.ActionRow( model = Adw.ActionRow(
title = f"<b>{name.capitalize()}</b> <small>by {model_info['author']}</small>", title = f"<b>{name.capitalize()}</b> <small>by {model_info['author']}</small>",
subtitle = f"<small>" + (_("(Image recognition capable)\n") if model_info["image"] else "") + f"{model_info['description']}</small>" subtitle = f"<small>" + (_("(Image recognition capable)\n") if model_info["image"] else "") + f"{model_info['description']}</small>",
name = name
) )
link_button = Gtk.Button( link_button = Gtk.Button(
icon_name = "globe-symbolic", icon_name = "globe-symbolic",

6
src/window.ui
View File

@ -648,7 +648,7 @@
</child> </child>
<child> <child>
<object class="GtkListBox" id="available_model_list_box"> <object class="GtkListBox" id="available_model_list_box">
<property name="selection-mode">none</property> <property name="selection-mode">single</property>
<style> <style>
<class name="boxed-list"/> <class name="boxed-list"/>
</style> </style>
@ -662,14 +662,14 @@
<object class="GtkScrolledWindow"> <object class="GtkScrolledWindow">
<child> <child>
<object class="GtkListBox" id="model_tag_list_box"> <object class="GtkListBox" id="model_tag_list_box">
<property name="valign">1</property>
<property name="margin-start">12</property> <property name="margin-start">12</property>
<property name="margin-end">12</property> <property name="margin-end">12</property>
<property name="margin-top">12</property> <property name="margin-top">12</property>
<property name="margin-bottom">12</property> <property name="margin-bottom">12</property>
<property name="selection-mode">none</property> <property name="selection-mode">single</property>
<style> <style>
<class name="boxed-list"/> <class name="boxed-list"/>
<class name="card"/>
</style> </style>
</object> </object>
</child> </child>