Update app.py

app.py

@@ -6,8 +6,9 @@ from sentence_transformers import SentenceTransformer
 import faiss
 from groq import Groq
 from reportlab.lib.pagesizes import A4
-from reportlab.lib.styles import getSampleStyleSheet
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
+from reportlab.lib import colors
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, ListFlowable, ListItem
 import io
 
 # -----------------------------
@@ -15,6 +16,7 @@ import io
 # -----------------------------
 REMOTEOK_URL = "https://remoteok.com/api"
 EMBED_MODEL = "BAAI/bge-small-en-v1.5"
+AI_MODEL = "openai/gpt-oss-120b"   # Groq model
 
 # Load embedding model
 @st.cache_resource
@@ -80,10 +82,6 @@ def generate_resume(resume_text, job):
     and this job description:\n{job['description']}\n
     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]
@@ -106,23 +104,87 @@ def generate_resume(resume_text, job):
     """
 
     chat_completion = groq_client.chat.completions.create(
-        model="openai/gpt-oss-120b",
+        model=AI_MODEL,
         messages=[{"role": "user", "content": prompt}],
         temperature=0.7,
     )
     return chat_completion.choices[0].message.content
 
-def build_pdf(resume_text):
+def generate_cover_letter(resume_text, job, name, email, phone):
+    prompt = f"""
+    You are an AI career assistant.
+    Given this resume:\n{resume_text}\n
+    and this job description:\n{job['description']}\n
+    Generate a professional, one-page cover letter tailored to this role.
+    Format it like this:
+
+    Dear Hiring Manager,
+
+    [Intro paragraph: Show enthusiasm and alignment with company/role]
+    [Body paragraph: Highlight 2-3 most relevant skills/experiences from resume]
+    [Closing paragraph: Express eagerness and thank them]
+
+    Sincerely,
+    {name}
+    {email} | {phone}
+    """
+
+    chat_completion = groq_client.chat.completions.create(
+        model=AI_MODEL,
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.7,
+    )
+    return chat_completion.choices[0].message.content
+
+def build_pdf(content, title="Resume", name="John Doe", email="john.doe@email.com", phone="+1 234 567 890"):
     buffer = io.BytesIO()
-    doc = SimpleDocTemplate(buffer, pagesize=A4)
+    doc = SimpleDocTemplate(buffer, pagesize=A4, leftMargin=40, rightMargin=40, topMargin=40, bottomMargin=40)
     styles = getSampleStyleSheet()
+
+    # Custom styles
+    header_style = ParagraphStyle("Header", parent=styles["Heading1"], fontSize=18, spaceAfter=6, textColor=colors.HexColor("#2C3E50"), alignment=1)
+    contact_style = ParagraphStyle("Contact", parent=styles["Normal"], fontSize=11, textColor=colors.HexColor("#566573"), alignment=1)
+    section_style = ParagraphStyle("Section", parent=styles["Heading2"], fontSize=13, spaceBefore=15, spaceAfter=8, textColor=colors.HexColor("#1B2631"))
+    normal_style = ParagraphStyle("Normal", parent=styles["Normal"], fontSize=11, leading=15)
+    bullet_style = ParagraphStyle("Bullet", parent=styles["Normal"], fontSize=11, leading=15, leftIndent=20)
+
     story = []
 
-    for line in resume_text.split("\n"):
-        if line.strip() == "":
-            story.append(Spacer(1, 12))
-        else:
-            story.append(Paragraph(line, styles["Normal"]))
+    # ---- HEADER ----
+    story.append(Paragraph(name, header_style))
+    story.append(Paragraph(f"{email} | {phone}", contact_style))
+    story.append(Spacer(1, 12))
+
+    # ---- BODY ----
+    if title == "Resume":
+        sections = content.split("**")
+        for sec in sections:
+            sec = sec.strip()
+            if not sec:
+                continue
+
+            if sec.lower().startswith("summary"):
+                story.append(Paragraph("Summary", section_style))
+            elif sec.lower().startswith("skills"):
+                story.append(Paragraph("Skills", section_style))
+            elif sec.lower().startswith("experience"):
+                story.append(Paragraph("Experience", section_style))
+            elif sec.lower().startswith("education"):
+                story.append(Paragraph("Education", section_style))
+            else:
+                if sec.startswith("- "):
+                    bullets = [s.strip("- ").strip() for s in sec.split("\n") if s.strip()]
+                    bullet_list = ListFlowable([ListItem(Paragraph(b, bullet_style)) for b in bullets], bulletType="bullet")
+                    story.append(bullet_list)
+                else:
+                    story.append(Paragraph(sec, normal_style))
+            story.append(Spacer(1, 8))
+    else:
+        # Treat as cover letter: keep paragraphs
+        for line in content.split("\n"):
+            if line.strip():
+                story.append(Paragraph(line.strip(), normal_style))
+                story.append(Spacer(1, 10))
 
     doc.build(story)
     buffer.seek(0)
@@ -133,11 +195,17 @@ def build_pdf(resume_text):
 # -----------------------------
 st.title("MATCHHIVE - AI Job Matcher")
 
+# Upload resume
 resume_file = st.file_uploader("Upload your resume (PDF or DOCX)", type=["pdf", "docx"])
 if resume_file:
     resume_text = extract_text_from_resume(resume_file)
 
     if resume_text.strip():
+        st.subheader("Contact Information")
+        name = st.text_input("Full Name", "John Doe")
+        email = st.text_input("Email", "john.doe@email.com")
+        phone = st.text_input("Phone", "+1 234 567 890")
+
         st.subheader("Fetching jobs...")
         jobs = fetch_jobs()
 
@@ -149,17 +217,28 @@ if resume_file:
                         f"[View Job Posting]({job['url']})  \n"
                         f"**Match Score:** {score:.2f}")
 
-            if st.button(f"Generate Resume for {job['position']}", key=job['id']):
-                tailored_resume = generate_resume(resume_text, job)
-
-                # 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",
-                )
+            col1, col2 = st.columns(2)
+
+            with col1:
+                if st.button(f"Generate Resume for {job['position']}", key=f"resume_{job['id']}"):
+                    tailored_resume = generate_resume(resume_text, job)
+                    edited_resume = st.text_area("Tailored Resume", tailored_resume, height=300)
+                    pdf_buffer = build_pdf(edited_resume, title="Resume", name=name, email=email, phone=phone)
+                    st.download_button(
+                        label="📥 Download Resume (PDF)",
+                        data=pdf_buffer,
+                        file_name="tailored_resume.pdf",
+                        mime="application/pdf",
+                    )
+
+            with col2:
+                if st.button(f"Generate Cover Letter for {job['position']}", key=f"cl_{job['id']}"):
+                    tailored_cl = generate_cover_letter(resume_text, job, name, email, phone)
+                    edited_cl = st.text_area("Cover Letter", tailored_cl, height=300)
+                    pdf_buffer = build_pdf(edited_cl, title="Cover Letter", name=name, email=email, phone=phone)
+                    st.download_button(
+                        label="📥 Download Cover Letter (PDF)",
+                        data=pdf_buffer,
+                        file_name="cover_letter.pdf",
+                        mime="application/pdf",
+                    )