Merge pull request #307 from FrancescoCaracciolo/master

Add per chat profiles

Author: qwersyk

data/io.github.qwersyk.Newelle.gschema.xml

@@ -176,5 +176,8 @@
         <key name="editor-color-scheme" type="s">
           <default>"Adwaita-dark"</default>
         </key>
+        <key name="remember-profile" type="b">
+          <default>false</default>
+        </key>
 	</schema>
 </schemalist>

src/controller.py

@@ -342,6 +342,7 @@ def load_settings(self, settings):
         self.virtualization = settings.get_boolean("virtualization")
         self.memory = settings.get_int("memory")
         self.hidden_files = settings.get_boolean("hidden-files")
+        self.remember_profile = settings.get_boolean("remember-profile")
         self.reverse_order = settings.get_boolean("reverse-order")
         self.remove_thinking = settings.get_boolean("remove-thinking")
         self.auto_generate_name = settings.get_boolean("auto-generate-name")

src/ui/settings.py

@@ -184,6 +184,12 @@ def update_zoom(x,y):
         self.settings.bind("hidden-files", switch, 'active', Gio.SettingsBindFlags.DEFAULT)
         self.interface.add(row)
 
+        row = Adw.ActionRow(title=_("Remember assistant profile per chat"), subtitle=_("When changing chat, the profile corresponding to the last generation is selected"))
+        switch = Gtk.Switch(valign=Gtk.Align.CENTER)
+        row.add_suffix(switch)
+        self.settings.bind("remember-profile", switch, 'active', Gio.SettingsBindFlags.DEFAULT)
+        self.interface.add(row)
+
         row = Adw.ActionRow(title=_("Send with ENTER"), subtitle=_("If enabled, messages will be sent with ENTER, to go to a new line use CTRL+ENTER. If disabled, messages will be sent with SHIFT+ENTER, and newline with enter"))
         switch = Gtk.Switch(valign=Gtk.Align.CENTER)
         row.add_suffix(switch)

src/window.py

@@ -1878,6 +1878,9 @@ def chose_chat(self, button, *a):
         old_chat_id = self.chat_id
         self.chat_id = int(button.get_name())
         self.chat = self.chats[self.chat_id]["chat"]
+        # Change profile 
+        if self.controller.newelle_settings.remember_profile and "profile" in self.chats[self.chat_id]:
+            self.switch_profile(self.chats[self.chat_id]["profile"])
         self.update_history()
         if old_chat_id > self.chat_id:
             self.chat_stack.set_transition_type(Gtk.StackTransitionType.SLIDE_UP)
@@ -2154,6 +2157,9 @@ def get_variable(self, name:str):
 
     def send_message(self, manual=True):
         """Send a message in the chat and get bot answer, handle TTS etc"""
+        # Save profile for generation 
+        self.chats[self.chat_id]["profile"] = self.current_profile
+
         GLib.idle_add(self.hide_placeholder)
         if manual:
             self.auto_run_times = 0