Update app.py

app.py

@@ -349,59 +349,13 @@ def generate_ai_background(
     seed: Optional[int] = None,
 ) -> str:
     """Generate AI background using Stable Diffusion."""
-    if not TORCH_AVAILABLE:
-        raise RuntimeError("PyTorch required for AI background generation")
-    
-    try:
-        from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
-    except ImportError:
-        raise RuntimeError("Please install diffusers: pip install diffusers transformers")
-    
-    # Setup generator
-    generator = torch.Generator(device=DEVICE)
-    if seed is None:
-        seed = random.randint(0, 2**31 - 1)
-    generator.manual_seed(seed)
-    
-    # Generate background
-    if init_image_path and os.path.exists(init_image_path):
-        # Image-to-image
-        pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
-            "runwayml/stable-diffusion-v1-5",
-            torch_dtype=torch.float16 if CUDA_AVAILABLE else torch.float32,
-            safety_checker=None
-        ).to(DEVICE)
-        
-        init_image = Image.open(init_image_path).convert("RGB").resize((width, height))
-        result = pipe(
-            prompt=prompt,
-            image=init_image,
-            strength=0.6,
-            num_inference_steps=num_steps,
-            guidance_scale=guidance_scale,
-            generator=generator
-        ).images[0]
-    else:
-        # Text-to-image
-        pipe = StableDiffusionPipeline.from_pretrained(
-            "runwayml/stable-diffusion-v1-5",
-            torch_dtype=torch.float16 if CUDA_AVAILABLE else torch.float32,
-            safety_checker=None
-        ).to(DEVICE)
-        
-        result = pipe(
-            prompt=prompt,
-            height=height,
-            width=width,
-            num_inference_steps=num_steps,
-            guidance_scale=guidance_scale,
-            generator=generator
-        ).images[0]
-    
-    # Save result
-    output_path = TEMP_DIR / f"ai_bg_{int(time.time())}_{uuid4().hex[:6]}.jpg"
-    result.save(output_path, quality=95)
-    return str(output_path)
+    # TEMPORARILY DISABLED due to PyTorch/Diffusers compatibility issue
+    # To fix: pip install --upgrade torch diffusers transformers
+    raise RuntimeError(
+        "AI Background temporarily disabled due to PyTorch/Diffusers version compatibility.\n"
+        "To fix: pip install --upgrade torch diffusers transformers accelerate\n"
+        "For now, please use Upload Image or Gradients instead."
+    )
 
 # ==============================================================================
 # MAIN PROCESSING PIPELINE