File size: 2,413 Bytes
7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 65cb957 7dcc5b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
# app/utils/parser.py
import re
from typing import List, Dict
def parse_questions(raw_text: str) -> List[Dict]:
"""
Parse a model-generated block into list of question dicts with keys:
number, question, options [A,B,C,D], answer, explanation, detailed_solution (optional)
"""
if not raw_text:
return []
text = raw_text.replace("\r\n", "\n").strip()
# Try to separate Detailed Solutions section if present
sol_section = ""
m = re.search(r'(?:Detailed Solutions:|Solutions:)\s*([\s\S]*)$', text, re.IGNORECASE)
if m:
sol_section = m.group(1).strip()
main_text = text[:m.start(0)].strip()
else:
main_text = text
parts = re.split(r'\n(?=\d+\.)', main_text)
questions = []
for idx, part in enumerate(parts):
part = part.strip()
if not part:
continue
q_text = re.split(r'\nA\)', part, maxsplit=1)[0]
q_text = re.sub(r'^\s*\d+\.\s*', '', q_text).strip()
opts = []
for letter in ['A','B','C','D']:
m_opt = re.search(rf'{letter}\)\s*(.*?)(?=\n[A-D]\)|\nAnswer:|\nExplanation:|$)', part, re.DOTALL)
opts.append(m_opt.group(1).strip() if m_opt else "")
ans_m = re.search(r'Answer:\s*([A-D])', part, re.IGNORECASE)
answer = ans_m.group(1).upper() if ans_m else ""
expl_m = re.search(r'Explanation:\s*([\s\S]*?)(?=$)', part, re.IGNORECASE)
explanation = expl_m.group(1).strip() if expl_m else ""
questions.append({
"number": idx + 1,
"question": q_text,
"options": opts,
"answer": answer,
"explanation": explanation,
"detailed_solution": ""
})
# Map detailed solutions if present
if sol_section:
sol_parts = re.split(r'\n(?=\d+\.)', sol_section)
sol_list = []
for sp in sol_parts:
sp = sp.strip()
if not sp:
continue
sp_text = re.sub(r'^\s*\d+\.\s*', '', sp).strip()
sol_list.append(sp_text)
for i, q in enumerate(questions):
if i < len(sol_list):
q['detailed_solution'] = sol_list[i]
else:
q['detailed_solution'] = q.get('explanation', '')
else:
for q in questions:
q['detailed_solution'] = q.get('explanation', '')
return questions
|