Spaces:
Runtime error
Runtime error
File size: 5,081 Bytes
5cbb2cb bbd2d91 5cbb2cb bbd2d91 5cbb2cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
import streamlit as st
import tempfile
import os
from pathlib import Path
from PyPDF2 import PdfReader
from openai import OpenAI
# Import our CrewAI resume optimization module.
from crewai_resume_optimization import optimize_resume, generate_interview_questions, extract_job_description
# ---------------------------
# Helper Function: Extract text from PDF (for resume)
# ---------------------------
def extract_pdf_text(uploaded_file):
# Check file size (limit: 10MB)
uploaded_file.seek(0, os.SEEK_END)
if uploaded_file.tell() > 10 * 1024 * 1024:
st.error("File exceeds 10MB limit.")
return ""
uploaded_file.seek(0)
reader = PdfReader(uploaded_file)
text = ""
for page in reader.pages:
page_text = page.extract_text()
if page_text:
text += page_text + "\n"
return text
# ---------------------------
# Sidebar: File Upload & Job Input Mode
# ---------------------------
st.sidebar.title("Upload your resume and add a job description")
# Resume upload (PDF only)
uploaded_resume = st.sidebar.file_uploader("Upload your resume (PDF, max 10MB)", type="pdf")
# Job input mode: "Job Description" vs "Job URL"
job_input_mode = st.sidebar.radio("Select Job Input Mode", ("Job Description", "Job URL"))
job_input = ""
if job_input_mode == "Job Description":
job_input = st.sidebar.text_area("Paste the job description here:")
elif job_input_mode == "Job URL":
job_input = st.sidebar.text_input("Enter the job URL:")
# Button to extract job description if URL mode is selected.
if job_input_mode == "Job URL" and st.sidebar.button("Extract Job Description"):
if job_input:
st.session_state.job_description = extract_job_description(job_input)
st.sidebar.success("Job description extracted!")
else:
st.sidebar.error("Please enter a job URL.")
# For Job Description mode, use the pasted job description.
if job_input_mode == "Job Description":
st.session_state.job_description = job_input
# Button to trigger resume optimization.
if st.sidebar.button("Optimize Resume"):
st.session_state.optimize_trigger = True
# ---------------------------
# Session State Initialization
# ---------------------------
if "resume_text" not in st.session_state:
st.session_state.resume_text = None
if "job_description" not in st.session_state:
st.session_state.job_description = None
if "optimized_resume" not in st.session_state:
st.session_state.optimized_resume = None
if "interview_questions" not in st.session_state:
st.session_state.interview_questions = None
# ---------------------------
# Process Resume Upload
# ---------------------------
if uploaded_resume is not None:
st.session_state.resume_text = extract_pdf_text(uploaded_resume)
if st.session_state.resume_text:
st.sidebar.success("Resume uploaded and processed successfully!")
else:
st.sidebar.error("Failed to extract text from the resume.")
# ---------------------------
# Main Area: Display Results
# ---------------------------
st.title("Resume Optimizer")
if st.session_state.resume_text is None:
st.info("Please upload your resume from the sidebar to begin.")
else:
if st.session_state.job_description:
st.subheader("Job Description")
st.write(st.session_state.job_description)
else:
st.info("Please provide a job description (paste or extract from URL) from the sidebar.")
if st.session_state.get("optimize_trigger", False):
if st.session_state.resume_text and st.session_state.job_description:
with st.spinner("Optimizing resume..."):
st.session_state.optimized_resume = optimize_resume(st.session_state.resume_text, st.session_state.job_description)
with st.spinner("Generating interview questions..."):
st.session_state.interview_questions = generate_interview_questions(st.session_state.optimized_resume, st.session_state.job_description)
st.success("Resume optimized and interview questions generated!")
st.session_state.optimize_trigger = False
else:
st.error("Please ensure you have uploaded your resume and provided a job description.")
if st.session_state.optimized_resume:
st.header("Optimized Resume")
st.markdown(st.session_state.optimized_resume)
st.download_button("Download Optimized Resume (md)",
st.session_state.optimized_resume,
file_name="tailored_resume.md",
mime="text/markdown")
if st.session_state.interview_questions:
st.header("Interview Questions")
st.markdown(st.session_state.interview_questions)
st.download_button("Download Interview Questions (md)",
st.session_state.interview_questions,
file_name="interview_materials.md",
mime="text/markdown")
st.info("You can open the downloaded markdown (.md) files with Notepad or any text editor.")
|