from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants

class Mot_survey(Page):
    form_model = 'player'
    form_fields = ['Intrinsic_mot_1', 'Intrinsic_mot_2', 'Emotional_reaction_1', 'Emotional_reaction_2', 'Emot_happy', 'Emot_disappointed']
class Control_survey_AI(Page):
    form_model = 'player'
    form_fields = ['Desirablity', 'Self_efficacy', 'Expectancy', 'test_human']
    def is_displayed(self):
        if self.participant.vars['treatment'] == 'AI':
            return True
        else:
            return False
class Control_survey_human(Page):
    form_model = 'player'
    form_fields = ['Desirablity', 'Self_efficacy', 'Expectancy', 'test_AI']
    def is_displayed(self):
        if self.participant.vars['treatment'] == 'human':
            return True
        else:
            return False
class Feedback(Page):
    form_model = 'player'
    form_fields = ['interested_in_results', 'contact']
class End(Page):
    form_model = 'player'
page_sequence = [Mot_survey, Control_survey_AI, Control_survey_human, Feedback, End]