from rest_framework import serializers


class GenerateTitleRequestSerializer(serializers.Serializer):
    prompt = serializers.CharField(trim_whitespace=True, allow_blank=False)
    model = serializers.CharField(
        required=False,
        allow_blank=True,
        trim_whitespace=True,
        max_length=64,
    )
    image_url = serializers.URLField(required=False, allow_blank=True)
    image_base64 = serializers.CharField(required=False, allow_blank=True)
    resolution = serializers.CharField(
        required=False,
        allow_blank=True,
        trim_whitespace=True,
        default="1K",
    )
    parameters = serializers.DictField(required=False, default=dict)


class GenerateImageRequestSerializer(serializers.Serializer):
    prompt = serializers.CharField(trim_whitespace=True, allow_blank=False)
    model = serializers.CharField(
        required=False,
        allow_blank=True,
        trim_whitespace=True,
        max_length=64,
    )
    image_url = serializers.URLField(required=False, allow_blank=True)
    image_base64 = serializers.CharField(required=False, allow_blank=True)
    resolution = serializers.CharField(
        required=False,
        allow_blank=True,
        trim_whitespace=True,
        default="1K",
    )
    aspect_ratio = serializers.CharField(
        required=False,
        allow_blank=True,
        trim_whitespace=True,
        default="1:1",
    )
    parameters = serializers.DictField(required=False, default=dict)