app

app.py

@@ -17,18 +17,16 @@ secret_model = os.environ.get("MODEL_PATH")
 # 从环境变量获取基础模型路径
 BASE_MODEL = os.environ.get("BASE_MODEL_ID")
 
-try:
-    from cascade.condition import Condition
-    from cascade.generate import generate
-    from cascade.lora_controller import set_lora_scale
-    FLUX_AVAILABLE = True
-except ImportError as e:
-    print(f"Warning: FLUX components not available: {e}")
-    FLUX_AVAILABLE = False
+from cascade.condition import Condition
+from cascade.generate import generate
+from cascade.lora_controller import set_lora_scale
 
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 
+# Global pipeline variable
+_global_pipe = None
+
 # 認証トークンを使ってファイルをダウンロード
 model_path = hf_hub_download(
     repo_id="Cascade-Inc/private_model",
@@ -49,6 +47,9 @@ def get_gpu_memory_gb() -> float:
     return torch.cuda.get_device_properties(0).total_memory / 1024**3
 
 def init_pipeline_if_needed():
+    global _global_pipe
+    if _global_pipe is not None:
+        return _global_pipe
     
     print("🚀 Initializing pipeline...")
     
@@ -92,8 +93,9 @@ def init_pipeline_if_needed():
     print("🎨 Loading Cascade weights...")
     _pipe.load_lora_weights(MODEL_PATH, adapter_name=ADAPTER_NAME)
     _pipe.set_adapters([ADAPTER_NAME])
-    pipe = _pipe
+    _global_pipe = _pipe
     print("✅ Pipeline initialized successfully!")
+    return _global_pipe
 
 def _to_pil_rgba(img: Any) -> Image.Image:
     """Convert various inputs to PIL RGBA image"""
@@ -252,24 +254,11 @@ def apply_style(image: Image.Image, style: str, width: int = 1024, height: int =
     return styled_image
 
 def generate_background_local(styled_image: Image.Image, prompt: str, steps: int = 10, width: int = 1024, height: int = 1024) -> Image.Image:
-    """Generate background using local FLUX model"""
+    """Generate background using local model"""
     width = int(width)
     height = int(height)
-
-    if not FLUX_AVAILABLE:
-        # Return a simple gradient background if FLUX is not available
-        if styled_image is None:
-            return Image.new("RGB", (width, height), (200, 200, 255))
-        styled_image = _center_subject_on_canvas(styled_image, width, height)
-        # Create a simple colored background
-        bg = Image.new("RGB", (width, height), (200, 220, 255))
-        if styled_image.mode == "RGBA":
-            bg.paste(styled_image, (0, 0), styled_image)
-        else:
-            bg.paste(styled_image, (0, 0))
-        return bg
     
-    init_pipeline_if_needed()
+    pipe = init_pipeline_if_needed()
     
     if styled_image is None:
         return Image.new("RGB", (width, height), (255, 255, 255))
@@ -291,7 +280,7 @@ def generate_background_local(styled_image: Image.Image, prompt: str, steps: int
     
     with set_lora_scale([ADAPTER_NAME], scale=3.0):
         result_img = generate(
-            #pipe,
+            pipe=pipe,
             model_config=model_config,
             prompt=prompt.strip() if prompt else "",
             conditions=[condition],