remove unused imports

clean up some docstrings

src/airunner/aihandler/llm/agent/actions/bash_execute.py

@@ -1,7 +1,7 @@
 import subprocess
 
 
-def bash_execute(self, command: str) -> str:
+def bash_execute(command: str) -> str:
     """
     Executes a bash command.
 
@@ -12,11 +12,9 @@ def bash_execute(self, command: str) -> str:
     :param command: str
     :return: str
     """
-    self.logger.debug(f"Executing bash command {command}")
     try:
         command = command.split(" ")
         result = subprocess.check_output(command, shell=False)
         return result.decode("utf-8")
     except Exception as e:
-        self.logger.error(e)
         return str(e)

src/airunner/aihandler/models/settings_db_handler.py

@@ -540,7 +540,7 @@ def get_chatbot_by_id(self, chatbot_id) -> Chatbot:
     def create_conversation(self):
         with self.get_db_session() as session:
             conversation = Conversation(
-                timestamp=datetime.datetime.utcnow(),
+                timestamp=datetime.datetime.now(datetime.timezone.utc),
                 title=""
             )
             session.add(conversation)

src/airunner/aihandler/models/settings_models.py

@@ -552,7 +552,7 @@ class WindowSettings(Base):
 class Conversation(Base):
     __tablename__ = 'conversations'
     id = Column(Integer, primary_key=True, autoincrement=True)
-    timestamp = Column(DateTime, default=datetime.datetime.utcnow)
+    timestamp = Column(DateTime, default=datetime.datetime.now(datetime.timezone.utc))
     title = Column(String, nullable=True)  # New column added
     messages = relationship("Message", back_populates="conversation", cascade="all, delete-orphan")
 
@@ -562,7 +562,7 @@ class Message(Base):
     id = Column(Integer, primary_key=True, autoincrement=True)
     role = Column(String, nullable=False)
     content = Column(String, nullable=False)
-    timestamp = Column(DateTime, default=datetime.datetime.utcnow)
+    timestamp = Column(DateTime, default=datetime.datetime.now(datetime.timezone.utc))
     conversation_id = Column(Integer, ForeignKey('conversations.id'))
     conversation = relationship("Conversation", back_populates="messages")
     name = Column(String, nullable=True)  # New column added
@@ -576,7 +576,7 @@ class Summary(Base):
     __tablename__ = 'summaries'
     id = Column(Integer, primary_key=True, autoincrement=True)
     content = Column(String, nullable=False)
-    timestamp = Column(DateTime, default=datetime.datetime.utcnow)
+    timestamp = Column(DateTime, default=datetime.datetime.now(datetime.timezone.utc))
     conversation_id = Column(Integer, ForeignKey('conversations.id'))
     conversation = relationship("Conversation", back_populates="summaries")
 

src/airunner/aihandler/stt/stt_handler.py

@@ -141,6 +141,7 @@ def run(
         """
         Run the model on the given inputs.
         :param inputs: str - The transcription of the audio data.
+        :param role: LLMChatRole - The role of the speaker.
         :return:
         """
         # Extract the tensor from the BatchFeature object

src/airunner/app_installer.py

@@ -11,8 +11,7 @@
     QObject, QTimer,
 )
 from PySide6.QtGui import (
-    Qt,
-    QWindow, QPixmap, QGuiApplication
+    Qt, QPixmap, QGuiApplication
 )
 from PySide6.QtWidgets import (
     QApplication, QSplashScreen,

src/airunner/widgets/canvas/canvas_widget.py

@@ -107,4 +107,4 @@ def do_draw(
         self.ui.canvas_container_size = self.ui.canvas_container.viewport().size()
 
     def save_image(self, image_path, image=None):
-        self.save_image(image_path, image, self.scene.items())
+        self.save_image(image_path, image)

src/airunner/widgets/canvas/custom_scene.py

@@ -274,7 +274,7 @@ def add_image_to_scene(
     ):
         """
         Adds a given image to the scene
-        :param image_data: dict containing the image to be added to the scene
+        :param image: Image object to add to the scene
         :param is_outpaint: bool indicating if the image is an outpaint
         :param outpaint_box_rect: QPoint indicating the root point of the image
         :param border_size: int indicating the size of the border

src/airunner/widgets/canvas/custom_view.py

@@ -348,6 +348,7 @@ def snap_to_grid(self, event: QMouseEvent, use_floor: bool = True) -> QMouseEven
         This is used to adjust the selection tool to the grid
         in real time during rubberband mode.
         :param event:
+        :param use_floor:
         :return:
         """
         if self.current_tool is CanvasToolName.SELECTION:

src/airunner/widgets/image/image_panel_widget.py

@@ -95,10 +95,6 @@ def load_files(self):
     def display_thumbnails(self):
         """
         Display thumbnails of images from the sorted_files list within the specified range.
-
-        Args:
-            start (int): The starting index of the range.
-            end (int): The ending index of the range.
         """
         for file in self.sorted_files[self.start:self.end]:
             if file.endswith(".png"):

src/airunner/windows/download_wizard/download_thread.py

@@ -33,9 +33,9 @@ def run(self):
                         #     repo_type=repo_type
                         # )
                         self.hf_downloader.download_model(
-                            path=model["path"],
-                            file_name=filename,
-                            callback=self.progress_updated.emit
+                            requested_path=model["path"],
+                            requested_file_name=filename,
+                            requested_callback=self.progress_updated.emit
                         )
                 else:
                     print("Skipping download for model with no files {}".format(model["name"]))

src/airunner/windows/main/ai_model_mixin.py

@@ -1,6 +1,5 @@
 from typing import List
 
-from airunner.aihandler.models.settings_models import AIModels
 from airunner.enums import SignalCode
 
 

src/airunner/windows/main/main_window.py

@@ -357,7 +357,7 @@ def on_bash_execute_signal(self, data: dict) -> str:
         """
         Takes a message from the LLM and strips bash commands from it.
         Passes bash command to the bash_execute function.
-        :param message:
+        :param data: dict
         :return:
         """
         args = data["args"]

src/airunner/windows/setup_wizard/ai_runner_license/ai_runner_license.py

@@ -1,5 +1,3 @@
-from PySide6.QtCore import Slot
-
 from airunner.windows.setup_wizard.user_agreement.agreement_page import AgreementPage
 from airunner.windows.setup_wizard.ai_runner_license.templates.airunner_license_ui import Ui_airunner_license
 

src/airunner/workers/worker.py

@@ -2,12 +2,12 @@
 import queue
 import threading
 
-from PySide6.QtCore import Signal, QThread, QSettings, QObject
+from PySide6.QtCore import Signal, QThread, QObject
 
 from airunner.enums import QueueType, SignalCode, WorkerState
 from airunner.aihandler.logger import Logger
 from airunner.mediator_mixin import MediatorMixin
-from airunner.settings import SLEEP_TIME_IN_MS, ORGANIZATION, APPLICATION_NAME
+from airunner.settings import SLEEP_TIME_IN_MS
 from airunner.windows.main.settings_mixin import SettingsMixin