Update app.py

app.py

@@ -1,8 +1,12 @@
 import os
 import gradio as gr
 import google.generativeai as genai
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import cm
 
-# Configure Gemini API (key must be set in Hugging Face Space secrets)
+# Configure Gemini API
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 # ---------- PROMPTS ----------
@@ -40,13 +44,10 @@ def transcribe(ans_file):
     yield "⏳ Processing transcription..."
     try:
         ans_uploaded = genai.upload_file(path=ans_file.name, display_name="Answer Sheet")
-
         model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
         resp = model.generate_content([TRANSCRIPTION_PROMPT, ans_uploaded])
-
         transcription = getattr(resp, "text", None) or resp.candidates[0].content.parts[0].text
-
-        yield transcription  # Markdown will render LaTeX properly
+        yield transcription
     except Exception as e:
         yield f"❌ Error during transcription: {e}"
 
@@ -59,15 +60,37 @@ def grade(qp_file, ms_file, transcription):
 
         model = genai.GenerativeModel("gemini-2.5-pro", generation_config={"temperature": 0})
         resp = model.generate_content([GRADING_PROMPT, qp_uploaded, ms_uploaded, transcription])
-
         grading = getattr(resp, "text", None) or resp.candidates[0].content.parts[0].text
 
-        yield grading
+        # Save grading report as PDF
+        pdf_path = "/tmp/grading_report.pdf"
+        doc = SimpleDocTemplate(pdf_path, pagesize=A4)
+        styles = getSampleStyleSheet()
+        elements = []
+
+        elements.append(Paragraph("📘 AI Teacher Assistant - Grading Report", styles["Title"]))
+        elements.append(Spacer(1, 0.5*cm))
+        for para in grading.split("\n"):
+            if para.strip():
+                elements.append(Paragraph(para, styles["Normal"]))
+                elements.append(Spacer(1, 0.2*cm))
+
+        doc.build(elements)
+
+        yield grading, pdf_path
     except Exception as e:
-        yield f"❌ Error during grading: {e}"
+        yield f"❌ Error during grading: {e}", None
 
 # ---------- GRADIO APP ----------
 with gr.Blocks(title="📘 AI Teacher Assistant") as demo:
+    # Inject MathJax for LaTeX rendering
+    gr.HTML("""
+    <script src="https://polyfill.io/v3/polyfill.min.js?features=es6"></script>
+    <script id="MathJax-script" async
+      src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js">
+    </script>
+    """)
+
     gr.Markdown("## 📘 AI Teacher Assistant\nUpload exam documents to transcribe and grade student answers step by step.")
 
     with gr.Row():
@@ -77,15 +100,16 @@ with gr.Blocks(title="📘 AI Teacher Assistant") as demo:
 
     # Step 1: Transcription
     transcribe_btn = gr.Button("Step 1: Transcribe Answer Sheet")
-    transcription_out = gr.Markdown(label="📄 Student Transcription", elem_id="transcription_box")
+    transcription_out = gr.Markdown(label="📄 Student Transcription")
 
     # Step 2: Grading
     grade_btn = gr.Button("Step 2: Grade the Student")
     grading_out = gr.Textbox(label="✅ Grading Report (Step-by-Step)", lines=20)
+    download_pdf = gr.File(label="⬇️ Download Grading Report (PDF)")
 
-    # Button Logic with streaming updates
+    # Button Logic
     transcribe_btn.click(fn=transcribe, inputs=[ans_file], outputs=[transcription_out])
-    grade_btn.click(fn=grade, inputs=[qp_file, ms_file, transcription_out], outputs=[grading_out])
+    grade_btn.click(fn=grade, inputs=[qp_file, ms_file, transcription_out], outputs=[grading_out, download_pdf])
 
 if __name__ == "__main__":
     demo.launch()