Update app.py

app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from utils.visualizer import generate_timeline
 from utils.parser import parse_cv, extract_education_level, identify_cv_type
 from utils.cv_scoring import calculate_cv_score
 from utils.suggestions import (
@@ -11,16 +10,58 @@ from utils.suggestions import (
 )
 from utils.visualizer import generate_timeline
 from PIL import Image
-import matplotlib.pyplot as plt
 from fpdf import FPDF
+import matplotlib.pyplot as plt
 import tempfile
-import base64
 import io
-import datetime
-from fpdf import FPDF
-from matplotlib.backends.backend_agg import RendererAgg
-import matplotlib.pyplot as plt
 
+# ------------------ PDF Generation Function ------------------
+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
+
+# ------------------ Streamlit App ------------------
 st.title("Universal Smart CV Analyzer & Career Roadmap")
 
 uploaded_file = st.file_uploader("Upload your CV in PDF format", type=["pdf"])
@@ -67,84 +108,16 @@ if uploaded_file is not None:
     fig = generate_timeline(cv_type, education_level, score)
     st.pyplot(fig)
 
-     st.subheader("Download Report")
-     if st.button("Generate PDF Report"):
-         pdf = generate_pdf_report(cv_type, education_level, score, score_breakdown, certs, edu, visa, advice, jobs, fig)
-         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
-        pdf.output(tmp_pdf.name)
-        with open(tmp_pdf.name, "rb") as f:
-             st.download_button(
-                label="📄 Download CV Analysis Report (PDF)",
-                data=f,
-                file_name="cv_analysis_report.pdf",
-                mime="application/pdf"
-            )
-
-
-
-
-    # --- Add Roadmap Chart Image ---
-    pdf.add_page()
-    pdf.set_font("Arial", "B", 14)
-    pdf.cell(0, 10, "Career Roadmap Timeline", ln=True)
-
-    img_buf = io.BytesIO()
-    fig.savefig(img_buf, format="PNG", bbox_inches="tight")
-    img_buf.seek(0)
-
-    # Convert to PIL Image to get correct dimensions for PDF scaling
-    image = Image.open(img_buf)
-    width, height = image.size
-    aspect = height / width
-    pdf_width = 180  # Max width for PDF
-    pdf.image(img_buf, x=15, y=None, w=pdf_width, h=pdf_width * aspect)
-
-    pdf_buffer = io.BytesIO()
-    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
+    # --- PDF Download Button ---
+    st.subheader("Download Report")
+    if st.button("Generate PDF Report"):
+        pdf = generate_pdf_report(cv_type, education_level, score, score_breakdown, certs, edu, visa, advice, jobs, fig)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
+            pdf.output(tmp_pdf.name)
+            with open(tmp_pdf.name, "rb") as f:
+                st.download_button(
+                    label="📄 Download CV Analysis Report (PDF)",
+                    data=f,
+                    file_name="cv_analysis_report.pdf",
+                    mime="application/pdf"
+                )