Update app.py

app.py

@@ -46,87 +46,43 @@ 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, more robust splitting ---
-    blocks = re.split(r'\n\n+', question_section)
+    # --- Split into blocks by double newline ---
+    blocks = re.split(r'\n\n+', question_section) # Split by one or more double newlines
 
-    # --- DEBUGGING PRINT: Print the blocks after splitting ---
-    st.write("### Debugging: Question Blocks after Splitting:")
-    st.write(blocks)
-
-    # --- Handle Potential Title (First Block) - More robust title check ---
+    # --- Handle Potential Title (First Block) ---
     if blocks:
         first_block = blocks[0].strip()
-        if not re.search(r'Question\s+\d+:', first_block, re.IGNORECASE): # Use re.search for more flexible title detection
-            blocks = blocks[1:] # Remove title block if it doesn't look like "Question 1:", etc.
+        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 ---
-    question_number = 1 # Keep track of expected question number for better parsing
-
     for block in blocks:
         block = block.strip()
-        if not block or len(block) < 5: # Skip completely empty blocks or very short blocks (less than 5 chars)
-            st.write(f"Debugging: Skipping very short or empty block: '{block}'") # Debugging skip message
+        if not block:
             continue
 
         lines = block.split('\n')
-        question_text = ""
-        options = {}
-        parsing_question_text = True # Flag to indicate we are parsing question text
-        question_label_found = False # Flag to track if "Question N:" label was found
-        block_has_question_content = False # Flag to check if block has any question content
-
+        # --- First line is question text (after removing bold markdown) ---
+        question_text = re.sub(r'\*\*|\*', '', lines[0]).strip()
 
-        for line_index, line in enumerate(lines):
+        options = {}
+        for line in lines[1:]:
             line = line.strip()
-            if not line: # Skip empty lines within a question block
-                continue
-
-            if parsing_question_text:
-                if re.match(r'^Question\s+\d+:', line, re.IGNORECASE): # Check for "Question N:"
-                    question_label_found = True # Mark label as found
-                    # **MODIFIED QUESTION TEXT EXTRACTION:**
-                    if len(lines) > line_index + 1: # Check if there's a line AFTER "Question N:"
-                        question_text = lines[line_index + 1].strip() # Take the *NEXT* line as question text
-                        parsing_question_text = False # Stop parsing question text, move to options
-                        block_has_question_content = True # Mark that we found question content
-                    else:
-                        st.warning(f"Warning: Question label '{line}' found, but no question text followed.")
-                        question_text = "" # No question text found after label
-                        parsing_question_text = False # Move to options parsing even if question text missing
-                # **MODIFIED - Moved this condition to be ELSE if question_label_found is False**
-                elif not question_label_found and not question_text: # Check if question_label NOT found AND question_text is still empty
-                    # If no "Question N:" label found yet in this block, and we are still parsing question text,
-                    # consider this line as question text (for the very first line if no label)
-                    question_text = line
-                    parsing_question_text = False # Move to options parsing
-                    block_has_question_content = True # Mark that we found question content
-
-
-            elif re.match(r'^[A-D]\)\s(.+)', line): # Parse options
-                match = re.match(r'^([A-D])\)\s(.+)', line)
-                if match:
-                    option_letter = match.group(1)
-                    option_text = match.group(2).strip()
-                    options[option_letter] = option_text
-                    block_has_question_content = True # Mark that we found option content
-
-        # --- DEBUGGING PRINT: Print question text and options for each block ---
-        st.write(f"#### Debugging: Block {question_number} - Parsed Data:")
-        st.write(f"- Question Text: '{question_text}'")
-        st.write(f"- Options: {options}")
-        st.write(f"- Block has question content: {block_has_question_content}") # Debugging flag
-
-        if question_text and block_has_question_content: # Only add question if we successfully extracted question text AND block has content
-            questions.append({'question': question_text, 'options': options})
-        else:
-            st.warning(f"Warning: Could not parse question text OR no question content found for block (potential issue in question {question_number}). Block content:\n{block}")
+            if re.match(r'^[A-D]\.\s', line): # Options are like "A. Elephant"
+                option_letter = line[0]
+                option_text = line[2:].strip()
+                options[option_letter] = option_text
 
-        question_number += 1 # Increment expected question number
+        if question_text: # Only append if we have a question text
+            questions.append({'question': question_text, 'options': options})
 
 
     # --- Answer Key Parsing (No change needed) ---
-    answer_lines = answer_key_section.strip('\n')
+    answer_lines = answer_key_section.strip().split('\n')
     for line in answer_lines:
         line = line.strip()
         match = re.match(r'(\d+)\.\s*([A-D])', line)
@@ -169,7 +125,7 @@ def display_question():
     if st.session_state.current_question_index < len(st.session_state.quiz_data):
         question_data = st.session_state.quiz_data[st.session_state.current_question_index]
         question_number = st.session_state.current_question_index + 1
-        st.markdown(f"**Question {question_number}:** {question_data['question']}") # Display question from parsed data
+        st.markdown(f"**Question {question_number}:** {question_data['question']}")
 
         options_list = [f"{key}. {value}" for key, value in question_data['options'].items()]
         user_choice = st.radio("Choose an answer:", options_list, key=f"q_{question_number}") # Unique key for radio buttons
@@ -217,38 +173,37 @@ if topic:
                 prompt = f"""
                 Generate a multiple-choice quiz on the topic of "{topic}".
                 The quiz should have 5 questions.
-                Label each question as "Question 1:", "Question 2:", etc.
-                Provide four plausible options for each question, labeled A), B), C), and D).
-                Clearly indicate the correct answer for each question at the end in a section called "Answer Key:".
-                Format the quiz for easy reading.
+                For each question, provide four plausible options labeled A, B, C, and D.
+                Clearly indicate the correct answer for each question at the end in a separate section called "Answer Key".
+                Format the quiz clearly for easy reading.
                 """
                 response = model.generate_content(prompt)
                 quiz_content = response.text
 
-                # --- DEBUGGING PRINTS UNCOMMENTED ---
+                # --- DEBUGGING PRINTS ---
                 st.write("### Raw Quiz Content from Gemini:")
                 st.code(quiz_content) # Display raw content in a code block for readability
 
                 parsed_quiz_data, answer_key = parse_quiz_content(quiz_content)
 
-                st.write("### Parsed Quiz Data:") # Added Parsed Data debug print back
+                st.write("### Parsed Quiz Data:")
                 st.write(parsed_quiz_data) # Display the parsed data structure
 
-                # --- END DEBUGGING PRINTS UNCOMMENTED ---
+                # --- END DEBUGGING PRINTS ---
 
 
-                if quiz_content: # Check if quiz content was generated
-                    if parsed_quiz_data: # Check if parsing was successful
+                if quiz_content: # Check if quiz_content was generated successfully (outer if)
+                    if parsed_quiz_data: # Check if parsing was successful (inner if)
                         st.session_state.quiz_data = parsed_quiz_data
                         st.session_state.current_question_index = 0
                         st.session_state.user_answers = []
                         st.session_state.quiz_completed = False
                         st.session_state.score = 0
                         st.success(f"Quiz on '{topic}' generated successfully! Let's begin.")
-                    else: # ELSE for parsed_quiz_data (parsing failed)
+                    else: # else associated with inner if parsed_quiz_data
                         st.error("Failed to parse quiz content. Please try generating again.")
                         st.session_state.quiz_data = None
-                else: # ELSE for quiz_content (quiz_content not generated - API issue)
+                else: # else associated with outer if quiz_content
                     st.error("Failed to generate quiz content. Please try again or check your API key.")
 
             except Exception as e: