update generators module

scripts/quiz/generators.py

@@ -1,33 +1,20 @@
+import random
 from .questions import generate_question_data
-from .utils import get_next_level
+from .utils import get_next_difficulty, generate_start_step
 
 
-def start_interactive_math(right_answers=0, wrong_answers=0, level="easy"):
-    if wrong_answers > 2:
-        wrong_answers = 0
-        right_answers = 0
-        level = get_next_level(level, False)
-    elif right_answers > 2:
-        right_answers = 0
-        wrong_answers = 0
-        level = get_next_level(level)
-        
-    question_data = generate_question_data(level)
+def start_interactive_math(difficulty=0.01, do_increase: True | False = True):
+    next_difficulty = get_next_difficulty(difficulty, do_increase)
+    start, step = generate_start_step(difficulty)
+    question_data = generate_question_data(start, step, question_num=random.randint(0, 5))
+
     question = question_data['question']
-    right_answer = question_data['answer']
-    cur_num = question_data['current_number']
-    ord_num = question_data['ordinal_number']
-    times = question_data['times']
+    start = question_data['start']
+    step = question_data['step']
 
-    numbers_group = [cur_num, ord_num, times]
     output = {
         "text": question,
-        "question_numbers": numbers_group,
-        "right_answer": right_answer,
-        'number_correct': right_answers,
-        'number_incorrect': wrong_answers,
-        'level': level,
-        "hints_used": 0
+        "difficulty": next_difficulty,
+        "question_numbers": [start, step, start + step]
     }
     return output
-