Update app.py

adding prompt varistions

app.py

@@ -43,6 +43,36 @@ def generate_canny_map(image: Image.Image) -> Image.Image:
     canny_image = Image.fromarray(edges).convert("RGB")
     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
 # ----------------------------