diff --git a/src/main.py b/src/main.py index 5e7fa1b..41a836a 100644 --- a/src/main.py +++ b/src/main.py @@ -234,7 +234,11 @@ class BavarderApplication(Adw.Application): if not self.setup_chat(): # NO MODELS: return _("Please download a model from Preferences by clicking on the Dot Menu at the top!") else: - with self.model.chat_session(): + for p in ["Hi", "Hello"]: + if p.lower() in prompt.lower(): + return _("Hello, I am Bavarder, a Chit-Chat AI") + system_template = """A chat between a curious user and an artificial intelligence assistant.""" + with self.model.chat_session(system_template): self.model.current_chat_session = chat["content"].copy() response = self.model.generate(prompt=prompt, top_k=1) else: diff --git a/src/threading.py b/src/threading.py index b8d614d..fbd6ccb 100644 --- a/src/threading.py +++ b/src/threading.py @@ -25,7 +25,7 @@ class KillableThread(threading.Thread): def localtrace(self, frame, event, arg): if self.killed: if event == 'line': - raise SystemExit() + raise Exception("Killed") return self.localtrace def kill(self): diff --git a/src/views/window.py b/src/views/window.py index e0c2fca..b49e1ca 100644 --- a/src/views/window.py +++ b/src/views/window.py @@ -374,8 +374,8 @@ class BavarderWindow(Adw.ApplicationWindow): del self.t self.toast.dismiss() except AttributeError: # nothing to stop - print("Nothing to stop") - except SystemExit: + pass + except Exception: self.t.join() del self.t self.toast.dismiss()