From d28f00c845d4161fddef60ccc5899b95ed4d25c9 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Thu, 7 May 2026 15:14:03 +0200
Subject: [PATCH] Improve model path resolution and enforce offline environment
 variables for diffusion generation

---
 src-tauri/src/main.rs | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/src-tauri/src/main.rs b/src-tauri/src/main.rs
index 2b908bf..4e379d9 100644
--- a/src-tauri/src/main.rs
+++ b/src-tauri/src/main.rs
@@ -361,9 +361,16 @@ async fn generate_map(
     let scheduler = cfg.scheduler.unwrap_or_else(|| "dpmsolver-sde".to_string());
     let upscale = cfg.upscale.unwrap_or_else(|| "none".to_string());
     let seam_inpaint = cfg.seam_inpaint.unwrap_or(false);
+    let local_model_path = root.join("models").join("sdxl360Diffusion_v10.safetensors");
     let model_path = cfg
         .model_path
-        .unwrap_or_else(|| "proximasan/sdxl-360-diffusion".to_string());
+        .unwrap_or_else(|| {
+            if local_model_path.exists() {
+                local_model_path.to_string_lossy().to_string()
+            } else {
+                "ProGamerGov/sdxl-360-diffusion".to_string()
+            }
+        });
     let base_model = cfg
         .base_model
         .unwrap_or_else(|| "stabilityai/stable-diffusion-xl-base-1.0".to_string());
@@ -444,7 +451,10 @@ async fn generate_map(
                 &mut log_file,
                 format!("  program: {}", python.to_string_lossy()),
             );
-            write_log_line(&mut log_file, "  env: PYTHONUNBUFFERED=1");
+            write_log_line(
+                &mut log_file,
+                "  env: PYTHONUNBUFFERED=1 HF_HUB_OFFLINE=1 TRANSFORMERS_OFFLINE=1 DIFFUSERS_OFFLINE=1",
+            );
             for arg in &args {
                 write_log_line(&mut log_file, format!("  arg: {arg}"));
             }
@@ -453,6 +463,9 @@ async fn generate_map(
             let mut cmd = Command::new(&python);
             cmd.current_dir(&root)
                 .env("PYTHONUNBUFFERED", "1")
+                .env("HF_HUB_OFFLINE", "1")
+                .env("TRANSFORMERS_OFFLINE", "1")
+                .env("DIFFUSERS_OFFLINE", "1")
                 .stdout(Stdio::piped())
                 .stderr(Stdio::piped());
             for arg in &args {