Update app.py

variations of prompt

app.py

@@ -7,7 +7,7 @@ import torch
 import numpy as np
 import cv2
 from diffusers import StableDiffusionControlNetImg2ImgPipeline, ControlNetModel
-from utils.planner import extract_scene_plan  # 🧠 Brain Layer
+from utils.planner import extract_scene_plan, generate_prompt_variations_from_scene  # 🧠 Brain Layer
 
 # ----------------------------
 # 🔧 Device Setup
@@ -44,35 +44,6 @@ def generate_canny_map(image: Image.Image) -> Image.Image:
     return canny_image
 
 
-# ----------------------------
-# 🧠 Generate Prompt Variations from Scene Plan
-# ----------------------------
-def generate_prompt_variations(base_prompt: str, scene_plan: dict, num_variations: int) -> list:
-    variations = []
-    scene = scene_plan.get("scene", {})
-    subject = scene_plan.get("subject", {})
-    layout = scene_plan.get("layout", {})
-
-    for i in range(num_variations):
-        enriched = f"{base_prompt}, in a {scene.get('environment', '')} setting with {scene.get('setting', '')}, featuring {subject.get('main_actor', '')}, layout includes"
-
-        # Foreground
-        fg = layout.get("foreground", {}).get("elements", [])
-        fg_text = ", ".join([f"{el['type']} at {el['position']}" for el in fg if 'type' in el and 'position' in el])
-        enriched += f" {fg_text} in foreground,"
-
-        # Background
-        bg = layout.get("background", {}).get("elements", [])
-        bg_text = ", ".join([f"{el['type']} at {el['position']}" for el in bg if 'type' in el and 'position' in el])
-        enriched += f" {bg_text} in background"
-
-        enriched += f", style variation {i+1}"
-        variations.append(enriched.strip())
-
-    return variations
-
-
-
 # ----------------------------
 # 🎨 Image Generation Function
 # ----------------------------
@@ -87,7 +58,7 @@ def process_image(prompt, image, num_variations):
         print("🧠 Scene plan extracted:", scene_plan)
 
         # Step 2: Generate Prompt Variations
-        prompt_list = generate_prompt_variations(prompt, scene_plan, num_variations)
+        prompt_list = generate_prompt_variations_from_scene(scene_plan, prompt, num_variations)
         print("🧠 Enriched Prompts:")
         for i, p in enumerate(prompt_list):
             print(f"  {i+1}: {p}")