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

class Instructions_revision(Page):
    def is_displayed(player):
        return player.round_number == 1

class Revision(Page):
    def vars_for_template(self):
        treatment = self.participant.vars['treatment']
        product_order = self.participant.vars['product_order']
        proudct_number = product_order[self.round_number-1]
        print("Current surgery number " + str(proudct_number))
        
        demand_table, algorithm_advice, actual_demand, demand_data, mu, sigma, min_level, max_level, reg_a, reg_b, reg_p = composer_fc.get_demand_table(proudct_number)
        
        series = []

        series.append({
            'name': ' ',
            'data': demand_data
        })
    
        return{
            'round_number' : self.round_number,
            'demand_table' : demand_table, 
            'highcharts_series' : series, 
            'min_level' : min_level,
            'max_level' : max_level
        }

    form_model = 'player'

def before_next_page(self):
        self.participant.vars['rounds'] = self.round_number
        product_order = self.participant.vars['product_order']
        self.participant.vars['initial_judgement_' + str(product_order[self.round_number-1])] = self.player.initial_judgement

page_sequence = [Instructions_revision, Revision]