variation in gallery

services/generator.py

@@ -1536,7 +1536,7 @@ CRITICAL REQUIREMENTS:
             "visual_styles": visual_styles,
         }
         
-        # Save to database (for first image)
+        # Save to database (save EACH image as a separate record for gallery visibility)
         # Ensure database connection is initialized
         if db_service and db_service.collection is None and settings.mongodb_url:
             try:
@@ -1544,42 +1544,50 @@ CRITICAL REQUIREMENTS:
             except Exception as e:
                 print(f"Warning: Could not connect to database: {e}")
         
-        first_image = generated_images[0] if generated_images else {}
-        if db_service and db_service.collection is not None and not first_image.get("error") and username:
-            try:
-                db_id = await db_service.save_ad_creative(
-                    niche=niche,
-                    title=None,  # No title for standard flow
-                    headline=ad_copy.get("headline", ""),
-                    primary_text=ad_copy.get("primary_text", ""),
-                    description=ad_copy.get("description", ""),
-                    body_story=ad_copy.get("body_story", ""),
-                    cta=ad_copy.get("cta", ""),
-                    psychological_angle=ad_copy.get("psychological_angle", ""),
-                    why_it_works=ad_copy.get("why_it_works", ""),
-                    username=username,  # Pass username
-                    image_url=first_image.get("image_url"),
-                    image_filename=first_image.get("filename"),
-                    image_model=first_image.get("model_used"),
-                    image_seed=first_image.get("seed"),
-                    image_prompt=first_image.get("image_prompt"),  # Save the final refined prompt
-                    generation_method="standard",
-                    angle_key=angle.get("key"),
-                    angle_name=angle.get("name"),
-                    angle_trigger=angle.get("trigger"),
-                    angle_category=angle.get("category"),
-                    concept_key=concept.get("key"),
-                    concept_name=concept.get("name"),
-                    concept_structure=concept.get("structure"),
-                    concept_visual=concept.get("visual"),
-                    concept_category=concept.get("category"),
-                    metadata=metadata,
-                )
-                if db_id:
-                    ad_id = db_id
-                    print(f"✓ Saved ad creative to database: {ad_id}")
-            except Exception as e:
-                print(f"Warning: Failed to save to database: {e}")
+        # Save each image variation as a separate database record
+        saved_ad_ids = []
+        if db_service and db_service.collection is not None and username:
+            for img_idx, image in enumerate(generated_images):
+                if image.get("error"):
+                    continue  # Skip failed images
+                try:
+                    db_id = await db_service.save_ad_creative(
+                        niche=niche,
+                        title=None,  # No title for standard flow
+                        headline=ad_copy.get("headline", ""),
+                        primary_text=ad_copy.get("primary_text", ""),
+                        description=ad_copy.get("description", ""),
+                        body_story=ad_copy.get("body_story", ""),
+                        cta=ad_copy.get("cta", ""),
+                        psychological_angle=ad_copy.get("psychological_angle", ""),
+                        why_it_works=ad_copy.get("why_it_works", ""),
+                        username=username,  # Pass username
+                        image_url=image.get("image_url"),
+                        image_filename=image.get("filename"),
+                        image_model=image.get("model_used"),
+                        image_seed=image.get("seed"),
+                        image_prompt=image.get("image_prompt"),  # Save the final refined prompt
+                        generation_method="standard",
+                        angle_key=angle.get("key"),
+                        angle_name=angle.get("name"),
+                        angle_trigger=angle.get("trigger"),
+                        angle_category=angle.get("category"),
+                        concept_key=concept.get("key"),
+                        concept_name=concept.get("name"),
+                        concept_structure=concept.get("structure"),
+                        concept_visual=concept.get("visual"),
+                        concept_category=concept.get("category"),
+                        metadata={**metadata, "variation_index": img_idx, "total_variations": len(generated_images)},
+                    )
+                    if db_id:
+                        saved_ad_ids.append(db_id)
+                        print(f"✓ Saved ad creative variation {img_idx + 1}/{len(generated_images)} to database: {db_id}")
+                except Exception as e:
+                    print(f"Warning: Failed to save variation {img_idx + 1} to database: {e}")
+        
+        # Use first saved ID as the main ad_id
+        if saved_ad_ids:
+            ad_id = saved_ad_ids[0]
         
         # Build response
         result = {
@@ -1791,7 +1799,7 @@ CONCEPT: {concept['name']}
         # Generate unique ID
         ad_id = str(uuid.uuid4())
         
-        # Save to database (for first image)
+        # Save to database (save EACH image as a separate record for gallery visibility)
         # Ensure database connection is initialized
         if db_service and db_service.collection is None and settings.mongodb_url:
             try:
@@ -1799,42 +1807,50 @@ CONCEPT: {concept['name']}
             except Exception as e:
                 print(f"Warning: Could not connect to database: {e}")
         
-        first_image = images[0] if images else {}
-        if db_service and db_service.collection is not None and not first_image.get("error") and username:
-            try:
-                db_id = await db_service.save_ad_creative(
-                    niche=niche,
-                    title=None,  # No title for matrix flow
-                    headline=ad_copy.get("headline", ""),
-                    primary_text=ad_copy.get("primary_text", ""),
-                    description=ad_copy.get("description", ""),
-                    body_story=ad_copy.get("body_story", ""),
-                    cta=ad_copy.get("cta", ""),
-                    psychological_angle=ad_copy.get("psychological_angle", angle.get("name", "")),
-                    why_it_works=ad_copy.get("why_it_works", ""),
-                    username=username,  # Pass username
-                    image_url=first_image.get("image_url"),
-                    image_filename=first_image.get("filename"),
-                    image_model=first_image.get("model_used"),
-                    image_seed=first_image.get("seed"),
-                    image_prompt=first_image.get("image_prompt"),  # Save the final refined prompt
-                    angle_key=angle.get("key"),
-                    angle_name=angle.get("name"),
-                    angle_trigger=angle.get("trigger"),
-                    angle_category=angle.get("category"),
-                    concept_key=concept.get("key"),
-                    concept_name=concept.get("name"),
-                    concept_structure=concept.get("structure"),
-                    concept_visual=concept.get("visual"),
-                    concept_category=concept.get("category"),
-                    generation_method="angle_concept_matrix",
-                    metadata={"generation_method": "angle_concept_matrix"},
-                )
-                if db_id:
-                    ad_id = db_id
-                    print(f"✓ Saved matrix ad creative to database: {ad_id}")
-            except Exception as e:
-                print(f"Warning: Failed to save to database: {e}")
+        # Save each image variation as a separate database record
+        saved_ad_ids = []
+        if db_service and db_service.collection is not None and username:
+            for img_idx, image in enumerate(images):
+                if image.get("error"):
+                    continue  # Skip failed images
+                try:
+                    db_id = await db_service.save_ad_creative(
+                        niche=niche,
+                        title=None,  # No title for matrix flow
+                        headline=ad_copy.get("headline", ""),
+                        primary_text=ad_copy.get("primary_text", ""),
+                        description=ad_copy.get("description", ""),
+                        body_story=ad_copy.get("body_story", ""),
+                        cta=ad_copy.get("cta", ""),
+                        psychological_angle=ad_copy.get("psychological_angle", angle.get("name", "")),
+                        why_it_works=ad_copy.get("why_it_works", ""),
+                        username=username,  # Pass username
+                        image_url=image.get("image_url"),
+                        image_filename=image.get("filename"),
+                        image_model=image.get("model_used"),
+                        image_seed=image.get("seed"),
+                        image_prompt=image.get("image_prompt"),  # Save the final refined prompt
+                        angle_key=angle.get("key"),
+                        angle_name=angle.get("name"),
+                        angle_trigger=angle.get("trigger"),
+                        angle_category=angle.get("category"),
+                        concept_key=concept.get("key"),
+                        concept_name=concept.get("name"),
+                        concept_structure=concept.get("structure"),
+                        concept_visual=concept.get("visual"),
+                        concept_category=concept.get("category"),
+                        generation_method="angle_concept_matrix",
+                        metadata={"generation_method": "angle_concept_matrix", "variation_index": img_idx, "total_variations": len(images)},
+                    )
+                    if db_id:
+                        saved_ad_ids.append(db_id)
+                        print(f"✓ Saved matrix ad creative variation {img_idx + 1}/{len(images)} to database: {db_id}")
+                except Exception as e:
+                    print(f"Warning: Failed to save matrix variation {img_idx + 1} to database: {e}")
+        
+        # Use first saved ID as the main ad_id
+        if saved_ad_ids:
+            ad_id = saved_ad_ids[0]
         
         return {
             "id": ad_id,
@@ -2089,9 +2105,6 @@ CONCEPT: {concept['name']}
                 print(f"Warning: No images generated for strategy {idx + 1}, skipping...")
                 continue
             
-            # Get first image for database
-            first_image = generated_images[0]
-            
             # Build ad copy
             strategy = creative_strategies[idx]
             headline = title or strategy.titleIdeas or "Check this out"
@@ -2099,44 +2112,53 @@ CONCEPT: {concept['name']}
             description_text = description or strategy.captionIdeas or ""
             cta = strategy.cta or "Learn More"
             
-            # Save to database
+            # Save to database (save EACH image as a separate record for gallery visibility)
             ad_id = str(uuid.uuid4())
-            db_id = None
+            saved_ad_ids = []
             
             if db_service and username:
-                try:
-                    db_id = await db_service.save_ad_creative(
-                        niche=niche,
-                        title=title or "",
-                        headline=headline,
-                        primary_text=primary_text,
-                        description=description_text,
-                        body_story=primary_text,  # Use primary_text as body_story
-                        cta=cta,
-                        psychological_angle=strategy.phsychologyTrigger or "",
-                        why_it_works=f"Angle: {strategy.angle}, Concept: {strategy.concept}",
-                        username=username,  # Pass username
-                        image_url=first_image.get("image_url"),
-                        image_filename=first_image.get("filename"),
-                        image_model=first_image.get("model_used"),
-                        image_prompt=first_image.get("image_prompt"),
-                        generation_method="extensive",
-                        metadata={
-                            "generation_method": "extensive",
-                            "target_audience": target_audience,
-                            "offer": offer,
-                            "strategy_index": idx,
-                            "psychology_trigger": strategy.phsychologyTrigger,
-                            "angle": strategy.angle,
-                            "concept": strategy.concept,
-                            "visual_direction": strategy.visualDirection,
-                        },
-                    )
-                    if db_id:
-                        ad_id = db_id
-                        print(f"✓ Saved extensive ad creative to database: {ad_id}")
-                except Exception as e:
-                    print(f"Warning: Failed to save to database: {e}")
+                for img_idx, image in enumerate(generated_images):
+                    if image.get("error"):
+                        continue  # Skip failed images
+                    try:
+                        db_id = await db_service.save_ad_creative(
+                            niche=niche,
+                            title=title or "",
+                            headline=headline,
+                            primary_text=primary_text,
+                            description=description_text,
+                            body_story=primary_text,  # Use primary_text as body_story
+                            cta=cta,
+                            psychological_angle=strategy.phsychologyTrigger or "",
+                            why_it_works=f"Angle: {strategy.angle}, Concept: {strategy.concept}",
+                            username=username,  # Pass username
+                            image_url=image.get("image_url"),
+                            image_filename=image.get("filename"),
+                            image_model=image.get("model_used"),
+                            image_prompt=image.get("image_prompt"),
+                            generation_method="extensive",
+                            metadata={
+                                "generation_method": "extensive",
+                                "target_audience": target_audience,
+                                "offer": offer,
+                                "strategy_index": idx,
+                                "psychology_trigger": strategy.phsychologyTrigger,
+                                "angle": strategy.angle,
+                                "concept": strategy.concept,
+                                "visual_direction": strategy.visualDirection,
+                                "variation_index": img_idx,
+                                "total_variations": len(generated_images),
+                            },
+                        )
+                        if db_id:
+                            saved_ad_ids.append(db_id)
+                            print(f"✓ Saved extensive ad creative variation {img_idx + 1}/{len(generated_images)} to database: {db_id}")
+                    except Exception as e:
+                        print(f"Warning: Failed to save extensive variation {img_idx + 1} to database: {e}")
+            
+            # Use first saved ID as the main ad_id
+            if saved_ad_ids:
+                ad_id = saved_ad_ids[0]
             
             all_results.append({
                 "id": ad_id,