from otree.api import *
from otree import settings
import numpy as np
import pandas as pd


doc = """Experiment 2 for Social Inference project
"""

# TO DOs ---------------------------------------------------------------------------------------------------------------

# OTREE CLASSES --------------------------------------------------------------------------------------------------------
class C(BaseConstants):
    NAME_IN_URL = 'SI_Exp2_FeedbackQuestions'
    PLAYERS_PER_GROUP = None
    NUM_ROUNDS = 1

class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    Q2 = models.IntegerField(label="Do you think Observer B was a human or a computer?", widget=widgets.RadioSelect, choices=[[1, 'a) Human'], [2, 'b) Computer'],] )
    Q1 = models.LongStringField(label="What did you think about Observer B and their answers? Did they do a good job? Feel free to share any observations or comments you have regarding Observer B.", blank=True)


class Q1(Page):
    template_name = 'SI_Exp2_FeedbackQuestion/Q.html'
    form_model = "player"
    form_fields = ["Q1"]
    def is_displayed(player: Player):
        session_name = player.session.config['name']
        return session_name != 'SI_Exp2_a'


class Q2(Page):
    template_name = 'SI_Exp2_FeedbackQuestion/Q.html'
    form_model = "player"
    form_fields = ["Q2"]
    def is_displayed(player: Player):
        session_name = player.session.config['name']
        return session_name != 'SI_Exp2_a'

class Debriefing(Page):
    template_name = 'SI_Exp2_FeedbackQuestion/Debriefing.html'
    def vars_for_template(player: Player):
        #print('------------------------------------------')
        #print('player.participant', player.participant)
        #print('player.id_in_group', player.id_in_group)
        #pay = player.participant.vars['payout']
        pay = player.participant.vars['you_get']
        #player.participant.vars['payout']
        #print('debriefing payout', pay)
        order_cellPrompts =  player.participant.vars['order']
        if pay > 3:
            pay = 3
        else:
            pay = pay
        return {'payout': pay, 'order': order_cellPrompts}

# PAGES ----------------------------------------------------------------------------------------------------------------

page_sequence = [Q1, Q2, Debriefing]