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

Fixed stop button

Browse Source
This commit is contained in:
jeffser 2024-09-17 19:46:28 -06:00
parent dd29077499
commit 1ee36b113a
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/custom_widgets/message_widget.py
View File

@ -458,8 +458,11 @@ class message(Gtk.Overlay):
window.save_history(chat) window.save_history(chat)
else: else:
if self.spinner: if self.spinner:
self.container.remove(self.spinner) GLib.idle_add(self.container.remove, self.spinner)
self.spinner = None self.spinner = None
chat_tab = window.chat_list_box.get_tab_by_name(chat.get_name())
if chat_tab.spinner:
GLib.idle_add(chat_tab.spinner.set_visible, False)
sys.exit() sys.exit()
def set_text(self, text:str=None): def set_text(self, text:str=None):

2
src/window.py
View File

@ -554,7 +554,7 @@ Generate a title following these rules:
if self.regenerate_button: if self.regenerate_button:
GLib.idle_add(self.chat_list_box.get_current_chat().remove, self.regenerate_button) GLib.idle_add(self.chat_list_box.get_current_chat().remove, self.regenerate_button)
try: try:
response = self.ollama_instance.request("POST", "api/chat", json.dumps(data), lambda data, message_element=message_element: GLib.idle_add(message_element.update_message, data)) response = self.ollama_instance.request("POST", "api/chat", json.dumps(data), lambda data, message_element=message_element: message_element.update_message(data))
if response.status_code != 200: if response.status_code != 200:
raise Exception('Network Error') raise Exception('Network Error')
except Exception as e: except Exception as e: