added imae adn prompt icons on contetn page

backend/igny8_core/modules/writer/serializers.py

@@ -1,4 +1,5 @@
 from rest_framework import serializers
+from django.db import models
 from .models import Tasks, Images, Content
 from igny8_core.modules.planner.models import Clusters, ContentIdeas
 
@@ -180,6 +181,8 @@ class ContentSerializer(serializers.ModelSerializer):
     """Serializer for Content model"""
     task_title = serializers.SerializerMethodField()
     sector_name = serializers.SerializerMethodField()
+    has_image_prompts = serializers.SerializerMethodField()
+    has_generated_images = serializers.SerializerMethodField()
     
     class Meta:
         model = Content
@@ -202,6 +205,8 @@ class ContentSerializer(serializers.ModelSerializer):
             'generated_at',
             'updated_at',
             'account_id',
+            'has_image_prompts',
+            'has_generated_images',
         ]
         read_only_fields = ['id', 'generated_at', 'updated_at', 'account_id']
     
@@ -225,4 +230,20 @@ class ContentSerializer(serializers.ModelSerializer):
             except Sector.DoesNotExist:
                 return None
         return None
+    
+    def get_has_image_prompts(self, obj):
+        """Check if content has any image prompts generated"""
+        # Check if any images exist with prompts for this content
+        return Images.objects.filter(
+            models.Q(content=obj) | models.Q(task=obj.task)
+        ).exclude(prompt__isnull=True).exclude(prompt='').exists()
+    
+    def get_has_generated_images(self, obj):
+        """Check if content has any generated images (status='generated' and has URL)"""
+        # Check if any images are generated (have status='generated' and image_url)
+        return Images.objects.filter(
+            models.Q(content=obj) | models.Q(task=obj.task),
+            status='generated',
+            image_url__isnull=False
+        ).exclude(image_url='').exists()