Upload pipeline_input_processor.py

core/pipelines/pipeline_input_processor.py

@@ -5,7 +5,7 @@ import gradio as gr
 from PIL import Image, ImageChops
 from typing import Dict, Any, List
 
-from core.settings import INPUT_DIR, MULTIPLIERS_MAP
+from core.settings import INPUT_DIR, MULTIPLIERS_MAP, LORA_DIR, EMBEDDING_DIR, VAE_DIR
 from utils.app_utils import (
     sanitize_filename,
     get_lora_path,
@@ -57,6 +57,9 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
                 lora_filename = None
                 if source == "File":
                     lora_filename = sanitize_filename(lora_id)
+                    local_path = os.path.join(LORA_DIR, lora_filename)
+                    if not os.path.exists(local_path):
+                        raise gr.Error(f"Uploaded LoRA file '{lora_id}' no longer exists on server. Please re-upload it.")
                 elif source in ("Civitai", "Hugging Face"):
                     local_path, status = get_lora_path(source, lora_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
                     if local_path: lora_filename = os.path.basename(local_path)
@@ -137,6 +140,9 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
                 emb_filename = None
                 if source == "File":
                     emb_filename = sanitize_filename(emb_id)
+                    local_path = os.path.join(EMBEDDING_DIR, emb_filename)
+                    if not os.path.exists(local_path):
+                        raise gr.Error(f"Uploaded Embedding file '{emb_id}' no longer exists on server. Please re-upload it.")
                 elif source in ("Civitai", "Hugging Face"):
                     local_path, status = get_embedding_path(source, emb_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
                     if local_path: emb_filename = os.path.basename(local_path)
@@ -368,6 +374,9 @@ def process_pipeline_inputs(ui_inputs: Dict[str, Any], progress: gr.Progress, wo
     if vae_source and vae_source != "None":
         if vae_source == "File":
             vae_name_override = sanitize_filename(vae_id)
+            local_path = os.path.join(VAE_DIR, vae_name_override)
+            if not os.path.exists(local_path):
+                raise gr.Error(f"Uploaded VAE file '{vae_id}' no longer exists on server. Please re-upload it.")
         elif vae_source in ("Civitai", "Hugging Face") and vae_id and vae_id.strip():
             local_path, status = get_vae_path(vae_source, vae_id, os.environ.get("CIVITAI_API_KEY", ""), progress)
             if local_path: vae_name_override = os.path.basename(local_path)