Update app.py

app.py

@@ -52,46 +52,53 @@ except Exception as e:
 # =========================
 
 def generate_quiz(text: str):
-    """Генерирует один вопрос и два варианта (correct, wrong) на русском языке."""
     prompt = (
-        "Сгенерируй один учебный вопрос по этому тексту и дай 1 правильный и 1 неправильный вариант ответа. "
-        "Формат вывода (разделять переносами строки): "
-        "QUESTION: ...\nCORRECT: ...\nWRONG: ...\nTEXT: " + text
+        "Сгенерируй один учебный вопрос по этому тексту и дай 1 правильный и 1 неправильный вариант.\n"
+        "Строго используй такой формат (каждая часть с новой строки):\n"
+        "QUESTION: ...\n"
+        "CORRECT: ...\n"
+        "WRONG: ...\n"
+        f"TEXT: {text}"
     )
+
     try:
         out = qa_model(prompt, max_length=200)[0]["generated_text"]
     except Exception as e:
         raise RuntimeError(f"Ошибка генерации вопроса: {e}")
 
-    question = ""
-    correct = ""
-    wrong = ""
+    # Унифицируем текст
+    data = out.replace("\r", "")
+    
+    # --- Пытаемся достать через регулярные выражения ---
+    q = re.search(r"QUESTION:\s*(.+)", data, re.IGNORECASE)
+    c = re.search(r"CORRECT:\s*(.+)", data, re.IGNORECASE)
+    w = re.search(r"WRONG:\s*(.+)", data, re.IGNORECASE)
 
-    # ИСПРАВЛЕНО: корректное split('\n')
-    for line in out.split('\n'):
-        if line.upper().startswith("QUESTION:"):
-            question = line.split(':', 1)[1].strip()
-        elif line.upper().startswith("CORRECT:"):
-            correct = line.split(':', 1)[1].strip()
-        elif line.upper().startswith("WRONG:"):
-            wrong = line.split(':', 1)[1].strip()
+    question = q.group(1).strip() if q else ""
+    correct = c.group(1).strip() if c else ""
+    wrong = w.group(1).strip() if w else ""
 
+    # --- Если пусто — fallback парсер ---
     if not (question and correct and wrong):
-        parts = out.split('CORRECT:')
-        if len(parts) > 1:
-            qpart = parts[0]
-            question = qpart.replace('QUESTION:', '').strip()
-            rest = parts[1]
-            if 'WRONG:' in rest:
-                correct, wrong = rest.split('WRONG:', 1)
-                correct = correct.strip()
-                wrong = wrong.strip()
-
+        lines = [ln.strip() for ln in data.split('\n') if ln.strip()]
+        for ln in lines:
+            if ln.lower().startswith("question"):
+                question = ln.split(":", 1)[1].strip()
+            elif ln.lower().startswith("correct"):
+                correct = ln.split(":", 1)[1].strip()
+            elif ln.lower().startswith("wrong"):
+                wrong = ln.split(":", 1)[1].strip()
+
+    # --- Если всё ещё пусто — ошибка ---
     if not (question and correct and wrong):
-        raise ValueError("Модель не смогла корректно сгенерировать вопрос/варианты")
+        raise ValueError(
+            f"Модель вывела неподходящий формат:\n---\n{out}\n---"
+        )
 
+    # Случайно перемешать варианты
     options = [correct, wrong]
     random.shuffle(options)
+
     return question, options, correct