Finalize design of standard image widget

src/airunner/styles/dark_theme/main.qss

@@ -630,6 +630,7 @@ QTableWidget::item:selected {
     background-color: @darkerprimary-color;
 }
 
+#standard_image_widget QPushButton,
 #image_widget QPushButton {
     border: 1px solid @darkerprimary-color;
 }

src/airunner/utils.py

@@ -476,6 +476,7 @@ def apply_opacity_to_image(image, target_opacity):
     target_opacity = 255 * target_opacity
     if target_opacity == 0:
         target_opacity = 1
+    image = image.convert("RGBA")
     r, g, b, a = image.split()
     a = a.point(lambda i: target_opacity if i > 0 else 0)
     image.putalpha(a)

src/airunner/widgets/canvas_plus/canvas_plus_widget.py

@@ -572,7 +572,6 @@ def handle_outpaint(self, outpaint_box_rect, outpainted_image, action=None, is_k
             new_image = Image.alpha_composite(new_image, new_image_b)
             new_image = Image.alpha_composite(new_image, new_image_a)
 
-        new_image.save("test.png")
         return new_image, image_root_point, image_pivot_point
 
     def load_image_from_path(self, image_path):

src/airunner/widgets/canvas_plus/standard_image_widget.py

@@ -12,6 +12,7 @@
 
 from PIL import Image
 from PIL import PngImagePlugin
+from PIL.ImageQt import ImageQt
 
 from airunner.widgets.canvas_plus.canvas_base_widget import CanvasBaseWidget
 from airunner.widgets.canvas_plus.templates.standard_image_widget_ui import Ui_standard_image_widget
@@ -35,9 +36,10 @@ def __init__(self, *args, **kwargs):
         self.ui.batch_container.hide()
         self.ui.delete_confirmation.hide()
         self.ui.tableWidget.hide()
+        self.ui.image_frame.hide()
+        self.ui.similar_groupbox.hide()
 
     def handle_image_data(self, data):
-        print(data)
         self.image_path = data["path"]
         images = data["images"]
         if len(images) == 1:
@@ -47,20 +49,17 @@ def handle_image_data(self, data):
             self.load_batch_images(images)
 
     def clear_batch_images(self):
-        for widget in self.ui.batch_container.layout().children():
+        for widget in self.ui.batch_container.findChildren(QLabel):
             widget.deleteLater()
 
     def load_batch_images(self, images):
+        self.ui.batch_container.show()
         self.clear_batch_images()
         images = images[:4]
         for image in images:
-            raw_data = image.tobytes("raw", "RGBA")
-            qimage = QImage(
-                raw_data, 
-                image.size[0], 
-                image.size[1], 
-                QImage.Format.Format_RGBA8888
-            )
+            # resize the image to 128x128
+            image = image.resize((128, 128))
+            qimage = ImageQt(image)
             pixmap = QPixmap.fromImage(qimage)
             label = QLabel()
             label.setPixmap(pixmap)
@@ -140,6 +139,8 @@ def set_pixmap(self, image_path=None, image=None):
             self.set_table_data(meta_data)
 
         self.ui.controls_container.show()
+        self.ui.image_frame.show()
+        self.ui.similar_groupbox.show()
 
     def handle_label_clicked(self, event):
         # create a popup window and show the full size image in it