refactor: update model and code organization

- Update Fluxlyptus model to newer version
- Reorganize imports in multiple files
- Remove unnecessary blank lines
- Improve code formatting and consistency

src/gui.py

@@ -27,7 +27,6 @@
     QVBoxLayout,
     QWidget,
 )
-
 from utils import ImageGeneratorThread, ImageLoader, TokenCounter
 
 

src/image_generator.py

@@ -3,9 +3,10 @@
 
 load_dotenv()
 
+
 class ImageGenerator:
     def __init__(self):
-        self.model = "rtuszik/fluxlyptus:4e304b52ad6745623fb29f3250d89df23ac38b42734887d9e0a4b3a31c648472"
+        self.model = "rtuszik/fluxlyptus:b23b9b488de7af95eba09786ef3156d345d979024712f54b3e5a32d61f14e568"
 
     def generate_images(self, params):
         try:
@@ -14,5 +15,6 @@ def generate_images(self, params):
         except Exception as e:
             raise ImageGenerationError(f"Error generating images: {str(e)}")
 
+
 class ImageGenerationError(Exception):
-    pass
+    pass

src/main.py

@@ -1,7 +1,6 @@
-from PyQt6.QtWidgets import QApplication
-
 from gui import ImageGeneratorGUI
 from image_generator import ImageGenerator
+from PyQt6.QtWidgets import QApplication
 
 
 def main():

src/utils.py

@@ -1,10 +1,13 @@
 import os
-from PyQt6.QtCore import QRunnable, QThread, pyqtSignal, QObject
-from PyQt6.QtWidgets import QWidget, QVBoxLayout, QLabel
+
+from PyQt6.QtCore import QObject, QRunnable, QThread, pyqtSignal
+from PyQt6.QtWidgets import QLabel, QVBoxLayout, QWidget
+
 
 class ImageLoaderSignals(QObject):
     finished = pyqtSignal(list)
 
+
 class ImageLoader(QRunnable):
     def __init__(self, folder_path):
         super().__init__()
@@ -21,6 +24,7 @@ def run(self):
         images.sort(key=lambda x: x[1], reverse=True)
         self.signals.finished.emit([img[0] for img in images])
 
+
 class ImageGeneratorThread(QThread):
     finished = pyqtSignal(list)
     error = pyqtSignal(str)
@@ -37,6 +41,7 @@ def run(self):
         except Exception as e:
             self.error.emit(str(e))
 
+
 class TokenCount:
     def __init__(self, model_name):
         self.model_name = model_name
@@ -45,6 +50,7 @@ def num_tokens_from_string(self, string: str) -> int:
         # This is a simplified implementation. You might want to use a proper tokenizer here.
         return len(string.split())
 
+
 class TokenCounter(QWidget):
     def __init__(self, text_edit, *args, **kwargs):
         super().__init__(*args, **kwargs)