Update app.py

app.py

@@ -401,17 +401,19 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
             with gr.Row():
                 generate_btn = gr.Button("📝 Generate 20 Questions", variant="primary", size="lg")
             
-            questions_output = gr.Textbox(
-                label="Generated Questions",
-                lines=20,
+            questions_output = gr.Markdown(
+                value="",
                 elem_classes="question-display"
             )
-            
+
             status_output = gr.Textbox(
                 label="Status",
                 interactive=False,
                 lines=1
             )
+
+            # Hidden file component to provide a real downloadable file
+            questions_file = gr.File(visible=False, label="Questions File")
             
             with gr.Row():
                 download_questions_btn = gr.Button("📥 Download Questions")
@@ -428,7 +430,9 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 session.current_grade = grade
                 session.current_topic = topic
                 
-                return formatted_text, formatted_text, questions_list, f"✅ Generated {len(questions_list)} questions on {topic}"
+                # For Markdown display, keep spacing and simple formatting
+                md_text = "\n\n".join([f"**{i+1}.** {q.split('.',1)[1].strip() if '.' in q else q}" for i, q in enumerate(questions_list)])
+                return md_text, md_text, questions_list, f"✅ Generated {len(questions_list)} questions on {topic}"
             
             generate_btn.click(
                 fn=generate_and_display,
@@ -439,17 +443,18 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
             # Download handler
             def download_qns():
                 if not session.current_questions:
-                    return "⚠️ Generate questions first!"
+                    return None, "⚠️ Generate questions first!"
                 try:
                     filepath = download_questions_file(session.current_questions, session.current_topic, session.current_grade)
-                    return f"✅ Downloaded! File: {filepath}"
+                    # Return the file path for gr.File and a friendly status
+                    return filepath, f"✅ Downloaded! File: {os.path.abspath(filepath)}"
                 except Exception as e:
-                    return f"❌ Error: {str(e)}"
+                    return None, f"❌ Error: {str(e)}"
             
             download_questions_btn.click(
                 fn=download_qns,
                 inputs=[],
-                outputs=status_output
+                outputs=[questions_file, status_output]
             )
         
         # ===== TAB 2: SUBMIT ANSWERS =====
@@ -569,6 +574,8 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
             
             with gr.Row():
                 download_feedback_btn = gr.Button("📥 Download Feedback")
+                # Hidden file component so the feedback can be downloaded
+                feedback_file = gr.File(visible=False, label="Feedback File")
                 save_status = gr.Textbox(
                     label="Download Status",
                     interactive=False
@@ -601,17 +608,17 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
             # Download feedback handler
             def download_feedback_handler():
                 if not session.last_feedback:
-                    return "⚠️ Get AI correction first!"
+                    return None, "⚠️ Get AI correction first!"
                 try:
                     filepath = download_feedback_file(session.last_feedback, session.current_topic, session.current_grade)
-                    return f"✅ Downloaded! File: {filepath}"
+                    return filepath, f"✅ Downloaded! File: {os.path.abspath(filepath)}"
                 except Exception as e:
-                    return f"❌ Error: {str(e)}"
+                    return None, f"❌ Error: {str(e)}"
             
             download_feedback_btn.click(
                 fn=download_feedback_handler,
                 inputs=[],
-                outputs=save_status
+                outputs=[feedback_file, save_status]
             )
         
         # ===== TAB 4: START NEW SESSION =====