Update app.py

app.py

@@ -8,6 +8,7 @@ import base64
 import torch
 from fpdf import FPDF
 import os
+import tempfile
 import glob
 
 # Preload models
@@ -65,9 +66,9 @@ def generate_pdf_report(question, answer, score, score_explanation, score_chart,
     pdf = FPDF()
     pdf.add_page()
 
-    # Find a system font
+    # Add the Unicode font (e.g., DejaVuSans)
     font_path = find_system_font()
-    pdf.add_font("SystemFont", "", font_path)  # Use the found system font
+    pdf.add_font("SystemFont", "", font_path)
     pdf.set_font("SystemFont", size=12)
 
     pdf.multi_cell(0, 10, f"Question: {question}")
@@ -92,15 +93,22 @@ def generate_pdf_report(question, answer, score, score_explanation, score_chart,
     pdf.multi_cell(0, 10, highlighted_context)
     pdf.ln()
 
-    # Add score chart image to PDF
+    # Decode and save the score chart image to a temporary file
     score_chart_image = io.BytesIO(base64.b64decode(score_chart))
-    pdf.image(score_chart_image, x=10, y=pdf.get_y(), w=100)
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmpfile:
+        tmpfile.write(score_chart_image.read())
+        tmpfile.flush()
+        tmpfile.close()
+        pdf.image(tmpfile.name, x=10, y=pdf.get_y(), w=100)
 
     # Save PDF to memory
     pdf_output = io.BytesIO()
     pdf.output(pdf_output)
     pdf_output.seek(0)
 
+    # Clean up temporary file
+    os.remove(tmpfile.name)
+
     return pdf_output
 
 def answer_question(model_name, file, question, status):