|
import streamlit as st |
|
from prompt import PROMPT_RESUME_ABOUT, PROMPT_ACCEPTANCE_PERCENTAGE, PROMPT_SKILL_IMPROVEMENTS |
|
from utils import * |
|
|
|
|
|
def main(): |
|
st.set_page_config( |
|
page_title="Resume (ATS) Review Using Gemini AI", |
|
page_icon="π", |
|
layout="wide" |
|
) |
|
|
|
|
|
st.title("Resume (ATS) Review Using Gemini AI π") |
|
|
|
col1, col2 = st.columns([1, 3]) |
|
|
|
with col1: |
|
uploaded_file = st.file_uploader("Please Upload your Resume", type=["pdf"]) |
|
|
|
if uploaded_file is None: |
|
st.warning("Please upload your image") |
|
else: |
|
st.success("PDF Uploaded Successfully") |
|
|
|
submit1 = st.button("Tell Me About the Resume") |
|
|
|
submit2 = st.button("How can I Improvise my Skills") |
|
|
|
submit3 = st.button("Percentage match") |
|
|
|
with col2: |
|
input_text = st.text_area("Job Description:", key= "input") |
|
with st.spinner('Wait for it...'): |
|
if submit1: |
|
if uploaded_file is not None: |
|
pdf_content = input_pdf_setup(uploaded_file) |
|
response = get_gemini_response(PROMPT_RESUME_ABOUT, pdf_content, input_text) |
|
st.subheader("Resume Evaluation:") |
|
st.write(response) |
|
elif submit2: |
|
if uploaded_file is not None: |
|
pdf_content = input_pdf_setup(uploaded_file) |
|
response = get_gemini_response(PROMPT_SKILL_IMPROVEMENTS, pdf_content, input_text) |
|
st.subheader("Skill Improvement Recommendations:") |
|
st.write(response) |
|
elif submit3: |
|
if uploaded_file is not None: |
|
pdf_content = input_pdf_setup(uploaded_file) |
|
response = get_gemini_response(PROMPT_ACCEPTANCE_PERCENTAGE, pdf_content, input_text) |
|
st.subheader("ATS System Evaluation:") |
|
st.write(response) |
|
st.success('Done!') |
|
|
|
if __name__ == "__main__": |
|
main() |
|
|