Update app.py

app.py

@@ -9,6 +9,7 @@ import scipy.io.wavfile as wavfile
 import traceback
 import random
 import re
+import json
 
 # =========================
 # ПАРАМЕТРЫ
@@ -46,51 +47,47 @@ except Exception as e:
 # ГЕНЕРАЦИЯ ВОПРОСА
 # =========================
 
+
+
 def generate_quiz(text: str):
     prompt = (
-        "Сгенерируй учебный вопрос по тексту и дай 1 правильный и 1 неправильный вариант ответа.\n"
-        "СТРОГО ИСПОЛЬЗУЙ ФОРМАТ:\n"
-        "QUESTION: ...\n"
-        "CORRECT: ...\n"
-        "WRONG: ...\n"
+        "Сгенерируй один учебный вопрос по тексту. "
+        "Ответ верни строго в формате JSON БЕЗ лишнего текста, без комментариев, только JSON.\n"
+        "{\n"
+        "  \"question\": \"...\",\n"
+        "  \"correct\": \"...\",\n"
+        "  \"wrong\": \"...\"\n"
+        "}\n"
         f"TEXT: {text}"
     )
 
     try:
-        out = qa_model(prompt, max_length=200)[0]["generated_text"]
+        out = qa_model(prompt, max_new_tokens=150)[0]["generated_text"].strip()
     except Exception as e:
         raise RuntimeError(f"Ошибка генерации вопроса: {e}")
 
-    text_out = out.replace("\r", "").strip()
-
-    # --- Regular expressions ---
-    q = re.search(r"QUESTION:\s*(.+)", text_out, re.IGNORECASE)
-    c = re.search(r"CORRECT:\s*(.+)", text_out, re.IGNORECASE)
-    w = re.search(r"WRONG:\s*(.+)", text_out, re.IGNORECASE)
-
-    question = q.group(1).strip() if q else ""
-    correct = c.group(1).strip() if c else ""
-    wrong = w.group(1).strip() if w else ""
+    # Иногда FLAN-T5 добавляет текст до/после JSON → пробуем вытащить JSON из строки
+    try:
+        json_str = out[out.index("{"): out.rindex("}") + 1]
+        data = json.loads(json_str)
+    except Exception:
+        raise ValueError(f"Модель вывела неправильный JSON:\n{out}")
 
-    # --- fallback ---
-    if not (question and correct and wrong):
-        lines = [l.strip() for l in text_out.split("\n") if l.strip()]
-        for l in lines:
-            if l.lower().startswith("question"):
-                question = l.split(":", 1)[1].strip()
-            elif l.lower().startswith("correct"):
-                correct = l.split(":", 1)[1].strip()
-            elif l.lower().startswith("wrong"):
-                wrong = l.split(":", 1)[1].strip()
+    # Проверяем данные
+    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}")
+        raise ValueError(f"Модель вернула некорректные поля:\n{data}")
 
     options = [correct, wrong]
     random.shuffle(options)
+
     return question, options, correct
 
 
+
 # =========================
 # АУДИО НА КАЗАХСКОМ
 # =========================