Update utils/planner.py

exttracting multiple prompts

utils/planner.py

@@ -44,3 +44,19 @@ def extract_scene_plan(prompt: str) -> dict:
             "layout": {},
             "rules": {}
         }
+
+
+# ----------------------------
+# 🧠 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 {layout.get('foreground', '')} in foreground and {layout.get('background', '')} in background, style variation {i+1}"
+        variations.append(enriched.strip())
+
+    return variations