difficulty_controller.py

class DifficultyController:
    """
    Adjusts world difficulty based on agent performance trends.
    """

    def __init__(self, config):

        self.config = config
        self.history = []

        self.difficulty = 1.0  # baseline

    # =====================================================
    # UPDATE SIGNAL
    # =====================================================
    def observe(self, performance_score):

        self.history.append(performance_score)

        if len(self.history) > 50:
            self.history.pop(0)

        self._update_difficulty()

    # =====================================================
    # COMPUTE ADJUSTMENT
    # =====================================================
    def _update_difficulty(self):

        if not self.history:
            return

        avg = sum(self.history) / len(self.history)

        # target band: agents should not dominate or collapse
        target = 1.0

        # bounded adjustment
        if avg > target + 0.2:
            self.difficulty += 0.05  # world gets harder

        elif avg < target - 0.2:
            self.difficulty -= 0.05  # world gets easier

        # clamp
        self.difficulty = max(0.5, min(3.0, self.difficulty))

    # =====================================================
    # GET CURRENT DIFFICULTY
    # =====================================================
    def get(self):

        return self.difficulty