Skip to content

refactor pod narrację #26

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gitesthuman wants to merge 2 commits into
base: staging
Choose a base branch
from
Open

refactor pod narrację #26

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e3666ed
refactor pod narrację
gitesthuman Dec 3, 2023
5452b6c
init modułu logic
gitesthuman Dec 3, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
0 assets/icon.png → assets/icons/icon.png
View file
Open in desktop
File renamed without changes
19 changes: 0 additions & 19 deletions init.py
View file
Open in desktop

This file was deleted.

0 components/__init__.py → logic/__init__.py
View file
Open in desktop
File renamed without changes.
128 changes: 14 additions & 114 deletions main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,119 +1,19 @@
from PyQt6 import QtCore
from PyQt6.QtGui import QIcon, QPixmap
from PyQt6.QtWidgets import QMainWindow, QHBoxLayout, QWidget, QVBoxLayout, QPushButton, QStackedLayout
import sys
import os

from logic.style_transfer import StyleTransfer
from widgets.home import HomeMenu
from widgets.library import LibraryMenu
from widgets.morphing import MorphingMenu
from widgets.settings import SettingsMenu
from widgets.style_image import StyleImageMenu
from widgets.style_video import StyleVideoMenu
from PyQt6.QtWidgets import QApplication
from widgets.main_window import MainWindow


class MainWindow(QMainWindow):
def sync():
app.sync()

def __init__(self):
super().__init__()

# initial configuration
self.setWindowTitle("neuroART")
self.resize(1400, 720)
self.setMinimumSize(QtCore.QSize(1360, 720))
self.central_widget = QWidget()
self.setCentralWidget(self.central_widget)

# initialize menus
self.home_menu = HomeMenu()
self.style_image_menu = StyleImageMenu()
self.style_video_menu = StyleVideoMenu()
self.morphing_menu = MorphingMenu()
self.settings_menu = SettingsMenu()
self.library_menu = LibraryMenu()

# main window layout
self.main_window_layout = QHBoxLayout()
self.central_widget.setLayout(self.main_window_layout)

# left menu
self.main_window_layout.addWidget(LeftMenu())

# stacked layout
self.stacked_layout = QStackedLayout()
self.main_window_layout.addLayout(self.stacked_layout)
self.stacked_layout.addWidget(self.style_image_menu)
self.stacked_layout.addWidget(self.style_video_menu)
self.stacked_layout.addWidget(self.home_menu)
self.stacked_layout.addWidget(self.library_menu)
self.stacked_layout.addWidget(self.settings_menu)
self.stacked_layout.addWidget(self.morphing_menu)

self.stacked_layout.setCurrentWidget(self.home_menu)

# load all models
StyleTransfer.load_models()

def change_current_widget(self, widget: QWidget):
self.stacked_layout.setCurrentWidget(widget)


class LeftMenu(QWidget):
def __init__(self):
super().__init__()

self.left_menu_layout = QVBoxLayout()
self.setLayout(self.left_menu_layout)

self.left_menu_layout.addWidget(MenuButton("Home", "assets/icons/home.png",
button_name='btn_leftmenu_home'))
self.left_menu_layout.addWidget(
MenuButton("Style Image", "assets/icons/shuffle.png",
button_name='btn_leftmenu_style_image'))
self.left_menu_layout.addWidget(
MenuButton("Style Video", "assets/icons/shuffle.png",
button_name='btn_leftmenu_style_video'))
self.left_menu_layout.addWidget(
MenuButton("Morphing", "assets/icons/color-filter.png",
button_name='btn_leftmenu_morphing'))
self.left_menu_layout.addWidget(
MenuButton("Library", "assets/icons/book.png", button_name='btn_leftmenu_library'))
self.left_menu_layout.addWidget(
MenuButton("Settings", "assets/icons/settings.png", button_name='btn_leftmenu_settings'))

self.left_menu_layout.addStretch()


class MenuButton(QPushButton):

def __init__(self, label, image_path, button_name: str = ''):
super().__init__()

icon = QIcon(QPixmap(image_path))
self.setIcon(icon)
self.setText(label)
self.setStyleSheet('text-align: left; padding: 5px 12px;')
self.setObjectName(button_name)
self.clicked.connect(self.button_click)

def button_click(self):
button = self.sender()
button_name = button.objectName()

window = MainWindow.window(self)
if button_name == 'btn_leftmenu_home':
window.change_current_widget(window.home_menu)

if button_name == 'btn_leftmenu_style_image':
window.change_current_widget(window.style_image_menu)

if button_name == 'btn_leftmenu_style_video':
window.change_current_widget(window.style_video_menu)

if button_name == 'btn_leftmenu_morphing':
window.change_current_widget(window.morphing_menu)

if button_name == 'btn_leftmenu_library':
window.change_current_widget(window.library_menu)

if button_name == 'btn_leftmenu_settings':
window.change_current_widget(window.settings_menu)
app = QApplication(sys.argv)
os.environ["theme"] = "light"
os.environ["TRAIN_EPOCHS"] = "100"
os.environ["INTERPOLATION"] = "TRUE"
os.environ["INTERPOLATION_STEP"] = "3"
window = MainWindow()
window.show()
app.exec()
2 changes: 1 addition & 1 deletion widgets/home.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ def __init__(self):
self.setLayout(self.home_layout)

self.logo = QLabel()
pixmap = QPixmap(os.path.abspath('assets/icon.png'))
pixmap = QPixmap(os.path.abspath('assets/icons/icon.png'))
scaled_pixmap = pixmap.scaled(self.logo.size() * 0.8, Qt.AspectRatioMode.KeepAspectRatio,
Qt.TransformationMode.SmoothTransformation)
self.logo.setPixmap(scaled_pixmap)
Expand Down
119 changes: 119 additions & 0 deletions widgets/main_window.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,119 @@
from PyQt6 import QtCore
from PyQt6.QtGui import QIcon, QPixmap
from PyQt6.QtWidgets import QMainWindow, QHBoxLayout, QWidget, QVBoxLayout, QPushButton, QStackedLayout

from logic.style_transfer import StyleTransfer
from .home import HomeMenu
from .library import LibraryMenu
from .morphing import MorphingMenu
from .settings import SettingsMenu
from .style_image import StyleImageMenu
from .style_video import StyleVideoMenu


class MainWindow(QMainWindow):

def __init__(self):
super().__init__()

# initial configuration
self.setWindowTitle("neuroART")
self.resize(1400, 720)
self.setMinimumSize(QtCore.QSize(1360, 720))
self.central_widget = QWidget()
self.setCentralWidget(self.central_widget)

# initialize menus
self.home_menu = HomeMenu()
self.style_image_menu = StyleImageMenu()
self.style_video_menu = StyleVideoMenu()
self.morphing_menu = MorphingMenu()
self.settings_menu = SettingsMenu()
self.library_menu = LibraryMenu()

# main window layout
self.main_window_layout = QHBoxLayout()
self.central_widget.setLayout(self.main_window_layout)

# left menu
self.main_window_layout.addWidget(LeftMenu())

# stacked layout
self.stacked_layout = QStackedLayout()
self.main_window_layout.addLayout(self.stacked_layout)
self.stacked_layout.addWidget(self.style_image_menu)
self.stacked_layout.addWidget(self.style_video_menu)
self.stacked_layout.addWidget(self.home_menu)
self.stacked_layout.addWidget(self.library_menu)
self.stacked_layout.addWidget(self.settings_menu)
self.stacked_layout.addWidget(self.morphing_menu)

self.stacked_layout.setCurrentWidget(self.home_menu)

# load all models
StyleTransfer.load_models()

def change_current_widget(self, widget: QWidget):
self.stacked_layout.setCurrentWidget(widget)


class LeftMenu(QWidget):
def __init__(self):
super().__init__()

self.left_menu_layout = QVBoxLayout()
self.setLayout(self.left_menu_layout)

self.left_menu_layout.addWidget(MenuButton("Home", "../assets/icons/home.png",
button_name='btn_leftmenu_home'))
self.left_menu_layout.addWidget(
MenuButton("Style Image", "../assets/icons/shuffle.png",
button_name='btn_leftmenu_style_image'))
self.left_menu_layout.addWidget(
MenuButton("Style Video", "../assets/icons/shuffle.png",
button_name='btn_leftmenu_style_video'))
self.left_menu_layout.addWidget(
MenuButton("Morphing", "../assets/icons/color-filter.png",
button_name='btn_leftmenu_morphing'))
self.left_menu_layout.addWidget(
MenuButton("Library", "../assets/icons/book.png", button_name='btn_leftmenu_library'))
self.left_menu_layout.addWidget(
MenuButton("Settings", "../assets/icons/settings.png", button_name='btn_leftmenu_settings'))

self.left_menu_layout.addStretch()


class MenuButton(QPushButton):

def __init__(self, label, image_path, button_name: str = ''):
super().__init__()

icon = QIcon(QPixmap(image_path))
self.setIcon(icon)
self.setText(label)
self.setStyleSheet('text-align: left; padding: 5px 12px;')
self.setObjectName(button_name)
self.clicked.connect(self.button_click)

def button_click(self):
button = self.sender()
button_name = button.objectName()

window = MainWindow.window(self)
if button_name == 'btn_leftmenu_home':
window.change_current_widget(window.home_menu)

if button_name == 'btn_leftmenu_style_image':
window.change_current_widget(window.style_image_menu)

if button_name == 'btn_leftmenu_style_video':
window.change_current_widget(window.style_video_menu)

if button_name == 'btn_leftmenu_morphing':
window.change_current_widget(window.morphing_menu)

if button_name == 'btn_leftmenu_library':
window.change_current_widget(window.library_menu)

if button_name == 'btn_leftmenu_settings':
window.change_current_widget(window.settings_menu)
6 changes: 3 additions & 3 deletions widgets/morphing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def __init__(self):
self.slider.setDisabled(True)

def handle_slider_value_change(self):
from main import MainWindow
from widgets.main_window import MainWindow
window = MainWindow.window(self)

morphing_menu = window.morphing_menu
Expand All @@ -173,7 +173,7 @@ def start_morphing_training(self):
self.save_button.setDisabled(True)
self.train_button.setDisabled(True)

from main import MainWindow
from widgets.main_window import MainWindow
window = MainWindow.window(self)
left_path = window.morphing_menu.left_container.image_path
right_path = window.morphing_menu.right_container.image_path
Expand All @@ -196,7 +196,7 @@ def start_morphing_training(self):
self.morphing_thread.start()

def morphing_training_finished(self, frames: list):
from main import MainWindow
from widgets.main_window import MainWindow
window = MainWindow.window(self)
morphing_menu = window.morphing_menu

Expand Down