class AbstractBCAgent():

    def update(self, community):
        opinions = [peer.opinion for peer in self.peers(community)]
        self.opinion = self.revise(opinions)

    def peers(self, community):
        peers = [
            agent for agent in community
            if self.distance(agent.opinion) <= epsilon
        ]
        return peers

    def distance(self, opinion):
        pass 

    def revise(self, opinions):
        pass

class NumericalBCAgent(AbstractBCAgent):

    def distance(self, opinion):
        """calculates distance between agent's and other opinion"""
        return abs(opinion - self.opinion)

    def revise(self, opinions):
        """revision through weighted opinion averaging"""
        alpha = self._parameters.get("alpha", .5)
        revision = alpha * self.opinion + (1-alpha) * np.mean(opinions)
        return revision

class NumericalBCAgent(AbstractBCAgent):

    def distance(self, opinion):
        """calculates distance between agent's and other opinion"""
        return abs(opinion - self.opinion)

    def revise(self, opinions):
        """revision through weighted opinion averaging"""
        alpha = self._parameters.get("alpha", .5)
        revision = alpha * self.opinion + (1-alpha) * np.mean(opinions)
        return revision

[
    #1
    "Consuming a vegan diet directly contributes to reducing greenhouse "
    "gas emissions, as animal agriculture is a significant source of "
    "environmental pollution.",
    #2
    "The scientific evidence supports the health benefits of a vegan diet, "
    "which can lead to a reduced risk of various diseases, such as diabetes, "
    "high blood pressure, and some cancers.",
    #3
    "Veganism doesn't support a healthy and balanced diet.",
    #4
    "There is a negative impact on the environment and economy when people "
    "follow a vegan diet.",
    #5
    "A vegan diet can prevent certain types of cancer.",
    #6
    "Reducing meat consumption is necessary to avoid a global food crisis.",
    #7
    "Contrary to popular belief, studies suggest that a well-planned "
    "traditional omnivorous diet may reduce the risk of certain diseases "
    "compared to a vegan diet.",
    #8
    "While plant-based diets have their benefits, they are not always easy "
    "to stick to in the long run.",
    #9
    "As someone who has been vegan for over a year, my energy levels have "
    "increased significantly while my risk of certain diseases has decreased.",
    #10
    "My personal experience as a vegan for two years has been plagued with "
    "deficiencies and malnutrition, leading to low energy levels and "
    "compromised health."
]

model = lmql.model(
    "local:HuggingFaceH4/zephyr-7b-alpha",
    device_map = "auto",
    load_in_8bit=True,
    low_cpu_mem_usage=True
)

class NaturalLanguageBCAgent(AbstractBCAgent):

    def distance(self, other):
        """distance as expected agreement level"""
        lmql_result = agreement_lmq(
            self.opinion, other, **kwargs
        )
        probs = lmql_result.variables.get("P(LABEL)")
        return sum([i*v for i, (_, v) in enumerate(probs)])/4.0

    def revise(self, peer_opinions):
        """natural language opinion revision"""
        revision = revise_lmq(
            self.opinion, peer_opinions, **kwargs
        )
        return revision