268 lines
7.9 KiB
Python
268 lines
7.9 KiB
Python
# main.py
|
|
#
|
|
# Copyright 2023
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import sys
|
|
import gi
|
|
import time
|
|
|
|
gi.require_version('Gtk', '4.0')
|
|
gi.require_version('Adw', '1')
|
|
gi.require_version('Xdp', '1.0')
|
|
gi.require_version('GtkSource', '5')
|
|
|
|
from gi.repository import Gtk, Gio, Adw, Xdp, GLib
|
|
from .views.window import BavarderWindow
|
|
from .views.about_window import AboutWindow
|
|
from .views.preferences_window import PreferencesWindow
|
|
from .constants import app_id
|
|
from .llm import LLM
|
|
|
|
import json
|
|
import os
|
|
|
|
user_config_dir = os.environ.get(
|
|
"XDG_CONFIG_HOME", os.environ["HOME"] + "/.config"
|
|
)
|
|
|
|
user_data_dir = os.environ.get(
|
|
"XDG_DATA_HOME", os.environ["HOME"] + "/.local/share"
|
|
)
|
|
|
|
user_cache_dir = os.environ.get(
|
|
"XDG_CACHE_HOME", os.environ["HOME"] + "/.cache"
|
|
)
|
|
|
|
model_path = os.path.join(user_cache_dir, "bavarder", "models")
|
|
|
|
class BavarderApplication(Adw.Application):
|
|
"""The main application singleton class."""
|
|
|
|
|
|
models = set()
|
|
model = None
|
|
action_running_in_background = False
|
|
number_of_win = 0
|
|
|
|
def __init__(self):
|
|
super().__init__(application_id='io.github.Bavarder.Bavarder',
|
|
flags=Gio.ApplicationFlags.DEFAULT_FLAGS)
|
|
self.create_action("quit", self.on_quit, ["<primary>q"])
|
|
self.create_action("close", self.on_close, ["<primary>w"])
|
|
self.create_action('about', self.on_about_action)
|
|
self.create_action('preferences', self.on_preferences_action, ['<primary>comma'])
|
|
self.create_action('new_chat', self.on_new_chat_action, ["<primary>n"])
|
|
self.create_action('ask', self.on_ask, ["Return"])
|
|
self.create_action('new_window', self.on_new_window, ["<primary><shift>n"])
|
|
|
|
self.data_path = os.path.join(user_data_dir, "bavarder")
|
|
|
|
if not os.path.exists(self.data_path):
|
|
os.makedirs(self.data_path)
|
|
|
|
if not os.path.exists(model_path):
|
|
os.makedirs(model_path)
|
|
|
|
self.list_models()
|
|
|
|
self.data_path = os.path.join(self.data_path, "data.json")
|
|
|
|
self.data = {
|
|
"chats": [],
|
|
"models": {}
|
|
}
|
|
|
|
if os.path.exists(self.data_path):
|
|
try:
|
|
with open(self.data_path, "r", encoding="utf-8") as f:
|
|
self.data = json.load(f)
|
|
except Exception:
|
|
pass
|
|
|
|
self.settings = Gio.Settings(schema_id=app_id)
|
|
|
|
self.model_name = self.settings.get_string("model")
|
|
|
|
self.create_stateful_action(
|
|
"set_model",
|
|
GLib.VariantType.new("s"),
|
|
GLib.Variant("s", self.model_name),
|
|
self.on_set_model_action
|
|
)
|
|
|
|
self.bot_name = self.settings.get_string("bot-name")
|
|
self.user_name = self.settings.get_string("user-name")
|
|
|
|
self.user_cache_dir = user_cache_dir
|
|
self.llm = LLM(self)
|
|
self.model_settings = {
|
|
"max_tokens": 200,
|
|
"temperature": 0.7,
|
|
"top_p": 0.9,
|
|
"top_k": 40,
|
|
}
|
|
|
|
def on_set_model_action(self, action, *args):
|
|
Gio.SimpleAction.set_state(self.lookup_action("set_model"), args[0])
|
|
self.model_name = args[0].get_string()
|
|
if self.win and self.win.chat:
|
|
self.win.chat["model"] = self.model_name
|
|
self.win.threads_row_activated_cb()
|
|
|
|
def save(self):
|
|
with open(self.data_path, "w", encoding="utf-8") as f:
|
|
json.dump(self.data, f)
|
|
self.settings.set_string("model", self.model_name)
|
|
self.settings.set_string("bot-name", self.bot_name)
|
|
self.settings.set_string("user-name", self.user_name)
|
|
|
|
def on_quit(self, action, *args, **kwargs):
|
|
"""Called when the user activates the Quit action."""
|
|
self.save()
|
|
self.quit()
|
|
|
|
def on_close(self, action, *args, **kwargs):
|
|
if self.number_of_win == 1:
|
|
self.on_quit(action, *args, **kwargs)
|
|
else:
|
|
self.win.destroy()
|
|
self.number_of_win -= 1
|
|
|
|
def on_new_chat_action(self, widget, _):
|
|
chat_id = 0
|
|
for chat in self.data["chats"]:
|
|
if chat["id"] > chat_id:
|
|
chat_id = chat["id"]
|
|
chat_id += 1
|
|
chat = {
|
|
"id": chat_id,
|
|
"title": "New Chat " + str(chat_id),
|
|
"starred": False,
|
|
"content": [],
|
|
"system_prompt": "",
|
|
}
|
|
|
|
self.data["chats"].append(chat)
|
|
self.win.load_threads()
|
|
|
|
def do_activate(self):
|
|
"""Called when the application is activated.
|
|
|
|
We raise the application's main window, creating it if
|
|
necessary.
|
|
"""
|
|
self.new_window()
|
|
|
|
@property
|
|
def win(self):
|
|
"""The application's main window."""
|
|
return self.props.active_window
|
|
|
|
def new_window(self, window=None):
|
|
if window:
|
|
win = self.props.active_window
|
|
else:
|
|
win = BavarderWindow(application=self)
|
|
self.number_of_win += 1
|
|
|
|
|
|
win.connect("close-request", self.on_close)
|
|
|
|
win.load_model_selector()
|
|
win.present()
|
|
|
|
|
|
def on_new_window(self, widget, _):
|
|
self.new_window()
|
|
|
|
|
|
def on_about_action(self, widget, _):
|
|
"""Callback for the app.about action."""
|
|
about = AboutWindow(self.win)
|
|
about.present()
|
|
|
|
def on_preferences_action(self, widget, _):
|
|
"""Callback for the app.preferences action."""
|
|
self.preferences_window = PreferencesWindow(self.win)
|
|
self.preferences_window.present()
|
|
|
|
|
|
def create_action(self, name, callback, shortcuts=None):
|
|
"""Add an application action.
|
|
|
|
Args:
|
|
name: the name of the action
|
|
callback: the function to be called when the action is
|
|
activated
|
|
shortcuts: an optional list of accelerators
|
|
"""
|
|
action = Gio.SimpleAction.new(name, None)
|
|
action.connect("activate", callback)
|
|
self.add_action(action)
|
|
|
|
if shortcuts:
|
|
self.set_accels_for_action(f"app.{name}", shortcuts)
|
|
|
|
def create_stateful_action(self, name, parameter_type, initial_state, callback, shortcuts=None):
|
|
"""Add a stateful application action."""
|
|
|
|
action = Gio.SimpleAction.new_stateful(
|
|
name, parameter_type, initial_state)
|
|
action.connect("activate", callback)
|
|
|
|
self.add_action(action)
|
|
|
|
if shortcuts:
|
|
self.set_accels_for_action(f"app.{name}", shortcuts)
|
|
|
|
def on_ask(self, widget, _):
|
|
try:
|
|
self.win.on_ask()
|
|
except AttributeError:
|
|
pass
|
|
|
|
def ask(self, prompt, chat, callback, error_callback):
|
|
self.llm.ask(prompt, chat, callback, error_callback)
|
|
|
|
def check_network(self):
|
|
return False
|
|
|
|
def list_models(self):
|
|
self.models = set()
|
|
for root, dirs, files in os.walk(model_path):
|
|
for model in files:
|
|
if model.endswith(".litertlm"):
|
|
self.models.add(model)
|
|
|
|
def delete_model(self, model_name):
|
|
model_file = os.path.join(model_path, model_name)
|
|
if os.path.exists(model_file):
|
|
os.remove(model_file)
|
|
self.list_models()
|
|
self.win.load_model_selector()
|
|
|
|
def clear_all_chats(self):
|
|
self.data["chats"] = []
|
|
self.save()
|
|
self.win.load_threads()
|
|
|
|
def main(version):
|
|
"""The application's entry point."""
|
|
app = BavarderApplication()
|
|
return app.run(sys.argv)
|