Update app.py

app.py

@@ -53,38 +53,59 @@ except Exception as e:
 # =========================
 def generate_quiz(text: str):
     """
-    Генерирует один вопрос и два варианта ответа (correct, wrong) на русском языке.
+    Генерирует один вопрос и два варианта ответа на основе текста.
+    Использует простой алгоритм выбора предложений из текста.
     """
-    prompt = (
-        "Сгенерируй один учебный вопрос по этому тексту и два варианта ответа (правильный и неправильный). "
-        "Верни в формате: Вопрос: ... Ответ1: ... Ответ2: ... Текст: " + text
-    )
     try:
-        out = qa_model(prompt, max_length=256)[0]["generated_text"]
-        import re
-        match = re.search(
-            r"Вопрос\s*[:\-]\s*(.*?)\s*Ответ1\s*[:\-]\s*(.*?)\s*Ответ2\s*[:\-]\s*(.*)", out, re.DOTALL | re.IGNORECASE
-        )
-        if not match:
-            lines = [l.strip() for l in out.splitlines() if l.strip()]
-            if len(lines) >= 3:
-                question, correct, wrong = lines[:3]
+        # Разбиваем текст на предложения
+        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 + "?"
             else:
-                parts = out.split(".")
-                question = parts[0] if len(parts) > 0 else "Вопрос"
-                correct = parts[1] if len(parts) > 1 else "Вариант 1"
-                wrong = parts[2] if len(parts) > 2 else "Вариант 2"
+                question = "Что сказано в тексте?"
+            
+            # Правильный ответ — первые несколько слов предложения
+            correct_answer = " ".join(words[:6]) + ("..." if len(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 "")
+            else:
+                wrong_answer = "Это не упоминается в тексте"
         else:
-            question, correct, wrong = match.groups()
-
-        question = question.strip()
-        correct = correct.strip()
-        wrong = wrong.strip()
-        options = [correct, wrong]
+            # Если только одно предложение — придумываем очевидную ложь
+            wrong_answer = "Это совершенно другая информация"
+        
+        # Перемешиваем варианты
+        options = [correct_answer, wrong_answer]
         random.shuffle(options)
-        return question, options, correct
+        
+        return question, options, correct_answer
+        
     except Exception as e:
-        raise ValueError(f"Ошибка генерации вопроса:\n{str(e)}")
+        raise ValueError(f"Ошибка генерации вопроса: {str(e)}")
 
 
 def synthesize_audio(text_ru: str):