Update app.py

app.py

@@ -49,30 +49,44 @@ except Exception as e:
 # Вспомогательные функции
 # =========================
 def generate_quiz(text: str):
-    """Генерирует один вопрос и два варианта (correct, wrong) на русском языке."""
+    """
+    Генерирует один вопрос и два варианта ответа (correct, wrong) на русском языке.
+    Парсит текст вместо строгого JSON.
+    """
     prompt = (
-        "Сгенерируй один учебный вопрос по этому тексту и дай 1 правильный и 1 неправильный вариант ответа. "
-        "Формат вывода JSON: {\"question\": \"...\", \"correct\": \"...\", \"wrong\": \"...\"}. Текст: " + text
+        "Сгенерируй один учебный вопрос по этому тексту и два варианта ответа (правильный и неправильный). "
+        "Верни в формате: Вопрос: ... Ответ1: ... Ответ2: ... Текст: " + text
     )
     try:
         out = qa_model(prompt, max_length=256)[0]["generated_text"]
-        # Пытаемся найти JSON в выводе модели
-        json_start = out.find("{")
-        json_end = out.rfind("}")
-        if json_start == -1 or json_end == -1:
-            raise ValueError(f"Модель вывела неподходящий формат:\n{out}")
-        import json
-        data = json.loads(out[json_start: json_end+1])
-        question = data.get("question", "").strip()
-        correct = data.get("correct", "").strip()
-        wrong = data.get("wrong", "").strip()
-        if not (question and correct and wrong):
-            raise ValueError(f"Неполные данные:\n{out}")
+        # Пробуем извлечь вопрос и два ответа через простые маркеры
+        import re
+        match = re.search(
+            r"Вопрос\s*[:\-]\s*(.*?)\s*Ответ1\s*[:\-]\s*(.*?)\s*Ответ2\s*[:\-]\s*(.*)", out, re.DOTALL | re.IGNORECASE
+        )
+        if not match:
+            # fallback: разделяем по строкам
+            lines = [l.strip() for l in out.splitlines() if l.strip()]
+            if len(lines) >= 3:
+                question, correct, wrong = lines[:3]
+            else:
+                # Если всё равно не получилось — берём первые 3 фразы
+                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"
+        else:
+            question, correct, wrong = match.groups()
+
+        question = question.strip()
+        correct = correct.strip()
+        wrong = wrong.strip()
         options = [correct, wrong]
         random.shuffle(options)
         return question, options, correct
     except Exception as e:
-        raise ValueError(f"Ошибка генерации вопроса:\n{str(e)}")
+        raise ValueError(f"Ошибка генерации вопроса:\n{str(e)}\nМодель вернула: {out}")
+
 
 
 def synthesize_audio(text_ru: str):