Merge pull request #949 from Capsize-Games/devastator

Devastator

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="airunner",
-    version="3.0.21",
+    version="3.1.0",
     author="Capsize LLC",
     description="A Stable Diffusion GUI",
     long_description=open("README.md", "r", encoding="utf-8").read(),

src/airunner/app.py

@@ -32,7 +32,6 @@
 from airunner.windows.main.settings_mixin import SettingsMixin
 from airunner.data.models.settings_models import ApplicationSettings, AIModels
 from airunner.windows.main.main_window import MainWindow
-from airunner.handlers.logger import Logger
 
 
 class App(
@@ -55,7 +54,6 @@ def __init__(
         """
         self.main_window_class_ = main_window_class or MainWindow
         self.app = None
-        self.logger = Logger(prefix=self.__class__.__name__)
         self.defendatron = defendatron
         self.splash = None
 
@@ -95,9 +93,7 @@ def create_paths(self):
                 "images"
             )
         ))
-        session = self.db_handler.get_db_session()
-        versions = session.query(distinct(AIModels.version)).filter(AIModels.category == 'stablediffusion').all()
-        session.close()
+        versions = self.session.query(distinct(AIModels.version)).filter(AIModels.category == 'stablediffusion').all()
         for version in versions:
             os.makedirs(
                 os.path.join(models_path, version[0], "embeddings"),
@@ -110,9 +106,7 @@ def create_paths(self):
         os.makedirs(images_path, exist_ok=True)
 
     def run_setup_wizard(self):
-        session = self.db_handler.get_db_session()
-        application_settings = session.query(ApplicationSettings).first()
-        session.close()
+        application_settings = self.session.query(ApplicationSettings).first()
         if application_settings.run_setup_wizard:
             AppInstaller()
 

src/airunner/app_installer.py

@@ -19,7 +19,6 @@
 from airunner.windows.download_wizard.download_wizard_window import DownloadWizardWindow
 from airunner.windows.main.settings_mixin import SettingsMixin
 from airunner.windows.setup_wizard.setup_wizard_window import SetupWizardWindow
-from airunner.handlers.logger import Logger
 
 
 class AppInstaller(
@@ -43,7 +42,6 @@ def __init__(
         self.download_wizard = None
         self.app = None
         self.close_on_cancel = close_on_cancel
-        self.logger = Logger(prefix=self.__class__.__name__)
 
         """
         Mediator and Settings mixins are initialized here, enabling the application