Support Windows

package.json

@@ -1,7 +1,7 @@
 {
   "name": "imagenie",
   "private": true,
-  "version": "0.2.0",
+  "version": "0.2.2",
   "type": "module",
   "scripts": {
     "dev": "vite",

src-tauri/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "imagenie"
-version = "0.2.0"
+version = "0.2.2"
 description = "A Tauri App"
 authors = ["zhongwei"]
 edition = "2021"
@@ -35,7 +35,7 @@ tauri-plugin-process = "2"
 tauri-plugin-shell = "2"
 tauri-plugin-window-state = "2.0.1"
 tracing = { version = "0.1.40", features = ["log"] }
-ort = { version = "2.0.0-rc.8" }
+ort = { version = "2.0.0-rc.9" }
 image = "0.25.4"
 ndarray = "0.16.1"
 half = "2.3"

src-tauri/src/image/model.rs

@@ -1,5 +1,5 @@
 use image::DynamicImage;
-use ort::{inputs, GraphOptimizationLevel, Session, Value};
+use ort::{inputs, session::builder::GraphOptimizationLevel, session::Session, value::Value};
 use std::marker::PhantomData;
 
 use crate::image::error::ImageProcessingError;

src-tauri/src/image/processor.rs

@@ -5,7 +5,7 @@ use std::marker::PhantomData;
 use super::{ImageModel, ImageProcessingError};
 
 pub struct ModelProcessor<M: ImageModel + Send + Sync> {
-    session: ort::Session,
+    session: ort::session::Session,
     _phantom: PhantomData<M>,
 }
 

src-tauri/src/image/types.rs

@@ -42,8 +42,8 @@ pub trait NumericType:
     + Send
     + Sync
     + std::fmt::Debug
-    + ort::IntoTensorElementType
-    + ort::PrimitiveTensorElementType
+    // + ort::IntoTensorElementType
+    // + ort::PrimitiveTensorElementType
     + 'static
 {
     fn from_f32(v: f32) -> Self;

src-tauri/tauri.conf.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://schema.tauri.app/config/2",
   "productName": "imagenie",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "identifier": "com.imagenie.app",
   "build": {
     "beforeDevCommand": "yarn dev",
@@ -24,10 +24,10 @@
     "windows": [],
     "security": {
       "csp": {
-        "default-src": "'self'",
         "style-src": "'self' 'unsafe-inline'",
-        "img-src": "'self' asset: https://raw.githubusercontent.com data:",
-        "connect-src": "'self' ipc: https://api.github.com",
+        "img-src": "'self' asset: http://asset.localhost https://raw.githubusercontent.com blob: data:",
+        "default-src": "'self' asset:",
+        "connect-src": "'self' tauri: http://tauri.localhost ipc: http://ipc.localhost http://localhost:8023 ipc: https://api.github.com https://api.iconify.design https://api.unisvg.com https://api.simplesvg.com",
         "script-src": "'self'"
       },
       "assetProtocol": {
@@ -54,6 +54,16 @@
     "windows": {
       "webviewInstallMode": {
         "type": "embedBootstrapper"
+      },
+      "nsis": {
+        "installMode": "perMachine",
+        "headerImage": null,
+        "sidebarImage": null,
+        "installerIcon": "icons/icon.ico",
+        "languages": [
+          "English",
+          "SimpChinese"
+        ]
       }
     }
   }

src/App.vue

@@ -1,20 +1,42 @@
 <!-- App.vue -->
 <template>
   <main class="container">
-    <InitializationScreen v-if="isInitializing" @initialization-complete="onInitializationComplete" />
+    <InitializationScreen v-if="needsInitialization" @initialization-complete="onInitializationComplete" />
     <MainLayout v-else />
   </main>
 </template>
 
 <script setup lang="ts">
-import { ref } from 'vue';
+import { ref, onMounted } from 'vue';
+import { invoke } from '@tauri-apps/api/core';
 import MainLayout from '@/components/MainLayout.vue';
 import InitializationScreen from '@/components/InitializationScreen.vue';
+import { FACE_RESTORATION_MODEL, IMAGE_UPSCALING_MODEL, BACKGROUND_REMOVAL_MODEL } from './config/models';
 
-const isInitializing = ref(true);
+const needsInitialization = ref(true);
+const MODELS = [BACKGROUND_REMOVAL_MODEL, FACE_RESTORATION_MODEL, IMAGE_UPSCALING_MODEL];
+
+onMounted(async () => {
+  try {
+    let requiresDownload = false;
+    for (const model of MODELS) {
+      const needsDownload = await invoke('check_model_exists', { modelName: model });
+      if (needsDownload) {
+        requiresDownload = true;
+        break;
+      }
+    }
+
+    if (!requiresDownload) {
+      needsInitialization.value = false;
+    }
+  } catch (error) {
+    console.error('Error checking models:', error);
+  }
+});
 
 function onInitializationComplete() {
-  isInitializing.value = false;
+  needsInitialization.value = false;
 }
 </script>