46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
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)
|