Update app.py

using multiple prompts

app.py

@@ -46,38 +46,47 @@ def generate_canny_map(image: Image.Image) -> Image.Image:
 # ----------------------------
 # 🎨 Image Generation Function
 # ----------------------------
+
+
 def process_image(prompt, image, num_variations):
     try:
         print("🧠 Prompt received:", prompt)
 
         # Step 1: Brain Layer
-        reasoning_json = extract_scene_plan(prompt)
-        print("🧠 Scene plan extracted:", reasoning_json)
+        scene_plan = extract_scene_plan(prompt)
+        print("🧠 Scene plan extracted:", scene_plan)
 
-        # Step 2: Prepare inputs
+        # Step 2: Generate Prompt Variations
+        prompt_list = generate_prompt_variations(prompt, scene_plan, num_variations)
+        print("🧠 Enriched Prompts:")
+        for i, p in enumerate(prompt_list):
+            print(f"  {i+1}: {p}")
+
+        # Step 3: Prepare image + canny
         image = image.resize((512, 512)).convert("RGB")
         canny_map = generate_canny_map(image)
 
+        # Step 4: Generate Images
         outputs = []
-        for i in range(num_variations):
-            print(f"🎨 Generating variation {i+1}")
+        for i, enriched_prompt in enumerate(prompt_list):
+            print(f"🎨 Generating image {i+1} with enriched prompt")
             result = pipe(
-                prompt=prompt,
+                prompt=enriched_prompt,
                 image=image,
                 control_image=canny_map,
                 num_inference_steps=40,
                 strength=0.9,
                 guidance_scale=7.5
             )
-            generated_image = result.images[0]
-            outputs.append(generated_image)
+            outputs.append(result.images[0])
 
-        return outputs, reasoning_json, canny_map
+        return outputs, scene_plan, canny_map
 
     except Exception as e:
         print("❌ Generation failed:", e)
         return ["❌ Error during generation"], {"error": str(e)}, None
 
+
 # ----------------------------
 # 🖼 Gradio UI
 # ----------------------------