Updated iamge prompt flow adn frotnend backend
This commit is contained in:
@@ -111,6 +111,7 @@ class TasksSerializer(serializers.ModelSerializer):
|
||||
class ImagesSerializer(serializers.ModelSerializer):
|
||||
"""Serializer for Images model"""
|
||||
task_title = serializers.SerializerMethodField()
|
||||
content_title = serializers.SerializerMethodField()
|
||||
|
||||
class Meta:
|
||||
model = Images
|
||||
@@ -118,6 +119,8 @@ class ImagesSerializer(serializers.ModelSerializer):
|
||||
'id',
|
||||
'task_id',
|
||||
'task_title',
|
||||
'content_id',
|
||||
'content_title',
|
||||
'image_type',
|
||||
'image_url',
|
||||
'image_path',
|
||||
@@ -139,6 +142,38 @@ class ImagesSerializer(serializers.ModelSerializer):
|
||||
except Tasks.DoesNotExist:
|
||||
return None
|
||||
return None
|
||||
|
||||
def get_content_title(self, obj):
|
||||
"""Get content title"""
|
||||
if obj.content:
|
||||
return obj.content.title or obj.content.meta_title
|
||||
return None
|
||||
|
||||
|
||||
class ContentImageSerializer(serializers.ModelSerializer):
|
||||
"""Serializer for individual image in grouped content images"""
|
||||
class Meta:
|
||||
model = Images
|
||||
fields = [
|
||||
'id',
|
||||
'image_type',
|
||||
'image_url',
|
||||
'image_path',
|
||||
'prompt',
|
||||
'status',
|
||||
'position',
|
||||
'created_at',
|
||||
'updated_at',
|
||||
]
|
||||
|
||||
|
||||
class ContentImagesGroupSerializer(serializers.Serializer):
|
||||
"""Serializer for grouped content images - one row per content"""
|
||||
content_id = serializers.IntegerField()
|
||||
content_title = serializers.CharField()
|
||||
featured_image = ContentImageSerializer(allow_null=True)
|
||||
in_article_images = ContentImageSerializer(many=True)
|
||||
overall_status = serializers.CharField() # 'pending', 'partial', 'complete', 'failed'
|
||||
|
||||
|
||||
class ContentSerializer(serializers.ModelSerializer):
|
||||
|
||||
Reference in New Issue
Block a user