May-25-2023, 01:41 AM
Not sure on why it happens as im still very new to python... Any suggestions or can you guys see any reason of why it might be happening??
import sys import os import cv2 from PyQt6.QtWidgets import QApplication, QLabel, QVBoxLayout, QPushButton, QScrollArea, QWidget, QMainWindow, QGridLayout, QFileDialog from PyQt6.QtGui import QPixmap, QImage, QPainter from PyQt6.QtCore import Qt, QThread, pyqtSignal class App(QMainWindow): def __init__(self, parent=None): super(App, self).__init__(parent) self.setWindowTitle("Character Creator") self.setFixedSize(600, 500) self.setStyleSheet("background-color: #3C3C3C;") self.categories = ['shirt', 'pants', 'shoes', 'accessories', 'eyes', 'hair', 'skin-colour'] # Add more categories as needed self.category_buttons = [] for i, category in enumerate(self.categories): button = QPushButton(category.upper(), self) button.move(10, 40 + i * 30) button.setFixedSize(100, 25) button.setStyleSheet("background-color: #1C1A1A; color: #EFEFEF;") button.clicked.connect(lambda _, c=category: self.show_category(c)) self.category_buttons.append(button) def show_category(self, category): category_root = CategoryWindow(self, category) category_root.show() class ImageLoaderThread(QThread): image_loaded = pyqtSignal(QPixmap, str) def __init__(self, category_folder, parent=None): super(ImageLoaderThread, self).__init__(parent) self.category_folder = category_folder def run(self): skin_parts = self.load_skin_parts(self.category_folder) for skin_part in skin_parts: pixmap = QPixmap(os.path.join(self.category_folder, skin_part)) self.image_loaded.emit(pixmap, skin_part) def load_skin_parts(self, folder): return os.listdir(folder) class CategoryWindow(QWidget): def __init__(self, parent=None, category=''): super(CategoryWindow, self).__init__(parent) self.setWindowTitle(category.upper()) self.setFixedSize(200, 500) self.setStyleSheet("background-color: #3C3C3C;") self.layout = QVBoxLayout() self.setLayout(self.layout) self.scrollArea = QScrollArea() self.scrollArea.setWidgetResizable(True) self.scrollAreaWidgetContents = QWidget(self.scrollArea) self.scrollAreaWidgetContents.setLayout(self.layout) self.scrollArea.setWidget(self.scrollAreaWidgetContents) self.layout.addWidget(self.scrollArea) category_folder = os.path.join(r'Desktop\MinecraftSkinEditor\clothes\alex', category) self.image_loader_thread = ImageLoaderThread(category_folder) self.image_loader_thread.image_loaded.connect(self.on_image_loaded) self.image_loader_thread.start() def on_image_loaded(self, pixmap, skin_part): label = ClickableQLabel() label.setPixmap(pixmap) label.clicked.connect(lambda sp=skin_part: self.on_skin_part_click(label, sp)) self.layout.addWidget(label) def on_skin_part_click(self, label, skin_part): print('Clicked on skin part:', skin_part) pixmap = label.pixmap().copy() # Open a dialog to select the skin part image file_dialog = QFileDialog() file_dialog.setWindowTitle("Select Skin Part Image") file_dialog.setFileMode(QFileDialog.FileMode.ExistingFile) file_dialog.setNameFilter("Images (*.png *.jpg *.jpeg *.gif)") if file_dialog.exec() == QFileDialog.DialogCode.Accepted: image_path = file_dialog.selectedFiles()[0] skin_part_image = cv2.imread(image_path) # Resize the skin part image to fit the preview resized_skin_part_image = cv2.resize(skin_part_image, (pixmap.width(), pixmap.height()), interpolation=cv2.INTER_LINEAR) # Convert the resized skin part image to QPixmap skin_part_qimage = QImage(resized_skin_part_image.data, resized_skin_part_image.shape[1], resized_skin_part_image.shape[0], QImage.Format_BGR888) skin_part_pixmap = QPixmap.fromImage(skin_part_qimage) # Paste the skin part image onto the preview image pixmap = pixmap.copy() # Create a copy to modify painter = QPainter(pixmap) painter.setCompositionMode(QPainter.CompositionMode.DestinationOver) painter.drawPixmap(0, 0, skin_part_pixmap) painter.end() # Update the label with the modified image label.setPixmap(pixmap) def create_complete_skin(self): skin_parts = [] # Collect the skin parts from the labels for label in self.findChildren(ClickableQLabel): pixmap = label.pixmap() if pixmap is not None: skin_parts.append(pixmap.toImage()) # Create a blank image for the complete skin complete_skin_width = 64 complete_skin_height = 64 complete_skin = QImage(complete_skin_width, complete_skin_height, QImage.Format_ARGB32) complete_skin.fill(Qt.GlobalColor.transparent) # Paste the skin parts onto the complete skin painter = QPainter(complete_skin) for i, skin_part in enumerate(skin_parts): x = i % 8 * 8 y = i // 8 * 8 painter.drawImage(x, y, skin_part) painter.end() # Save the complete skin save_path, _ = QFileDialog.getSaveFileName(self, "Save Complete Skin", "", "PNG Files (*.png)") if save_path: complete_skin.save(save_path) class ClickableQLabel(QLabel): clicked = pyqtSignal() def __init__(self, parent=None): QLabel.__init__(self, parent) def mousePressEvent(self, event): self.clicked.emit() if __name__ == "__main__": app = QApplication(sys.argv) window = App() window.show() sys.exit(app.exec())