Update app.py

app.py

@@ -688,11 +688,6 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
     
     # Global state for questions
     questions_state = gr.State(value=[])
-    # Shared question display (visible across tabs so students don't switch panels)
-    questions_display = gr.HTML(
-        value="",
-        elem_classes="question-display"
-    )
     
     # Student name input
     with gr.Group():
@@ -707,7 +702,9 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
         return f"Welcome, {session.student_name}!"
     
     student_name_input.change(update_student_name, student_name_input, None)
-    
+    # Shared question display (placed below name and above the tabs to avoid large empty header)
+    questions_display = gr.HTML(value="", elem_classes="question-display")
+
     with gr.Tabs():
         # ===== TAB 1: GENERATE QUESTIONS =====
         with gr.Tab("1️⃣ Generate Questions"):
@@ -805,16 +802,23 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
         with gr.Tab("2️⃣ Answer Questions"):
             gr.Markdown("### Step 2: Answer the Questions")
             gr.Markdown("Write out your working and answers. You can use any of the three methods below.")
-            
+
+            # Two-column layout: left = questions, right = answer methods
             with gr.Row():
-                method_info = gr.Textbox(
-                    label="How to Submit Answers",
-                    value="Method 1: Draw/Write in canvas\nMethod 2: Upload photo of your written work\nMethod 3: Type your answers directly",
-                    interactive=False,
-                    lines=3
-                )
-            
-            with gr.Tabs():
+                with gr.Column(scale=6):
+                    gr.Markdown("**Questions (keep visible while answering)**")
+                    # show the same shared questions_display here
+                    q_left = questions_display
+                with gr.Column(scale=6):
+                    with gr.Row():
+                        method_info = gr.Textbox(
+                            label="How to Submit Answers",
+                            value="Method 1: Draw/Write in canvas\nMethod 2: Upload photo of your written work\nMethod 3: Type your answers directly",
+                            interactive=False,
+                            lines=3
+                        )
+                    
+                    with gr.Tabs():
                 # Method 1: Draw
                 with gr.Tab("✏️ Draw Answers"):
                     canvas = gr.Sketchpad(
@@ -883,11 +887,11 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                         elem_classes="answer-input"
                     )
             
-            # Combined submit button
-            with gr.Row():
-                submit_btn = gr.Button(" Submit for Correction", variant="primary", size="lg")
-            
-            submit_status = gr.Textbox(
+                    # Combined submit button
+                    with gr.Row():
+                        submit_btn = gr.Button(" Submit for Correction", variant="primary", size="lg")
+                    
+                    submit_status = gr.Textbox(
                 label="Submission Status",
                 interactive=False,
                 lines=2