Skip to content

develop changes: save / reload #17

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

Merged
samamou merged 4 commits into from
Feb 2, 2026
Merged

develop changes: save / reload #17

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
4b335b3
ui: add a model training view
jcelerier Feb 2, 2026
9d44494
Allow to save / reload, plus minor UI adjustments
jcelerier Feb 2, 2026
262d451
ui: add a check for required downloads on start
jcelerier Feb 2, 2026
23d11e1
update app desc
Feb 2, 2026
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
2 changes: 1 addition & 1 deletion .github/workflows/build.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ jobs:
app-organization: "SAT"
app-domain: "sat.qc.ca"
app-identifier: "ca.qc.sat.koaia"
app-description: "pose and blob detection software"
app-description: "A tool for exploring generative ai"
app-copyright: "Société des Arts Technologiques"
app-version: "1.0"
app-png: "qml/koaia/resources/images/koaia_logo.png"
Expand Down
321 changes: 318 additions & 3 deletions qml/Main.qml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,68 @@ ApplicationWindow {
title: "koaia"
property var logger

// CUDA availability check
readonly property bool hasCuda: System.availableCudaDevice() > 0

// Required packages
readonly property var requiredPackages: [
{ uuid: "6aa9d4fd-a45b-4736-af8a-56168b48af42", name: "uv (Python package manager)" },
{ uuid: "28d97cbc-bc84-4e74-9bc2-fe0f0fccf87c", name: "CUDA libraries" },
{ uuid: "d0add344-cb23-4d3f-817b-3619505565d1", name: "LibreDiffusion" }
]

// Track installed packages
property var installedPackageUuids: []
property bool allPackagesInstalled: false
property bool isCheckingPackages: true
property bool needsRestart: false // True when packages were installed this session

function checkInstalledPackages() {
var installed = Library.installedPackages()
var uuids = []
for (var i = 0; i < installed.length; i++) {
uuids.push(installed[i].uuid)
}

var previouslyInstalled = installedPackageUuids.length
installedPackageUuids = uuids

// Check if all required packages are installed
var allInstalled = true
for (var j = 0; j < requiredPackages.length; j++) {
if (uuids.indexOf(requiredPackages[j].uuid) === -1) {
allInstalled = false
break
}
}

// If packages were installed this session, mark that restart is needed
if (allInstalled && !allPackagesInstalled && previouslyInstalled > 0) {
needsRestart = true
}

allPackagesInstalled = allInstalled
isCheckingPackages = false
}

function isPackageInstalled(uuid) {
return installedPackageUuids.indexOf(uuid) !== -1
}

Component.onCompleted: {
Library.refreshAvailablePackages()
checkInstalledPackages()
}

// Timer to periodically check package installation status
Timer {
id: packageCheckTimer
interval: 2000
repeat: true
running: !allPackagesInstalled
onTriggered: checkInstalledPackages()
}

DarkStyle {
id: dark_style
}
Expand Down Expand Up @@ -60,18 +122,259 @@ ApplicationWindow {

property int currentViewIndex: 0
readonly property int mainViewIndex: 0
readonly property int logViewIndex: 1
readonly property int modelViewIndex: 1
readonly property int logViewIndex: 2

AboutDialog {
id: aboutDialog
parentWindow: mainWindow
}

// Error message when no CUDA device is available
Rectangle {
id: noCudaMessage
anchors.fill: parent
color: appStyle.backgroundColor
visible: !hasCuda

ColumnLayout {
anchors.centerIn: parent
spacing: 20

Image {
Layout.preferredWidth: 80
Layout.preferredHeight: 80
Layout.alignment: Qt.AlignHCenter
source: "koaia/resources/images/koaia_logo.png"
fillMode: Image.PreserveAspectFit
}

Label {
Layout.alignment: Qt.AlignHCenter
text: "CUDA Device Not Found"
font.pixelSize: appStyle.fontSizeTitle
font.bold: true
color: appStyle.textColor
}

Label {
Layout.alignment: Qt.AlignHCenter
Layout.maximumWidth: 400
text: "This application requires an NVIDIA GPU with CUDA support.\n\nPlease ensure you have an NVIDIA GPU of at least generation RTX 3xxx (Ampere architecture or newer) installed with the appropriate drivers."
font.pixelSize: appStyle.fontSizeBody
color: appStyle.textColorSecondary
wrapMode: Text.WordWrap
horizontalAlignment: Text.AlignHCenter
}

Button {
Layout.alignment: Qt.AlignHCenter
Layout.topMargin: 10
text: "Exit"
font.pixelSize: appStyle.fontSizeBody
onClicked: Qt.exit(0)
}
}
}

// Missing packages screen
Rectangle {
id: missingPackagesScreen
anchors.fill: parent
color: appStyle.backgroundColor
visible: hasCuda && !allPackagesInstalled

ColumnLayout {
anchors.centerIn: parent
spacing: 20
width: Math.min(500, parent.width - 40)

Image {
Layout.preferredWidth: 80
Layout.preferredHeight: 80
Layout.alignment: Qt.AlignHCenter
source: "koaia/resources/images/koaia_logo.png"
fillMode: Image.PreserveAspectFit
}

Label {
Layout.alignment: Qt.AlignHCenter
text: "Required Packages"
font.pixelSize: appStyle.fontSizeTitle
font.bold: true
color: appStyle.textColor
}

Label {
Layout.alignment: Qt.AlignHCenter
Layout.fillWidth: true
text: "The following packages are required to run Koaia. Please install any missing packages. This may take several minutes per package."
font.pixelSize: appStyle.fontSizeBody
color: appStyle.textColorSecondary
wrapMode: Text.WordWrap
horizontalAlignment: Text.AlignHCenter
}

// Package list
Rectangle {
Layout.fillWidth: true
Layout.preferredHeight: packageListColumn.implicitHeight + 20
color: appStyle.backgroundColorSecondary
border.color: appStyle.borderColor
border.width: 1
radius: appStyle.borderRadius

ColumnLayout {
id: packageListColumn
anchors.fill: parent
anchors.margins: 10
spacing: 10

Repeater {
model: requiredPackages

RowLayout {
Layout.fillWidth: true
spacing: 10

// Status icon
Rectangle {
width: 24
height: 24
radius: 12
color: isPackageInstalled(modelData.uuid) ? "#4CAF50" : appStyle.borderColor

Label {
anchors.centerIn: parent
text: isPackageInstalled(modelData.uuid) ? "\u2713" : ""
font.pixelSize: 14
font.bold: true
color: "white"
}
}

// Package name
Label {
Layout.fillWidth: true
text: modelData.name
font.pixelSize: appStyle.fontSizeBody
color: appStyle.textColor
}

// Status / Install button
Button {
visible: !isPackageInstalled(modelData.uuid)
text: "Install"
font.pixelSize: appStyle.fontSizeSmall
onClicked: Library.installPackage(modelData.uuid)
}

Label {
visible: isPackageInstalled(modelData.uuid)
text: "Installed"
font.pixelSize: appStyle.fontSizeSmall
color: "#4CAF50"
}
}
}
}
}

// Install all button
Button {
Layout.alignment: Qt.AlignHCenter
Layout.topMargin: 10
text: "Install All Missing Packages"
font.pixelSize: appStyle.fontSizeBody
visible: !allPackagesInstalled
onClicked: {
for (var i = 0; i < requiredPackages.length; i++) {
if (!isPackageInstalled(requiredPackages[i].uuid)) {
Library.installPackage(requiredPackages[i].uuid)
}
}
}
}

// Restart required message
Rectangle {
Layout.fillWidth: true
Layout.topMargin: 10
visible: allPackagesInstalled && needsRestart
color: "#2E7D32"
radius: appStyle.borderRadius
implicitHeight: restartColumn.implicitHeight + 20

ColumnLayout {
id: restartColumn
anchors.fill: parent
anchors.margins: 10
spacing: 10

Label {
Layout.alignment: Qt.AlignHCenter
text: "All packages installed successfully!"
font.pixelSize: appStyle.fontSizeBody
font.bold: true
color: "white"
}

Label {
Layout.alignment: Qt.AlignHCenter
Layout.fillWidth: true
text: "Please restart Koaia to complete the setup."
font.pixelSize: appStyle.fontSizeBody
color: "white"
wrapMode: Text.WordWrap
horizontalAlignment: Text.AlignHCenter
}

Button {
Layout.alignment: Qt.AlignHCenter
text: "Quit Koaia"
font.pixelSize: appStyle.fontSizeBody
onClicked: Qt.exit(0)
}
}
}

// Progress indicator
Label {
Layout.alignment: Qt.AlignHCenter
visible: !allPackagesInstalled
text: "Checking installation status..."
font.pixelSize: appStyle.fontSizeSmall
color: appStyle.textColorSecondary

SequentialAnimation on opacity {
loops: Animation.Infinite
NumberAnimation { from: 1.0; to: 0.3; duration: 800 }
NumberAnimation { from: 0.3; to: 1.0; duration: 800 }
}
}

// Restart note (shown while installing)
Label {
Layout.alignment: Qt.AlignHCenter
Layout.fillWidth: true
Layout.topMargin: 10
visible: !allPackagesInstalled
text: "Note: After all packages are installed, you will need to restart Koaia."
font.pixelSize: appStyle.fontSizeSmall
font.italic: true
color: appStyle.textColorSecondary
wrapMode: Text.WordWrap
horizontalAlignment: Text.AlignHCenter
}
}
}

RowLayout {
id: rowLayout
anchors.fill: parent
spacing: 0

visible: hasCuda && allPackagesInstalled

Rectangle {
id: sidebar
width: appStyle.sidebarWidth
Expand Down Expand Up @@ -111,6 +414,15 @@ ApplicationWindow {
onClicked: currentViewIndex = mainViewIndex
}

CustomButton {
id: modelButton
text: "MODEL"
Layout.fillWidth: true
Layout.topMargin: appStyle.spacing
isActive: currentViewIndex === modelViewIndex
onClicked: currentViewIndex = modelViewIndex
}

CustomButton {
id: logButton
text: "LOGS"
Expand All @@ -128,12 +440,15 @@ ApplicationWindow {

StackLayout {
id: stackView
enabled: hasCuda
Layout.fillWidth: true
Layout.fillHeight: true
currentIndex: currentViewIndex

MainView {
}
ModelView {
}
LogView {
}
}
Expand Down
Loading
Loading