Spaces:
Runtime error
Runtime error
| from dotenv import load_dotenv | |
| load_dotenv() | |
| import base64 | |
| import streamlit as st | |
| import os | |
| import io | |
| import textwrap | |
| import tempfile | |
| from PIL import Image | |
| import fitz | |
| import google.generativeai as genai | |
| from prompts import ( | |
| input_prompt1, | |
| input_prompt2, | |
| input_prompt3, | |
| input_prompt4, | |
| input_prompt5, | |
| input_prompt6, | |
| input_prompt7, | |
| input_prompt8, | |
| ) | |
| genai.configure(api_key=os.getenv("GOOGLE_API_KEY")) | |
| def extract_field_from_description(description): | |
| return input_role | |
| def construct_prompt(template, field): | |
| return template.replace("[User-Specified Field]", field) | |
| def to_markdown(text): | |
| text = text.replace("•", " *") | |
| return textwrap.indent(text, "> ", predicate=lambda _: True) | |
| def get_gemini_response(input, pdf_content, prompt): | |
| model = genai.GenerativeModel("gemini-pro-vision") | |
| response = model.generate_content([input, pdf_content[0], prompt]) | |
| # return response.text | |
| return to_markdown(response.text) | |
| def input_pdf_setup(uploaded_file): | |
| if uploaded_file is not None: | |
| file_content = uploaded_file.getvalue() | |
| if file_content: | |
| try: | |
| doc = fitz.open(stream=file_content, filetype="pdf") | |
| if len(doc) > 0: | |
| page = doc.load_page(0) # first page | |
| pix = page.get_pixmap() | |
| img_byte_arr = pix.tobytes("png") # get bytes directly | |
| pdf_parts = [ | |
| { | |
| "mime_type": "image/png", | |
| "data": base64.b64encode(img_byte_arr).decode(), | |
| } | |
| ] | |
| return pdf_parts | |
| else: | |
| raise ValueError("PDF does not contain any pages.") | |
| except Exception as e: | |
| st.error(f"Error processing PDF: {e}") | |
| raise e | |
| finally: | |
| doc.close() | |
| else: | |
| st.error("Uploaded file is empty or invalid.") | |
| raise ValueError("Uploaded file is empty or invalid.") | |
| else: | |
| st.error("No file uploaded.") | |
| raise FileNotFoundError("No file uploaded") | |
| # Streamlit App | |
| st.set_page_config( | |
| page_title="ATS Resume Expert", | |
| page_icon="🔥", | |
| ) | |
| st.header("ATS Friendly Resume and Analysis") | |
| input_name = st.text_input("Enter your name: ", key="name", placeholder="John Doe") | |
| input_role = st.text_input( | |
| "Role you are applying for: ", key="role", placeholder="Software Engineer" | |
| ) | |
| input_text = st.text_area( | |
| "Job Description: ", key="input", placeholder="Paste the job description here..." | |
| ) | |
| uploaded_file = st.file_uploader("Upload your resume(PDF)", type=["pdf"]) | |
| if "input_text" not in st.session_state: | |
| st.session_state["input_text"] = input_text | |
| if "input_name" not in st.session_state: | |
| st.session_state["input_name"] = input_name | |
| if "input_role" not in st.session_state: | |
| st.session_state["input_role"] = input_role | |
| if "input_text" not in st.session_state: | |
| st.session_state["input_text"] = input_text | |
| if "uploaded_file" not in st.session_state: | |
| st.session_state["uploaded_file"] = uploaded_file | |
| if uploaded_file is not None: | |
| st.write("PDF uploaded successfully!") | |
| # Prompts here | |
| field = extract_field_from_description(input_text) | |
| dynamic_prompt1 = construct_prompt(input_prompt1, field) | |
| dynamic_prompt2 = construct_prompt(input_prompt2, field) | |
| dynamic_prompt3 = construct_prompt(input_prompt3, field) | |
| dynamic_prompt4 = construct_prompt(input_prompt4, field) | |
| dynamic_prompt5 = construct_prompt(input_prompt5, field) | |
| dynamic_prompt6 = construct_prompt(input_prompt6, field) | |
| dynamic_prompt7 = construct_prompt(input_prompt7, field) | |
| dynamic_prompt8 = construct_prompt(input_prompt8, field) | |
| # Check if both the job description and the PDF file are provided | |
| if input_text and uploaded_file: | |
| try: | |
| pdf_content = input_pdf_setup(uploaded_file) | |
| print(pdf_content) | |
| is_pdf_valid = True | |
| except Exception as e: | |
| st.error(f"Error processing the PDF file: {e}") | |
| is_pdf_valid = False | |
| if is_pdf_valid: | |
| # Process for each tab if PDF is valid | |
| field = extract_field_from_description(input_text) | |
| st.subheader("Name : " + input_name) | |
| st.subheader("Role : " + input_role) | |
| tab1, tab2, tab3, tab4, tab5, tab6, tab7, tab8 = st.tabs( | |
| [ | |
| "HR/Human Resume Analysis", | |
| "ATS/AI Resume Analysis", | |
| "Skills Review", | |
| "ATS Friendly Resume", | |
| "Cover Letter", | |
| "LinkedIn", | |
| "Interview Questions", | |
| "Similar Companies", | |
| ] | |
| ) | |
| with tab1: | |
| st.header("HR/Human - Detailed Resume Analysis") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt1, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab2: | |
| st.header("ATS/AI - Detailed Resume Analysis") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt2, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab3: | |
| st.header("Skills Recommended for the Role") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt3, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab4: | |
| st.header("ATS Friendly and optimized Resume") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt4, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab5: | |
| st.header("Cover Letter for your resume") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt5, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab6: | |
| st.header("LinkedIn Profile Recommendations") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt6, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab7: | |
| st.header("Interview Questions and Questions to ask the interviewer") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt7, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| with tab8: | |
| st.header("Similar companies you can apply for") | |
| if uploaded_file is not None: | |
| # pdf_content = input_pdf_setup(uploaded_file) | |
| response = get_gemini_response(dynamic_prompt8, pdf_content, input_text) | |
| st.write(response) | |
| else: | |
| st.write("Please upload the resume") | |
| else: | |
| st.write("Please provide the job description and upload the resume PDF.") | |