from otree.api import *
import random

doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'D_Outro'
    PLAYERS_PER_GROUP = None
    NUM_ROUNDS = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    completed_survey = models.BooleanField(doc="indicates whether a participant has completed the survey.",
                                           initial=False,
                                           blank=True)

    # Covariates
    comprehension_1 = models.IntegerField(
        doc="Overall, I understood the rules of the first task well.",
        label="Overall, I understood the rules of the first task well.",
        widget=widgets.RadioSelect,
        choices=[1, 2, 3, 4, 5, 6, 7])

    comprehension_2 = models.IntegerField(
        doc="Overall, I understood the rules of the first second well.",
        label="Overall, I understood the rules of the second task well.",
        widget=widgets.RadioSelect,
        choices=[1, 2, 3, 4, 5, 6, 7])


    # Demographics
    age = models.IntegerField(label="Please enter your age",
                              min=18,
                              max=99)

    gender = models.IntegerField(
        label="Please select your gender.",
        choices=[
            [1, "Female"],
            [2, "Male"],
            [3, "Other"],
            [4, "Prefer not to say"],
        ]
    )

    # OTF
    OTF = models.LongStringField(
        label="Did you encounter any difficulties answering the study or do you have any comments?", blank=True)

    # gen AI
    genAI = models.StringField(
        doc="Did you use ChatGPT or any other generative AI to better understand the instructions? (Note: we will not reject your submission, if you did. We need the information to assess our overall data quality.)",
        label="Did you  use ChatGPT or any other generative AI to better understand the instructions? (Note: we will not reject your submission, if you did. We need the information to assess our overall data quality.)",
        widget=widgets.RadioSelect,
        choices=["Yes", "No"],
        blank=False)

    genAI_OTF = models.LongStringField(
        label="If so, did it advise you on what to do? Did you follow the advice?", blank=True)


# PAGES
class Open_Text(Page):
    form_model = "player"
    form_fields = ['OTF']

    @staticmethod
    def is_displayed(player):
        if player.participant.is_dropout == False:
            if player.participant.waited_too_long == False:
                return True
            else:
                return False
        else:
            return False

class Covariates(Page):
    form_model = "player"
    form_fields = ['comprehension_1', 'comprehension_2']

    @staticmethod
    def is_displayed(player):
        if player.participant.is_dropout == False:
            if player.participant.waited_too_long == False:
                return True
            else:
                return False
        else:
            return False

class Demographics(Page):
    form_model = "player"
    form_fields = ['age', 'gender', 'genAI', 'genAI_OTF']

    @staticmethod
    def is_displayed(player):
        if player.participant.is_dropout == False:
            if player.participant.waited_too_long == False:
                return True
            else:
                return False
        else:
            return False


    @staticmethod
    def before_next_page(player, timeout_happened):
        player.participant.finished = True
        player.completed_survey = player.participant.finished


class Debriefing(Page):
    form_model = 'player'

    @staticmethod
    def vars_for_template(player: Player):
        return dict(
            final_payoff=player.participant.payoff_plus_participation_fee().to_real_world_currency(player.session),
        )


page_sequence = [Open_Text, Covariates, Demographics, Debriefing]