Update app.py

app.py

@@ -10,6 +10,10 @@ from utils.suggestions import (
     get_job_listings,
 )
 from utils.visualizer import generate_timeline
+from PIL import Image
+import matplotlib.pyplot as plt
+from fpdf import FPDF
+import tempfile
 import base64
 import io
 import datetime
@@ -63,60 +67,8 @@ if uploaded_file is not None:
     fig = generate_timeline(cv_type, education_level, score)
     st.pyplot(fig)
 
-from PIL import Image
-
-def generate_pdf_report():
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
 
-    def write_line(text):
-        pdf.multi_cell(0, 10, text)
 
-    write_line(f"CV Analysis Report - {datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}\n")
-    write_line(f"CV Type: {cv_type}")
-    write_line(f"Education Level: {education_level}")
-    write_line(f"Overall CV Score: {score} / 100\n")
-
-    write_line("Score Breakdown:")
-    for k, v in score_breakdown.items():
-        write_line(f"- {k}: {v}")
-    write_line("")
-
-    write_line("Certification Suggestions:")
-    if isinstance(certs, list):
-        for c in certs:
-            write_line(f"- {c}")
-    else:
-        write_line(str(certs))
-    write_line("")
-
-    write_line("Higher Education Suggestions:")
-    if isinstance(edu, list):
-        for e in edu:
-            write_line(f"- {e}")
-    else:
-        write_line(str(edu))
-    write_line("")
-
-    write_line("Visa Recommendations:")
-    if isinstance(visa, list):
-        for v in visa:
-            write_line(f"- {v}")
-    else:
-        write_line(str(visa))
-    write_line("")
-
-    write_line("Career Advice:")
-    write_line(str(advice))
-    write_line("")
-
-    write_line("Job Listings:")
-    if isinstance(jobs, list):
-        for job in jobs:
-            write_line(f"- {job}")
-    else:
-        write_line(str(jobs))
 
     # --- Add Roadmap Chart Image ---
     pdf.add_page()
@@ -138,3 +90,48 @@ def generate_pdf_report():
     pdf.output(pdf_buffer)
     pdf_buffer.seek(0)
     return pdf_buffer
+
+def generate_pdf_report(cv_type, education_level, score, score_breakdown, certs, edu, visa, advice, jobs, fig):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+
+    pdf.cell(200, 10, txt="CV Analysis Report", ln=True, align="C")
+    pdf.ln(10)
+
+    pdf.cell(200, 10, txt=f"CV Type: {cv_type}", ln=True)
+    pdf.cell(200, 10, txt=f"Education Level: {education_level}", ln=True)
+    pdf.cell(200, 10, txt=f"CV Score: {score}/100", ln=True)
+    pdf.ln(5)
+
+    pdf.set_font("Arial", style="B", size=12)
+    pdf.cell(200, 10, txt="Score Breakdown:", ln=True)
+    pdf.set_font("Arial", size=12)
+    for k, v in score_breakdown.items():
+        pdf.cell(200, 10, txt=f"- {k}: {v}", ln=True)
+    pdf.ln(5)
+
+    def add_list_to_pdf(title, items):
+        pdf.set_font("Arial", style="B", size=12)
+        pdf.cell(200, 10, txt=title, ln=True)
+        pdf.set_font("Arial", size=12)
+        if isinstance(items, list):
+            for item in items:
+                pdf.multi_cell(0, 10, f"- {item}")
+        else:
+            pdf.multi_cell(0, 10, str(items))
+        pdf.ln(5)
+
+    add_list_to_pdf("Certification Suggestions:", certs)
+    add_list_to_pdf("Higher Education Suggestions:", edu)
+    add_list_to_pdf("Visa Recommendations:", visa)
+    add_list_to_pdf("Career Advice:", advice)
+    add_list_to_pdf("Job Listings:", jobs)
+
+    # Save the figure as an image and insert into PDF
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmpfile:
+        fig.savefig(tmpfile.name, bbox_inches='tight')
+        pdf.add_page()
+        pdf.image(tmpfile.name, x=10, y=20, w=pdf.w - 20)
+
+    return pdf