Update app.py

app.py

@@ -46,39 +46,28 @@ def parse_quiz_content(quiz_content):
     question_section = quiz_parts[0].strip()
     answer_key_section = quiz_parts[1].strip()
 
-    # --- Split into blocks by double newline ---
-    blocks = re.split(r'\n\n+', question_section) # Split by one or more double newlines
-
-    # --- Handle Potential Title (First Block) ---
-    if blocks:
-        first_block = blocks[0].strip()
-        if not re.match(r'^\d+\.\s', first_block): # Check if the first block DOES NOT start with a question number
-            # Assume the first block is the title and remove it
-            blocks = blocks[1:] # Slice to remove the first block (title)
-        else:
-            st.warning("Unexpected format: First block starts with a question number. Title might not be removed correctly.")
-
-
-    # --- Process Question Blocks ---
-    for block in blocks:
-        block = block.strip()
-        if not block:
-            continue
-
-        lines = block.split('\n')
-        # --- First line is question text (after removing bold markdown) ---
-        question_text = re.sub(r'\*\*|\*', '', lines[0]).strip()
-
-        options = {}
-        for line in lines[1:]:
-            line = line.strip()
-            if re.match(r'^[A-D]\.\s', line): # Options are like "A. Elephant"
+    question_lines = question_section.strip().split('\n')
+    current_question = None
+    for line in question_lines:
+        line = line.strip()
+        if not line:
+            continue # Skip empty lines
+
+        if re.match(r'^\d+\.\s', line): # Start of a new question
+            if current_question: # Save previous question if exists
+                questions.append(current_question)
+            current_question = {'options': {}} # Start new question
+            current_question['question'] = re.sub(r'\*\*|\*', '', line).strip() # Question text, remove bold
+        elif re.match(r'^[A-D]\)\s', line) or re.match(r'^[A-D]\.\s', line): # Option line (handle both ')' and '.' after letter)
+            if current_question:
                 option_letter = line[0]
-                option_text = line[2:].strip()
-                options[option_letter] = option_text
+                option_text = line[2:].strip() # Remove "A) " or "A. "
+                current_question['options'][option_letter] = option_text
+        elif re.match(r'^\*\*.*?\*\*$', line) or re.match(r'^\*.*?\*$', line) : # Check for title again, just in case after split. Remove bold title if found in lines.
+            pass # Ignore title lines within question section if any.
 
-        if question_text: # Only append if we have a question text
-            questions.append({'question': question_text, 'options': options})
+    if current_question: # Append the last question
+        questions.append(current_question)
 
 
     # --- Answer Key Parsing (No change needed) ---