Update app.py

app.py

@@ -530,11 +530,8 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 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")
+                download_questions_btn = gr.DownloadButton("📥 Download Questions (PDF)")
                 copy_btn = gr.Button("📋 Copy Questions")
             
             # Generate questions handler
@@ -558,27 +555,25 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 outputs=[questions_output, questions_output, questions_state, status_output]
             )
             
-            # Download handler
+            # Download handler - returns file path for gr.DownloadButton
             def download_qns():
                 if not session.current_questions:
-                    return None, "⚠️ Generate questions first!"
+                    return None
                 try:
                     # Prefer PDF export; fall back to plain text if PDF library missing
                     pdf_path = download_questions_pdf(session.current_questions, session.current_topic, session.current_grade)
                     if pdf_path:
-                        abs_path = os.path.abspath(pdf_path)
-                        return abs_path, f"✅ Click the download button above to save: {os.path.basename(pdf_path)}"
+                        return pdf_path
                     # Fallback to text
                     filepath = download_questions_file(session.current_questions, session.current_topic, session.current_grade)
-                    abs_path = os.path.abspath(filepath)
-                    return abs_path, f"✅ Click the download button above to save: {os.path.basename(filepath)}"
+                    return filepath
                 except Exception as e:
-                    return None, f"❌ Error: {str(e)}"
+                    return None
             
             download_questions_btn.click(
                 fn=download_qns,
                 inputs=[],
-                outputs=[questions_file, status_output]
+                outputs=download_questions_btn
             )
         
         # ===== TAB 2: SUBMIT ANSWERS =====
@@ -697,9 +692,7 @@ 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")
+                download_feedback_btn = gr.DownloadButton("📥 Download Feedback (TXT)")
                 save_status = gr.Textbox(
                     label="Download Status",
                     interactive=False
@@ -729,21 +722,20 @@ with gr.Blocks(title="UNEB Exam Prep - Primary 6 & 7", theme=gr.themes.Soft(), c
                 outputs=feedback_output
             )
             
-            # Download feedback handler
+            # Download feedback handler - returns file path for gr.DownloadButton
             def download_feedback_handler():
                 if not session.last_feedback:
-                    return None, "⚠️ Get AI correction first!"
+                    return None
                 try:
                     filepath = download_feedback_file(session.last_feedback, session.current_topic, session.current_grade)
-                    abs_path = os.path.abspath(filepath)
-                    return abs_path, f"✅ Click the download button above to save: {os.path.basename(filepath)}"
+                    return filepath
                 except Exception as e:
-                    return None, f"❌ Error: {str(e)}"
+                    return None
             
             download_feedback_btn.click(
                 fn=download_feedback_handler,
                 inputs=[],
-                outputs=[feedback_file, save_status]
+                outputs=download_feedback_btn
             )
         
         # ===== TAB 4: START NEW SESSION =====