Update app.py

app.py

@@ -558,7 +558,7 @@ css_styles = """
         background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
         padding: 30px;
         border-radius: 15px;
-        color: red;
+        color: white;
         margin-bottom: 20px;
     }
     
@@ -797,11 +797,40 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 
                 # Method 3: Type
                 with gr.Tab(" Type Answers"):
-                    gr.Markdown("Type your answers here, numbered Q1, Q2, etc.")
+                    gr.Markdown("""
+### 📝 Type Your Answers
+
+**Instructions:** Type each answer in the box below. Use one of these formats:
+
+- **Option A (Recommended):** Type answers separated by blank lines
+  ```
+  Answer to question 1 goes here
+  
+  Answer to question 2 goes here
+  
+  Answer to question 3 goes here
+  ```
+
+- **Option B:** Type with labels
+  ```
+  Answer 1: Your answer here
+  Answer 2: Your answer here
+  Answer 3: Your answer here
+  ```
+
+- **Option C:** Type with Q labels
+  ```
+  Q1: Your answer here
+  Q2: Your answer here
+  Q3: Your answer here
+  ```
+                    """)
+                    
                     typed_answers = gr.Textbox(
-                        label="Type your answers",
-                        lines=15,
-                        placeholder="Q1. [Your answer]\nQ2. [Your answer]\n..."
+                        label="Type Your Answers Here",
+                        lines=20,
+                        placeholder="Type your answers using any of the formats shown above...",
+                        elem_classes="answer-input"
                     )
             
             # Combined submit button
@@ -814,44 +843,92 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 lines=2
             )
             
-            # Submission handler
+            # Improved submission handler with better parsing
             def submit_answers(canvas_input, upload_input, typed_input):
                 if not session.current_questions:
                     return " Generate questions first in Step 1!"
                 
                 # Check which method was used
                 if canvas_input is not None:
-                    # Convert canvas to text representation
                     session.current_answers = {0: "[Canvas drawing - sent to AI for analysis]"}
                     return f" Canvas submission received with {len(session.current_questions)} questions. Processing..."
+                
                 elif upload_input is not None:
                     session.current_answers = {0: "[Uploaded image - sent to AI for analysis]"}
                     return f" Photo submission received with {len(session.current_questions)} questions. Processing..."
+                
                 elif typed_input and typed_input.strip():
-                    # Parse typed answers
+                    # Parse typed answers with improved logic
                     session.current_answers = {}
-                    lines = typed_input.split('\n')
+                    lines = typed_input.strip().split('\n')
+                    
+                    # First, try to detect the format
+                    # Format detection: Look for patterns like "Answer 1:", "Q1:", or just free text separated by blank lines
+                    answer_list = []
+                    current_answer = []
+                    answer_number = 0
+                    
                     for line in lines:
-                        if line.strip().startswith('Q') and '.' in line:
-                            try:
-                                parts = line.split('.', 1)
-                                q_num = int(parts[0].strip('Q').strip()) - 1
-                                answer_text = parts[1].strip() if len(parts) > 1 else ""
-                                session.current_answers[q_num] = answer_text
-                            except:
-                                pass
+                        line = line.strip()
+                        
+                        # Skip completely empty lines if we have content in current_answer
+                        if not line:
+                            if current_answer:
+                                answer_list.append('\n'.join(current_answer))
+                                current_answer = []
+                            continue
+                        
+                        # Check for labeled formats (Q1:, Answer 1:, etc.)
+                        is_labeled = False
+                        label_content = None
+                        
+                        # Check for "Q1:", "Q2:" pattern
+                        if ':' in line:
+                            parts = line.split(':', 1)
+                            label_part = parts[0].strip().upper()
+                            content_part = parts[1].strip() if len(parts) > 1 else ""
+                            
+                            # Matches "Q1", "ANSWER 1", "ANS 1", etc.
+                            if (label_part.startswith('Q') and label_part[1:].split()[0].isdigit()) or \
+                               ('ANSWER' in label_part) or \
+                               ('ANS' in label_part):
+                                is_labeled = True
+                                # Save previous answer if exists
+                                if current_answer:
+                                    answer_list.append('\n'.join(current_answer))
+                                    current_answer = []
+                                # Start new answer with content
+                                if content_part:
+                                    current_answer.append(content_part)
+                                answer_number += 1
+                                continue
+                        
+                        # If not labeled format, treat as continuation of current answer
+                        current_answer.append(line)
+                    
+                    # Don't forget the last answer
+                    if current_answer:
+                        answer_list.append('\n'.join(current_answer))
+                    
+                    # Map answers to question indices
+                    for i, answer_text in enumerate(answer_list):
+                        if answer_text.strip():  # Only include non-empty answers
+                            session.current_answers[i] = answer_text.strip()
                     
                     if session.current_answers:
-                        return f"✅ Received {len(session.current_answers)} typed answers. Processing for correction..."
+                        answered = len(session.current_answers)
+                        total = len(session.current_questions)
+                        return f"✅ Received {answered}/{total} answer(s). Processing for AI correction..."
                     else:
-                        return "⚠️ Could not parse your answers. Use format: Q1. [answer], Q2. [answer], etc."
+                        return "⚠️ Could not find any answers. Please type your answers using one of the suggested formats."
                 else:
-                    return "⚠️ Please provide answers using one of the three methods."
+                    return "⚠️ Please provide answers using one of the three methods (draw, upload, or type)."
             
             submit_btn.click(
                 fn=submit_answers,
                 inputs=[canvas, upload_image, typed_answers],
                 outputs=submit_status
+
             )
         
         # ===== TAB 3: AI CORRECTION =====