cascade

app.py

@@ -18,11 +18,9 @@ secret_model = os.environ.get("MODEL_PATH")
 BASE_MODEL = os.environ.get("BASE_MODEL_ID")
 
 try:
-    from diffusers import FluxTransformer2DModel
-    from diffusers.pipelines import FluxPipeline
-    from flux.condition import Condition
-    from flux.generate import generate
-    from flux.lora_controller import set_lora_scale
+    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}")
@@ -44,10 +42,6 @@ temp_dir = os.path.join(os.path.expanduser("~"), "gradio_temp")
 os.makedirs(temp_dir, exist_ok=True)
 os.environ["GRADIO_TEMP_DIR"] = temp_dir
 
-# Global state
-pipe: Union[FluxPipeline, None] = None
-use_int8 = False
-
 ADAPTER_NAME = "subject"
 MODEL_PATH = model_path
 
@@ -55,9 +49,6 @@ def get_gpu_memory_gb() -> float:
     return torch.cuda.get_device_properties(0).total_memory / 1024**3
 
 def init_pipeline_if_needed():
-    global pipe
-    if pipe is not None:
-        return
     
     print("🚀 Initializing pipeline...")
     
@@ -246,7 +237,7 @@ def _place_subject_on_canvas_rect(
     canvas.alpha_composite(rotated, dest=(paste_x, paste_y))
     return canvas
 
-def apply_style(image: Image.Image, style: str, width: int = 800, height: int = 800) -> Image.Image:
+def apply_style(image: Image.Image, style: str, width: int = 1024, height: int = 1024) -> Image.Image:
     """Apply specified style to image with custom dimensions"""
     if image is None:
         # Create default transparent image
@@ -319,23 +310,23 @@ def create_simple_app():
     example_prompts = [
         {
             "title": "Handcrafted Leather Wallet",
-            "prompt": "A hand-stitched, dark brown leather wallet lies half-open on a wooden desk with a map, next to a brass pen and compass. A stack of classic books is in the background. A warm desk lamp from the right highlights the leather texture. Classic, rustic style."
+            "prompt": "A premium lifestyle advertisement for a hand-stitched dark brown leather wallet. The wallet is half-open on a timeworn walnut desk, revealing the suede interior and a few vintage travel tickets. Surround it with a rolled map, brass fountain pen, and antique compass to emphasize heritage craftsmanship. Soft amber light from a desk lamp on the right grazes the grainy leather and creates gentle shadow falloff, while a blurred wall of old books fills the background. Overall tone is classic, rustic, and aspirational."
         },
         {
             "title": "Sparkling Water with Fresh Lemons", 
-            "prompt": "A dewy glass bottle of sparkling water on a white marble countertop, next to a sliced lemon and ice cubes. The background is a soft-focus, pale blue gradient. Lighting is bright, even, and cool-toned from above. Clean, crisp, minimalist style."
+            "prompt": "A product hero shot for a premium sparkling water infused with fresh lemons. Place a dewy glass bottle at the center of a white marble countertop, with a tall tumbler filled with effervescent water, thin lemon wheels, and crystal-clear ice cubes beside it. Scatter a few lemon zest curls and condensation droplets for sensory detail. Use a soft-focus pale blue and white gradient background to communicate freshness, and bathe the scene in bright, cool, top-down lighting that creates sharp reflections. Keep the styling ultra-clean, crisp, and minimalist."
         },
         {
             "title": "High-tech Smartwatch",
-            "prompt": "A titanium smartwatch with an illuminated screen rests on a black matte slate rock. The background is a blurred cityscape at night with neon bokeh. A sharp, direct light from the top-left highlights the watch's metallic edge. Futuristic, tech-focused style."
+            "prompt": "A cinematic tech advertisement for a titanium smartwatch with an always-on illuminated screen displaying futuristic UI graphics. Position the watch on a jagged slab of matte black slate to contrast its polished chamfered edges. Behind it, place a blurred nighttime cityscape with teal and magenta neon bokeh to suggest urban energy. Hit the product with a sharp, directional spotlight from the top left to carve out highlights along the bezel and bracelet, while subtle rim lighting separates it from the background. Mood is sleek, futuristic, and performance-driven."
         },
         {
             "title": "Japanese Ramen Bowl",
-            "prompt": "A ceramic bowl of tonkotsu ramen with chashu pork and a soft-boiled egg on a wooden table, with chopsticks beside it. Rising steam is caught in soft overhead light. The background is a blurred, cozy izakaya. Warm, authentic, and appetizing style."
+            "prompt": "A mouthwatering food advertisement for a ceramic bowl of tonkotsu ramen. Present silky broth with two slices of torched chashu pork, a jammy soft-boiled egg, nori sheets, scallions, and sesame seeds arranged artfully. Place the bowl on a rustic wooden table with lacquered chopsticks resting on a ceramic holder, plus a tiny dish of pickled ginger for color. Capture wisps of steam drifting upward in soft overhead light, while the background falls into a blurred, amber-toned izakaya interior with paper lanterns. Atmosphere is warm, authentic, and comforting."
         },
         {
             "title": "Japanese Peach Iced Tea",
-            "prompt": "A bottle of Japanese peach iced tea beside a tall glass with tea and sparkling ice cubes. The background is a soft, warm peach and beige gradient. Lit with bright, soft light to appear crisp and refreshing. The style is clean, minimalist, and refined."
+            "prompt": "A commercial advertisement for a Japanese peach-flavored iced tea. The composition features the product bottle placed next to a tall, elegant glass filled with the tea and sparkling ice cubes. The background is a soft, warm gradient of peach and beige, creating a gentle and sophisticated atmosphere. The overall style is clean, minimalist, and refined, with bright, soft lighting that highlights the crisp, refreshing quality of the beverage."
         }
     ]