fix(presets): support multi-preset toggle and add active border indicator

.github/workflows/build.yml

@@ -35,9 +35,14 @@ on:
         required: false
         type: string
         default: ''
+      android-signing:
+        required: false
+        type: string
+        default: ''
 
 jobs:
   build:
+    if: ${{ inputs.mobile != 'android' || inputs.android-signing != 'false' }}
     runs-on: ${{ inputs.platform }}
     steps:
       - name: Checkout repository

.github/workflows/ci.yml

@@ -6,7 +6,24 @@ on:
       - main
 
 jobs:
+  # Forks do not have the Android signing secrets, so the Android build is skipped there.
+  android-signing:
+    runs-on: ubuntu-latest
+    outputs:
+      available: ${{ steps.check.outputs.available }}
+    steps:
+      - id: check
+        env:
+          ANDROID_KEY_BASE64: ${{ secrets.ANDROID_KEY_BASE64 }}
+        run: |
+          if [ -n "$ANDROID_KEY_BASE64" ]; then
+            echo "available=true" >> "$GITHUB_OUTPUT"
+          else
+            echo "available=false" >> "$GITHUB_OUTPUT"
+          fi
+
   build:
+    needs: android-signing
     strategy:
       fail-fast: false
       matrix:
@@ -65,6 +82,7 @@ jobs:
       build-args: ${{ matrix.args }}
       asset-prefix: ${{ matrix.asset-prefix }}
       mobile: ${{ matrix.mobile }}
+      android-signing: ${{ needs.android-signing.outputs.available }}
     secrets: inherit
     permissions:
       contents: write

.github/workflows/release.yml

@@ -5,7 +5,24 @@ on:
     types: [created]
 
 jobs:
+  # Forks do not have the Android signing secrets, so the Android build is skipped there.
+  android-signing:
+    runs-on: ubuntu-latest
+    outputs:
+      available: ${{ steps.check.outputs.available }}
+    steps:
+      - id: check
+        env:
+          ANDROID_KEY_BASE64: ${{ secrets.ANDROID_KEY_BASE64 }}
+        run: |
+          if [ -n "$ANDROID_KEY_BASE64" ]; then
+            echo "available=true" >> "$GITHUB_OUTPUT"
+          else
+            echo "available=false" >> "$GITHUB_OUTPUT"
+          fi
+
   release:
+    needs: android-signing
     strategy:
       fail-fast: false
       matrix:
@@ -66,6 +83,7 @@ jobs:
       asset-name-pattern: '[name]_v[version]_[platform]_[arch][ext]'
       asset-prefix: ${{ matrix.asset-prefix }}
       mobile: ${{ matrix.mobile }}
+      android-signing: ${{ needs.android-signing.outputs.available }}
     secrets: inherit
     permissions:
       contents: write

src-tauri/src/ai_denoise.rs

@@ -307,6 +307,8 @@ pub fn apply_ai_denoise<'a>(
         (adjustments["aiDenoiseAmount"].as_f64().unwrap_or(100.0) as f32 / 100.0).clamp(0.0, 1.0);
     let detail =
         (adjustments["aiDenoiseDetail"].as_f64().unwrap_or(0.0) as f32 / 100.0).clamp(0.0, 1.0);
+    let color =
+        (adjustments["aiDenoiseColor"].as_f64().unwrap_or(100.0) as f32 / 100.0).clamp(0.0, 1.0);
     if amount <= 0.0 {
         return Cow::Borrowed(base);
     }
@@ -336,7 +338,7 @@ pub fn apply_ai_denoise<'a>(
     let denoised: &[f16] = resized.as_deref().unwrap_or(&entry.data);
 
     let luma_keep = 1.0 - amount * (1.0 - detail);
-    let chroma_keep = 1.0 - amount;
+    let chroma_keep = 1.0 - amount * color;
 
     let mut out = base.to_rgb32f();
     out.par_chunks_mut(BLEND_CHUNK)

src-tauri/src/ai_processing.rs

@@ -26,6 +26,9 @@ use tauri::Manager;
 use tokenizers::Tokenizer;
 use tokio::sync::Mutex as TokioMutex;
 
+use crate::face_parsing::{FaceModels, load_face_parser_session};
+use crate::face_refine::FaceCropCache;
+
 use crate::face_masking::FaceAnalysis;
 
 const ENCODER_URL: &str = "https://huggingface.co/CyberTimon/RapidRAW-Models/resolve/main/sam_vit_b_01ec64_encoder.onnx?download=true";
@@ -124,6 +127,16 @@ const NORMAL_MAX_SIZE: u32 = 768;
 const NORMAL_NUM_TOKENS: i64 = 2500;
 const NORMAL_SHA256: &str = "24eacb5dc7a2c54c7bc98f7de085ffbed79ad006ea5b664c2c2cdc02ff3a52f0";
 
+const FACE_DETECTOR_URL: &str = "https://huggingface.co/opencv/face_detection_yunet/resolve/main/face_detection_yunet_2023mar.onnx?download=true";
+const FACE_DETECTOR_FILENAME: &str = "face_detection_yunet_2023mar.onnx";
+const FACE_DETECTOR_SHA256: &str =
+    "8f2383e4dd3cfbb4553ea8718107fc0423210dc964f9f4280604804ed2552fa4";
+
+const FACE_PARSER_URL: &str =
+    "https://github.com/yakhyo/face-parsing/releases/download/v0.0.1/resnet18.onnx";
+const FACE_PARSER_FILENAME: &str = "face_parsing_resnet18.onnx";
+const FACE_PARSER_SHA256: &str = "0d9bd318e46987c3bdbfacae9e2c0f461cae1c6ac6ea6d43bbe541a91727e33f";
+
 type SharedSession = Arc<Mutex<Session>>;
 
 const GPU_IDLE: u8 = 0;
@@ -140,6 +153,20 @@ struct ModelAsset {
     name: &'static str,
 }
 
+const FACE_DETECTOR_ASSET: ModelAsset = ModelAsset {
+    filename: FACE_DETECTOR_FILENAME,
+    url: FACE_DETECTOR_URL,
+    sha256: Some(FACE_DETECTOR_SHA256),
+    name: "Face Detection Model",
+};
+
+const FACE_PARSER_ASSET: ModelAsset = ModelAsset {
+    filename: FACE_PARSER_FILENAME,
+    url: FACE_PARSER_URL,
+    sha256: Some(FACE_PARSER_SHA256),
+    name: "Face Regions Model",
+};
+
 const SAM_ENCODER_ASSET: ModelAsset = ModelAsset {
     filename: ENCODER_FILENAME,
     url: ENCODER_URL,
@@ -691,6 +718,8 @@ pub struct AiState {
     pub esam3_features: Option<Arc<Esam3Features>>,
     pub face_analysis: Option<Arc<FaceAnalysis>>,
     pub depth_map: Option<CachedDepthMap>,
+    pub face_models: Option<Arc<FaceModels>>,
+    pub face_crops: Option<Arc<FaceCropCache>>,
 }
 
 impl AiState {
@@ -1356,6 +1385,51 @@ pub async fn get_or_init_lama_model(
     Ok(lama_model)
 }
 
+pub async fn get_or_init_face_models(
+    app_handle: &tauri::AppHandle,
+    ai_state_mutex: &Mutex<Option<AiState>>,
+    ai_init_lock: &TokioMutex<()>,
+) -> Result<Arc<FaceModels>> {
+    let loaded = || {
+        ai_state_mutex
+            .lock()
+            .unwrap()
+            .as_ref()
+            .and_then(|state| state.face_models.clone())
+    };
+    if let Some(face_models) = loaded() {
+        return Ok(face_models);
+    }
+
+    let _guard = ai_init_lock.lock().await;
+
+    if let Some(face_models) = loaded() {
+        return Ok(face_models);
+    }
+
+    let models_dir = get_models_dir(app_handle)?;
+    download_and_verify_model(app_handle, &models_dir, &FACE_DETECTOR_ASSET).await?;
+    download_and_verify_model(app_handle, &models_dir, &FACE_PARSER_ASSET).await?;
+
+    let _ = ort::init().with_name("AI-Face").commit();
+    let detector = Session::builder()?.commit_from_file(models_dir.join(FACE_DETECTOR_FILENAME))?;
+    let parser = load_face_parser_session(&models_dir.join(FACE_PARSER_FILENAME))?;
+    let face_models = Arc::new(FaceModels {
+        detector: Mutex::new(detector),
+        parser: Mutex::new(parser),
+    });
+
+    crate::register_exit_handler();
+
+    ai_state_mutex
+        .lock()
+        .unwrap()
+        .get_or_insert_with(AiState::default)
+        .face_models = Some(face_models.clone());
+
+    Ok(face_models)
+}
+
 pub async fn get_or_init_normal_model(
     app_handle: &tauri::AppHandle,
     ai_state_mutex: &Mutex<Option<AiState>>,

src-tauri/src/app_settings.rs

@@ -153,6 +153,7 @@ pub fn all_available_adjustments() -> HashSet<String> {
         "aiDenoiseEnabled",
         "aiDenoiseMethod",
         "aiDenoiseAmount",
+        "aiDenoiseColor",
         "aiDenoiseDetail",
         "chromaticAberrationRedCyan",
         "chromaticAberrationBlueYellow",

src-tauri/src/cache_utils.rs

@@ -94,7 +94,12 @@ fn hash_ai_denoise(adjustments: &serde_json::Value, hasher: &mut DefaultHasher)
     if enabled {
         hash_effect_keys(
             adjustments,
-            &["aiDenoiseMethod", "aiDenoiseAmount", "aiDenoiseDetail"],
+            &[
+                "aiDenoiseMethod",
+                "aiDenoiseAmount",
+                "aiDenoiseColor",
+                "aiDenoiseDetail",
+            ],
             hasher,
         );
         crate::ai_denoise::cache_epoch().hash(hasher);