import random

import numpy as np
from otree.api import *


author = 'Your name here'
doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'display_image'
    players_per_group = None
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    number_image_displayed = models.IntegerField()
    rating = models.FloatField()


# FUNCTIONS
# PAGES
class MyPage(Page):
    form_model = 'player'
    form_fields = ['rating']

    @staticmethod
    def vars_for_template(player: Player):
        number = random.randint(1, 2)
        player.number_image_displayed = number
        return dict(image_path='display_image/{}.jpg'.format(number))


class Results(Page):
    pass


page_sequence = [MyPage, Results]