from otree.api import *

US_HISTORY_PATH = 'prisoner/JeopardyStudyQuestions/AmericanHistory.csv'
GEOGRAPHY_PATH = 'prisoner/JeopardyStudyQuestions/Geography.csv'
LITERATURE_PATH = 'prisoner/JeopardyStudyQuestions/Literature.csv'
POP_CULTURE_PATH = 'prisoner/JeopardyStudyQuestions/PopCulture.csv'
SCIENCE_PATH = 'prisoner/JeopardyStudyQuestions/Science.csv'
ANIMAL_NAMES = ['Elephant', 'Falcon', 'Kangaroo', 'Tiger', 'Giraffe', 'Panda', 'Owl', 'Bear']
COLOR_NAMES = ['Purple', 'Blue', 'Green', 'Orange', 'Yellow', 'Red', 'Lavender'] 

class C(BaseConstants):
    NAME_IN_URL = 'user'
    PLAYERS_PER_GROUP = 4
    NUM_ROUNDS = 15 
    PAYOFF_A = cu(100)
    PAYOFF_B = cu(200)
    PAYOFF_C = cu(300)
    AGENT1_NAME, AGENT2_NAME, AGENT3_NAME, AGENT4_NAME = 'Player 1', 'Player 2', 'Player 3', 'Player 4'

    # Categories
    US_HISTORY, US_HISTORY_STR = 'AH', 'US_history'
    GEOGRAPHY, GEOGRAPHY_STR = 'GEO', 'geography'
    LITERATURE, LITERATURE_STR = 'LIT', 'literature'
    POP_CULTURE, POP_CULTURE_STR = 'PC', 'pop_culture'
    SCIENCE, SCIENCE_STR = 'SC', 'science'

    # Difficulties
    EASY, EASY_STRING = '1', 'Easy'
    MEDIUM, MEDIUM_STRING = '2', 'Medium'
    HARD, HARD_STRING = '3', 'Hard'

    # Scores
    EASY_SCORE = 100
    MEDIUM_SCORE = 200
    HARD_SCORE = 300
    
    # AI Correctness Probability Values
    AI_EASY_QUESTION_PROB = 0.75
    AI_MEDIUM_QUESTION_PROB = 0.7
    AI_HARD_QUESTION_PROB = 0.6

    # AI Mean Confidence Values
    AI_EASY_CONF_MEAN, AI_EASY_CONF_STD = 0.85, 0.10
    AI_MEDIUM_CONF_MEAN, AI_MEDIUM_CONF_STD = 0.75, 0.12
    AI_HARD_CONF_MEAN, AI_HARD_CONF_STD = 0.65, 0.15

    # Discussion Page minimum time seconds
    DISCUSSION_PAGE_MIN_TIME = 45