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

Added back regenerate button

Browse Source
This commit is contained in:
jeffser
2024-08-26 14:05:42 -06:00
parent d2271a7ade
commit db198e10c0
3 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/custom_widgets/chat_widget.py
View File

@@ -70,6 +70,7 @@ class chat(Gtk.ScrolledWindow):
)
self.messages = {}
self.welcome_screen = None
self.regenerate_button = None
self.busy = False
def stop_message(self):
@@ -180,7 +181,19 @@ class chat(Gtk.ScrolledWindow):
messages_dict[message_id]['files'] = files
return messages_dict
def show_regenerate_button(self, msg:message):
if self.regenerate_button:
self.remove(self.regenerate_button)
self.regenerate_button = Gtk.Button(
child=Adw.ButtonContent(
icon_name='update-symbolic',
label=_('Regenerate Response')
),
css_classes=["suggested-action"],
halign=3
)
self.regenerate_button.connect('clicked', lambda *_: msg.action_buttons.regenerate_message())
self.container.append(self.regenerate_button)
class chat_tab(Gtk.ListBoxRow):
__gtype_name__ = 'AlpacaChatTab'

5
src/custom_widgets/message_widget.py
View File

@@ -342,12 +342,13 @@ class action_buttons(Gtk.Box):
message_element = self.get_parent()
if not chat.busy:
message_element.set_text()
message_element.container.remove(message_element.footer)
if message_element.footer:
message_element.container.remove(message_element.footer)
message_element.remove_overlay(self)
message_element.action_buttons = None
history = window.convert_history_to_ollama(chat)[:list(chat.messages).index(message_element.message_id)]
data = {
"model": window.get_current_model(1),
"model": window.model_selector.get_model(),
"messages": history,
"options": {"temperature": window.model_tweaks["temperature"], "seed": window.model_tweaks["seed"]},
"keep_alive": f"{window.model_tweaks['keep_alive']}m"