Update app.py

app.py

@@ -576,4 +576,26 @@ with gr.Blocks(title="LeadIB AI Grading (Final Flow — Verbose)") as demo:
         qpms_box = gr.Textbox(label="📑 QP+MS Transcript", lines=12)
         as_box = gr.Textbox(label="📝 AS Transcript", lines=12)
 
-    grading_output_box = gr.Textbox(label="🧾 Grading (Ma
+    grading_output_box = gr.Textbox(label="🧾 Grading (Markdown)", lines=20)
+    grading_pdf_file = gr.File(label="📥 Download Grading PDF")
+    imprint_pdf_file = gr.File(label="📥 Download Imprinted PDF (Optional)")
+
+    def run_pipeline(qp_file_obj, ms_file_obj, ans_file_obj, imprint_flag):
+        qp_path = qp_file_obj.name
+        ms_path = ms_file_obj.name
+        ans_path = ans_file_obj.name
+
+        qpms_text, as_text, grading_text, grading_pdf_path, imprinted_pdf_path = align_and_grade_pipeline(
+            qp_path, ms_path, ans_path, imprint=imprint_flag
+        )
+
+        return qpms_text or "", as_text or "", grading_text or "", grading_pdf_path, imprinted_pdf_path
+
+    run_button.click(
+        fn=run_pipeline,
+        inputs=[qp_file, ms_file, ans_file, imprint_toggle],
+        outputs=[qpms_box, as_box, grading_output_box, grading_pdf_file, imprint_pdf_file]
+    )
+
+if __name__ == "__main__":
+    demo.launch()