Update app.py

app.py

@@ -54,60 +54,71 @@ except Exception as e:
 def generate_quiz(text: str):
     """
     Генерирует один вопрос и два варианта ответа на основе текста.
-    Использует простой алгоритм выбора предложений из текста.
+    Алгоритмы:
+      1. Базовый: случайное предложение и первые слова.
+      2. Пропуск ключевого слова.
+      3. Вопрос о числе/дате.
     """
     try:
-        # Разбиваем текст на предложения
         sentences = [s.strip() for s in text.replace("!", ".").replace("?", ".").split(".") if s.strip()]
-        
         if len(sentences) < 1:
-            raise ValueError("Текст слишком короткий, не смог найти предложения")
-        
-        # Выбираем случайное предложение для вопроса
-        question_sentence = random.choice(sentences)
-        
-        # Разбиваем предложение на слова
-        words = question_sentence.split()
-        
-        if len(words) <= 3:
-            # Если предложение слишком короткое
-            correct_answer = question_sentence
-            question = f"Что сказано в этом предложении?"
-        else:
-            # Формируем вопрос
-            first_word = words[0].lower()
-            if first_word in ["кто", "что", "где", "когда", "почему", "как", "сколько"]:
-                question = question_sentence + "?"
+            raise ValueError("Текст слишком короткий")
+
+        algo = random.choice([1, 2, 3])
+        # ------------------------
+        if algo == 1:
+            # Базовый алгоритм
+            question_sentence = random.choice(sentences)
+            words = question_sentence.split()
+            if len(words) <= 3:
+                correct_answer = question_sentence
+                question = "Что сказано в этом предложении?"
             else:
                 question = "Что сказано в тексте?"
+                correct_answer = " ".join(words[:6]) + ("..." if len(words) > 6 else "")
             
-            # Правильный ответ — первые несколько слов предложения
-            correct_answer = " ".join(words[:6]) + ("..." if len(words) > 6 else "")
+            wrong_sentence = random.choice([s for s in sentences if s != question_sentence] or ["Другая информация"])
+            wrong_words = wrong_sentence.split()
+            wrong_answer = " ".join(wrong_words[:6]) + ("..." if len(wrong_words) > 6 else "")
         
-        # Генерируем неправильный вариант
-        if len(sentences) > 1:
-            # Берём другое предложение из текста
-            wrong_options = [s for s in sentences if s != question_sentence]
-            if wrong_options:
-                wrong_sentence = random.choice(wrong_options)
-                wrong_words = wrong_sentence.split()
-                wrong_answer = " ".join(wrong_words[:6]) + ("..." if len(wrong_words) > 6 else "")
+        # ------------------------
+        elif algo == 2:
+            # Пропуск ключевого слова
+            question_sentence = random.choice(sentences)
+            words = question_sentence.split()
+            if len(words) > 2:
+                key_word = random.choice(words)
+                question = question_sentence.replace(key_word, "_____")
+                correct_answer = key_word
+                wrong_answer = random.choice([w for w in words if w != key_word] or ["другое"])
             else:
-                wrong_answer = "Это не упоминается в тексте"
-        else:
-            # Если только одно предложение — придумываем очевидную ложь
-            wrong_answer = "Это совершенно другая информация"
+                # fallback
+                return generate_quiz(text)
+        
+        # ------------------------
+        elif algo == 3:
+            # Вопрос о числе или дате
+            import re
+            question_sentence = random.choice(sentences)
+            numbers = re.findall(r'\d+', question_sentence)
+            if numbers:
+                number = random.choice(numbers)
+                question = question_sentence.replace(number, "_____")
+                correct_answer = number
+                wrong_answer = str(int(number)+random.randint(1,5))
+            else:
+                # fallback к базовому
+                return generate_quiz(text)
         
-        # Перемешиваем варианты
         options = [correct_answer, wrong_answer]
         random.shuffle(options)
-        
         return question, options, correct_answer
-        
+
     except Exception as e:
         raise ValueError(f"Ошибка генерации вопроса: {str(e)}")
 
 
+
 def synthesize_audio(text_ru: str):
     """Переводит русскую строку на казахский, синтезирует аудио и возвращает путь к файлу .wav"""
     translation = translator(text_ru, src_lang="rus_Cyrl", tgt_lang="kaz_Cyrl")