from rest_framework import serializers
from .models import DetectionResult, DetectionDetail

class DetectionDetailSerializer(serializers.ModelSerializer):
    face_crop_image = serializers.ImageField(use_url=True)

    class Meta:
        model = DetectionDetail
        fields = [
            'gender', 'matched_name', 'laborer_id',
            'match_score', 'laborer_details', 'face_crop_image'
        ]

class DetectionResultSerializer(serializers.ModelSerializer):
    original_image = serializers.ImageField(use_url=True)
    debug_image = serializers.ImageField(use_url=True)
    detections = DetectionDetailSerializer(many=True, read_only=True)

    class Meta:
        model = DetectionResult
        fields = [
            'id', 'original_image', 'debug_image',
            'detected_total', 'detected_males', 'detected_females',
            'timestamp', 'detections'
        ]