from rest_framework import serializers


class AskRequestSerializer(serializers.Serializer):
    conversationId = serializers.CharField(required=False, allow_null=True, allow_blank=True)
    lessonId = serializers.CharField(required=False, allow_null=True, allow_blank=True)
    chapterId = serializers.CharField(required=False, allow_null=True, allow_blank=True)
    gradeId = serializers.IntegerField(required=False, allow_null=True)
    questionId = serializers.CharField(required=False, allow_null=True, allow_blank=True)
    userQuestion = serializers.CharField()
    stream = serializers.BooleanField(required=False, default=False)


class AskResponseSerializer(serializers.Serializer):
    conversationId = serializers.CharField()
    messageId = serializers.CharField()
    answer = serializers.CharField()
    suggestedFollowUps = serializers.ListField(child=serializers.CharField())
    remainingQuota = serializers.IntegerField()


class MessageSerializer(serializers.Serializer):
    messageId = serializers.CharField()
    role = serializers.CharField()
    text = serializers.CharField()


class ConversationMessagesResponseSerializer(serializers.Serializer):
    conversationId = serializers.CharField()
    lessonId = serializers.CharField(allow_null=True)
    messages = MessageSerializer(many=True)


class FeedbackRequestSerializer(serializers.Serializer):
    isHelpful = serializers.BooleanField()


class FeedbackResponseSerializer(serializers.Serializer):
    success = serializers.BooleanField()