Updated iamge prompt flow adn frotnend backend

backend/igny8_core/modules/writer/models.py

@@ -138,7 +138,7 @@ class Content(SiteSectorBaseModel):
 
 
 class Images(SiteSectorBaseModel):
-    """Images model for task-related images (featured, desktop, mobile)"""
+    """Images model for content-related images (featured, desktop, mobile, in-article)"""
     
     IMAGE_TYPE_CHOICES = [
         ('featured', 'Featured Image'),
@@ -147,11 +147,21 @@ class Images(SiteSectorBaseModel):
         ('in_article', 'In-Article Image'),
     ]
     
+    content = models.ForeignKey(
+        Content,
+        on_delete=models.CASCADE,
+        related_name='images',
+        null=True,
+        blank=True,
+        help_text="The content this image belongs to (preferred)"
+    )
     task = models.ForeignKey(
         Tasks,
         on_delete=models.CASCADE,
         related_name='images',
-        help_text="The task this image belongs to"
+        null=True,
+        blank=True,
+        help_text="The task this image belongs to (legacy, use content instead)"
     )
     image_type = models.CharField(max_length=50, choices=IMAGE_TYPE_CHOICES, default='featured')
     image_url = models.URLField(blank=True, null=True, help_text="URL of the generated/stored image")
@@ -164,23 +174,33 @@ class Images(SiteSectorBaseModel):
     
     class Meta:
         db_table = 'igny8_images'
-        ordering = ['task', 'position', '-created_at']
+        ordering = ['content', 'position', '-created_at']
         verbose_name = 'Image'
         verbose_name_plural = 'Images'
         indexes = [
+            models.Index(fields=['content', 'image_type']),
             models.Index(fields=['task', 'image_type']),
             models.Index(fields=['status']),
+            models.Index(fields=['content', 'position']),
             models.Index(fields=['task', 'position']),
         ]
     
     def save(self, *args, **kwargs):
-        """Automatically set account, site, and sector from task"""
-        if self.task:
+        """Automatically set account, site, and sector from content or task"""
+        # Prefer content over task
+        if self.content:
+            self.account = self.content.account
+            self.site = self.content.site
+            self.sector = self.content.sector
+        elif self.task:
             self.account = self.task.account
             self.site = self.task.site
             self.sector = self.task.sector
         super().save(*args, **kwargs)
     
     def __str__(self):
-        return f"{self.task.title} - {self.image_type}"
+        content_title = self.content.title if self.content else None
+        task_title = self.task.title if self.task else None
+        title = content_title or task_title or 'Unknown'
+        return f"{title} - {self.image_type}"