model.py

"""
Dummy model module.

This file contains a placeholder model class `DummyGenderModel` that returns
random predictions. Replace the `predict` method or the whole class with your
trained model loading + inference logic when ready.
"""

import random
from typing import Dict
from PIL.Image import Image

class DummyGenderModel:
    def __init__(self):
        self.version = "dummy-v0.1"
        self._loaded = True

    def is_loaded(self) -> bool:
        return self._loaded

    def predict(self, img: Image) -> Dict:
        """
        Return a random prediction. Expects a PIL Image.

        Returns a dict:
        {
            "label": "male" or "female",
            "confidence": float between 0.5 and 1.0,
            "probabilities": {"male": p_male, "female": p_female}
        }
        """
        # random base probability
        p_male = random.random()
        p_female = 1.0 - p_male

        # calibrate to avoid extremely low confidences — push to [0.5, 1.0]
        p_male = 0.5 + 0.5 * p_male
        p_female = 1.0 - p_male

        if p_male >= p_female:
            label = "male"
            # confidence = p_male
        else:
            label = "female"
            # confidence = p_female

        return {
            "label": label,
            # "confidence": round(confidence, 4),
            # "probabilities": {"male": round(p_male, 4), "female": round(p_female, 4)}
        }