add ci/cd pipeline (#2)

* add pre-commit

Signed-off-by: Niclas Wesemann <niclas.wesmann@motius.de>

Author: nwesem

.github/workflows/pre-commit.yml

@@ -0,0 +1,14 @@
+name: pre-commit
+
+on:
+  pull_request:
+  push:
+    branches: [main]
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-python@v3
+    - uses: pre-commit/action@v3.0.1

src/touchdesigner_plugin/stream_diffusion_server.py

@@ -1,19 +1,17 @@
-import glob
+import os
 import socket
 import struct
-import os
 import sys
-from typing import Literal, Dict, Optional
+from typing import Literal
 
 import cv2 as cv
 import numpy as np
 
 from touchdesigner_plugin.constants import HOST, JPEG_ENCODE_QUALITY_PERCENT, PORT
-from touchdesigner_plugin.utilities import get_random_image
 
 sys.path.append(os.path.join(os.path.dirname(__file__), "..", ".."))
 
-from utils.wrapper import StreamDiffusionWrapper
+from utils.wrapper import StreamDiffusionWrapper  # noqa: E402
 
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
@@ -25,7 +23,7 @@ def __init__(self):
         self.server_socket.listen(5)
 
         model_id_or_path: str = "KBlueLeaf/kohaku-v2.1"
-        lora_dict: Optional[Dict[str, float]] = None
+        lora_dict: dict[str, float] | None = None
         width: int = 512
         height: int = 512
         acceleration: Literal["none", "xformers", "tensorrt"] = "xformers"
@@ -54,12 +52,11 @@ def send_image(self, client_socket):
         # image = get_random_image(nb_circles=3, noise_background=False)
         image = np.array(self.stream())
 
-
         # NOTE: decrease JPEG_ENCODE_QUALITY_PERCENT to increase performance
         _, image_data = cv.imencode(
             ".jpg", image, [int(cv.IMWRITE_JPEG_QUALITY), JPEG_ENCODE_QUALITY_PERCENT]
         )
-    
+
         image_size = struct.pack("!I", len(image_data))
         client_socket.send(image_size)
         client_socket.send(image_data.tobytes())
@@ -78,7 +75,7 @@ def run(self):
                     prompt=prompt,
                     num_inference_steps=50,
                 )
-                
+
                 # Send image to client
                 self.send_image(client_socket)