Update app.py

app.py

@@ -4,8 +4,11 @@ import pdfplumber
 import docx
 from sentence_transformers import SentenceTransformer
 import faiss
-import numpy as np
 from groq import Groq
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+import io
 
 # -----------------------------
 # CONFIG
@@ -72,11 +75,34 @@ def match_jobs(resume_text, jobs, top_k=5):
 
 def generate_resume(resume_text, job):
     prompt = f"""
-    You are an AI career assistant. 
+    You are an AI career assistant.
     Given this resume:\n{resume_text}\n
     and this job description:\n{job['description']}\n
-    Generate a tailored one-page resume that highlights relevant skills and experience.
-    Keep it concise and professional.
+    Generate a structured resume in this format:
+
+    Name & Contact
+    -----------------
+    [Fill with placeholder if not in resume]
+
+    Summary
+    -----------------
+    [2-3 line summary tailored for the job]
+
+    Skills
+    -----------------
+    - Skill 1
+    - Skill 2
+    - Skill 3
+
+    Experience
+    -----------------
+    Job Title | Company | Dates
+    • Achievement 1
+    • Achievement 2
+
+    Education
+    -----------------
+    Degree | Institution | Year
     """
 
     chat_completion = groq_client.chat.completions.create(
@@ -86,6 +112,21 @@ def generate_resume(resume_text, job):
     )
     return chat_completion.choices[0].message.content
 
+def build_pdf(resume_text):
+    buffer = io.BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=A4)
+    styles = getSampleStyleSheet()
+    story = []
+
+    for line in resume_text.split("\n"):
+        if line.strip() == "":
+            story.append(Spacer(1, 12))
+        else:
+            story.append(Paragraph(line, styles["Normal"]))
+
+    doc.build(story)
+    buffer.seek(0)
+    return buffer
 
 # -----------------------------
 # STREAMLIT UI
@@ -110,4 +151,15 @@ if resume_file:
 
             if st.button(f"Generate Resume for {job['position']}", key=job['id']):
                 tailored_resume = generate_resume(resume_text, job)
-                st.text_area("Tailored Resume", tailored_resume, height=300)
+
+                # Show editable preview
+                edited_resume = st.text_area("Tailored Resume", tailored_resume, height=300)
+
+                # Build PDF
+                pdf_buffer = build_pdf(edited_resume)
+                st.download_button(
+                    label="📥 Download Resume (PDF)",
+                    data=pdf_buffer,
+                    file_name="tailored_resume.pdf",
+                    mime="application/pdf",
+                )