Spaces:
Runtime error
Runtime error
File size: 4,116 Bytes
790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 790a22c a18ebc5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
import random
from typing import Literal
def generate_question_data(level: Literal["easy", "medium", "hard"] = "easy"):
"""generate question, its numbers and proper answer"""
nums = generate_numbers_by_level(level)
cur_num = nums['current_number'] # current number
ord_num = nums['ordinal_number'] # ordinal number
seq_up_by_one = generate_number_sequence(cur_num, ord_num=1, times=1) # sequence with ord_num = 1, times = 1
count_up_by_one_questions = [
{
"question": f"Let's practice counting. After {cur_num}, what number is next?\n{seq_up_by_one}",
"current_number": cur_num,
"ordinal_number": 1,
"times": 1,
"answer": cur_num + 1
}
]
seq_up_by_ord = generate_number_sequence(cur_num, ord_num, times=1) # sequence with times = 1
count_up_by_ord_questions = [
{
"question": f"What number comes {ord_num} number after {cur_num}?\n{seq_up_by_ord}",
"current_number": cur_num,
"ordinal_number": ord_num,
"times": 1,
"answer": cur_num + ord_num
},
{
"question": f"If we count up {ord_num} from {cur_num}, what number is next?\n{seq_up_by_ord}",
"current_number": cur_num,
"ordinal_number": ord_num,
"times": 1,
"answer": cur_num + ord_num
}
]
times = 1 if level == "easy" else nums['times']
times_ord_seq = generate_number_sequence(cur_num, ord_num, times)
times_ord_questions = [
{
"question": f"We're counting up by {times}s. What number is {ord_num} after {cur_num}?\n{times_ord_seq}",
"current_number": cur_num,
"ordinal_number": ord_num,
"times": times,
"answer": cur_num + ord_num * times
}
]
times_only_seq = generate_number_sequence(cur_num, 1, times) # sequence with ordinal number = 1
times_only_questions = [
{
"question": f"Let's count up by {times}s. What number is next if we start from {cur_num}?\n{times_only_seq}",
"current_number": cur_num,
"ordinal_number": 1,
"times": times,
"answer": cur_num + times
}
]
questions = [*count_up_by_one_questions, *count_up_by_ord_questions, *times_only_questions, *times_ord_questions]
random_choice = random.choice(questions)
return random_choice
def generate_numbers_by_level(level: Literal["easy", "medium", "hard"] = "easy"):
"""generate current number, ordinal number and times parameter
returns
dict with params:
:param current_number: current number
:param ordinal numebr: the number we count up by
:param times: the number of times we count up by ordinal number"""
if level == "easy":
cur_num = random.randint(1, 8)
ord_num = random.randint(1, 2)
times = 1
elif level == "medium":
cur_num = random.randint(1, 94)
ord_num = random.randint(1, 3)
times = random.randint(1, 2)
elif level == "hard":
cur_num = random.randint(1, 488)
ord_num = random.randint(1, 4)
times = random.randint(1, 2)
return {
"current_number": cur_num,
"ordinal_number": ord_num,
"times": times
}
def generate_number_sequence(cur_num, ord_num, times=1):
"""generate one of 2 sequences. For example we want 55 to be a right answer, then sequences can be:
52 53 54 ...
... 56 57 58
parameters
:cur_num: current number
:ord_num: ordinal number
:times: times"""
max_num = cur_num + times * ord_num
seq_before = [str(num) for num in range(max_num - times, 0, -times)][:3][::-1]
seq_after = [str(num) for num in range(max_num + times, max_num + 4 * times, times)]
seq_before.append("...")
seq_after.insert(0, "...")
seqs = []
if len(seq_before) == 4:
seqs.append(seq_before)
if len(seq_after) == 4:
seqs.append(seq_after)
rand_seq = " ".join(random.choice(seqs))
return rand_seq
|