Upload 13 files

app.py

@@ -1,277 +1,20 @@
-import re
+from src.ui import *
+from src.utils import is_chatbot_loaded
 
-import pdfkit
-import streamlit as st
 
-from src.chatbot.chatgpt import Chatgpt
-from src.utils import parse_pdf, build_html_resume
-
-section_examples = {'summary': 'I have passion for new tech',
-                    'workExperience': 'Tell about my ability to lead projects',
-                    'education': 'Describe my degree type in more details'}
-
-openai_key_info = 'https://platform.openai.com/account/api-keys'
-
-
-def list_section(section_name, section_data):
-    description_key = 'description'
-
-    item_keys = list(section_data[0].keys())
-    item_keys.remove(description_key)
-    for item_id, section_item in enumerate(section_data):
-        cols = st.columns(len(item_keys))
-        for col, key in zip(cols, item_keys):
-            col.text_input(key, section_item[key], key=f'{section_name}_{item_id}_{key}')
-        st.text_area(description_key, section_item[description_key], key=f'{section_name}_{item_id}_{description_key}')
-
-        recruiter_subsection(section_name, section_example=section_examples[section_name], item_id=item_id)
-        st.markdown('***')
-
-
-def skills_section(section_name, skills_data):
-    num_columns = 3
-    for skills_row in range(0, len(skills_data), num_columns):
-        cols = st.columns([3, 1] * num_columns)
-        skills_row_names = skills_data[skills_row: skills_row + num_columns]
-        for item_id, skill in enumerate(skills_row_names):
-            skill_id = skills_row + item_id
-            cols[item_id * 2].text_input(' ', value=skill, key=f'{section_name}_{skill_id}', label_visibility='hidden')
-            cols[item_id * 2 + 1].markdown('## ')
-            if cols[item_id * 2 + 1].button('x', key=f'{section_name}_{skill_id}_remove_skill'):
-                _remove_skill(skill_id, skills_data)
-
-    skill_subsection(section_name)
-    st.markdown('***')
-
-
-def _remove_skill(skill_id, skills_data):
-    del skills_data[skill_id]
-    st.experimental_rerun()
-
-
-def skill_subsection(section_name, item_id=0):
-    key = f'{section_name}_{item_id}_add_skill'
-    cols = st.columns([12, 1])
-    new_skill = cols[0].text_input("Add skill", key=key)
-    cols[1].markdown('##')
-    clicked = cols[1].button("\+")
-    if clicked and new_skill:
-        st.write(new_skill)
-        st.session_state['resume_data'][section_name].append(new_skill)
-        st.write(st.session_state['resume_data'][section_name])
-        st.experimental_rerun()
-
-
-def recruiter_subsection(section_name, section_example, item_id=0):
-    with st.container():
-        cols = st.columns([3, 10], gap='small')
-        cols[0].write('\n')
-        cols[0].write('\n')
-        button_clicked = cols[0].button("Auto Section Improve", key=f'{section_name}_{item_id}_improve_auto')
-        trigger_key = 'Add a special request'
-        user_request_template = f"{trigger_key} to the bot here... e.g. {section_example}."
-
-        user_request = cols[1].text_input("section_example", value=user_request_template,
-                                          key=f'{section_name}_{item_id}_improve_manual', label_visibility='hidden')
-        if button_clicked:
-            user_request = '' if trigger_key in user_request else user_request
-            section_key = get_item_key(section_name, item_id)
-            section_text = st.session_state[section_key]
-            new_section_text = st.session_state['chatbot'].improve_section(section_text, user_request)
-
-            update_resume_data(new_section_text, section_name, item_id)
-            st.experimental_rerun()
-
-
-def get_item_key(section_name, item_id=0):
-    section_key = ''
-    if section_name in ['workExperience', 'education']:
-        key = 'description'
-        section_key = f'{section_name}_{item_id}_{key}'
-    elif section_name == 'summary':
-        section_key = f'{section_name}'
-    return section_key
-
-
-def update_resume_data(text_input, section_name, item_id=0):
-    if section_name in ['workExperience', 'education']:
-        key = 'description'
-        st.session_state['resume_data'][section_name][item_id][key] = text_input
-    elif section_name == 'summary':
-        section_key = f'{section_name}'
-        st.session_state['resume_data'][section_key] = text_input
-
-
-def summary_section(section_name, summary_data):
-    st.text_area(section_name, summary_data, key=f'{section_name}', label_visibility='hidden')
-    recruiter_subsection(section_name, section_examples[section_name])
-
-
-def contact_info_section(section_name, info_data):
-    for key, value in info_data.items():
-        if value:
-            st.text_input(key.title(), value, key=f'{section_name}_{key}')
-    st.markdown('***')
-
-
-def header():
-    st.text_input('name', st.session_state.resume_data['name'], key="name")
-    st.text_input('title', st.session_state.resume_data['title'], key="title")
-
-
-def body():
-    section_dict = {'contactInfo': contact_info_section, 'summary': summary_section, 'workExperience': list_section,
-                    'education': list_section, 'skills': skills_section}
-    tabs_names = [key_to_tab_name(key) for key in section_dict.keys()]
-    tabs = st.tabs(tabs_names)
-    for tab, key in zip(tabs, section_dict):
-        section_func = section_dict[key]
-        with tab:
-            section_func(key, st.session_state['resume_data'][key])
-
-
-def key_to_tab_name(input_string):
-    return re.sub(r'([A-Z])', r' \1', input_string).strip().title()
-
-
-def sidebar():
-    with st.sidebar:
-        uploaded_file = st.file_uploader('Upload PDF Resume', type="PDF")
-        if uploaded_file and _is_new_file(uploaded_file) and is_chatbot_loaded():
-            _init_resume(uploaded_file)
-
-        if is_data_loaded() and is_chatbot_loaded():
-            st.button("Improve More", on_click=_improve_more)
-            st.download_button('Download PDF', file_name='out.pdf', mime="application/json", data=download_pdf())
-
-
-def download_pdf():
-    resume_data = format_resume_data()
-    html_resume = build_html_resume(resume_data)
-    options = {'page-size': 'A4', 'margin-top': '0.75in', 'margin-right': '0.75in', 'margin-bottom': '0.75in',
-               'margin-left': '0.75in', 'encoding': "UTF-8", 'no-outline': None}
-    return pdfkit.from_string(html_resume, options=options, css='src/css/main.css')
-
-
-def _improve_more():
-    print("Improving resume")
-    st.session_state['resume_data'] = st.session_state['chatbot'].improve_resume(st.session_state['resume_data'])
-
-
-def _init_chatbot():
-    cols = st.columns([6, 1, 1])
-    api_key = cols[0].text_input("Enter OpenAI API key")
-    cols[1].markdown("#")
-    api_submit = cols[1].button("Submit")
-
-    cols[2].markdown("#")
-    get_info = cols[2].button("Get key")
-    if get_info:
-        st.info(f"Get your key at: {openai_key_info}")
-    if api_submit:
-        if Chatgpt.validate_api(api_key):
-            st.session_state['chatbot'] = Chatgpt(api_key)
-            st.experimental_rerun()
-
-        else:
-            st.error("Not valid API key - try again...")
-
-
-def is_chatbot_loaded():
-    return st.session_state.get('chatbot')
-
-
-def _is_new_file(uploaded_file):
-    return uploaded_file.id != st.session_state.get('file_id', '')
-
-
-def _init_resume(uploaded_file):
-    resume_data = parse_pdf(uploaded_file)
-    st.session_state['resume_data'] = st.session_state['chatbot'].improve_resume(resume_data)
-    st.session_state['file_id'] = uploaded_file.id
-    st.experimental_rerun()
-
-
-def format_resume_data():
-    current_state = st.session_state
-    resume_data = {}
-    contact_info = {}
-    work_experience = []
-    education = []
-    skills = []
-
-    resume_data['name'] = current_state.get('name', '')
-    resume_data['title'] = current_state.get('title', '')
-
-    contact_info_keys = ['linkedin', 'github', 'email', 'address']
-    for key in contact_info_keys:
-        contact_info[key] = current_state.get(f'contactInfo_{key}', '')
-    resume_data['contactInfo'] = contact_info
-
-    resume_data['summary'] = current_state.get('summary', '')
-
-    work_experience_keys = ['workExperience_{}_title', 'workExperience_{}_company', 'workExperience_{}_dates',
-                            'workExperience_{}_description']
-    education_keys = ['education_{}_degree', 'education_{}_school', 'education_{}_dates', 'education_{}_description']
-
-    total_work_experience = count_entries(st.session_state, 'workExperience')
-    total_education = count_entries(st.session_state, 'education')
-
-    for i in range(total_work_experience):
-        work_experience.append(
-            {key.split('_')[2]: current_state.get(key.format(i), '') for key in work_experience_keys})
-
-    for i in range(total_education):
-        education.append({key.split('_')[2]: current_state.get(key.format(i), '') for key in education_keys})
-
-    resume_data['workExperience'] = work_experience
-    resume_data['education'] = education
-
-    total_skills = count_entries(st.session_state, 'skills')
-
-    for i in range(total_skills):
-        skill_key = f'skills_{i}'
-
-        skills.append(current_state.get(skill_key, ''))
-    resume_data['skills'] = skills
-
-    return resume_data
-
-
-def count_entries(input_dict, entry_type):
-    max_index = max([int(key.split("_")[1]) for key in input_dict.keys() if key.startswith(f"{entry_type}_")],
-                    default=0)
-    return max_index + 1
-
-
-def title():
-    st.title("ChatCV - AI Resume Builder")
-
-
-def upload_resume_header():
-    st.success("Upload PDF Resume - Let the magic begin...")
-
-
-def is_data_loaded():
-    return st.session_state.get('resume_data')
-
-
-def _main():
+def main():
     title()
     if is_chatbot_loaded():
         sidebar()
 
         if is_data_loaded():
-            header()
+            resume_header()
             body()
-
         else:
-            upload_resume_header()
+            user_info()
     else:
-        _init_chatbot()
+        init_chatbot()
 
 
 if __name__ == '__main__':
-    _main()
-
-    # bootstrap 4 collapse example
+    main()

src/chatbot/chatgpt.py

@@ -1,18 +1,20 @@
 import ast
 import json
+import logging
 import re
 from pathlib import Path
 from typing import Dict
-import logging
 
 import requests
 from revChatGPT.Official import Chatbot
 
-from .prompts import get_prompt, data_format
+from src.chatbot.prompts import get_prompt, data_format
 
 logging.basicConfig(filename='chatgpt.log', level=logging.INFO, format='%(asctime)s %(message)s',
                     datefmt='%m/%d/%Y %I:%M:%S %p')
 
+openai_key_info = 'https://platform.openai.com/account/api-keys'
+
 
 class Chatgpt:
     def __init__(self, api_key):
@@ -61,11 +63,13 @@ class Chatgpt:
 
     def _ask(self, chatgpt_input):
         logging.info("Asking chatbot for response")
-        response = self.chatbot.ask(chatgpt_input)
-        answer = response['choices'][0]['text']
-        logging.info("Received response from chatbot")
-        logging.info(f"Response: {answer}")
-
+        try:
+            response = self.chatbot.ask(chatgpt_input)
+            answer = response['choices'][0]['text']
+            logging.info("Received response from chatbot")
+            logging.info(f"Response: {answer}")
+        except Exception:
+            answer = ""
         return answer
 
     def parse_json_from_string(self, json_string):
@@ -113,9 +117,8 @@ class Chatgpt:
 
     def clean_section_response(self, input_string):
         try:
-            start = input_string.index('"')
-            end = input_string.rindex('"') + 1
-            input_string = input_string[start:end]
+            input_string = re.sub('^\W*"', "", input_string)
+            input_string = re.sub('"\W*$', "", input_string)
         except ValueError:
             pass
         input_string = self.remove_prefix(input_string)

src/chatbot/prompts.py

@@ -6,8 +6,10 @@ data_format = {'name': '', 'title': '',
                                   {'title': '', 'company': '', 'dates': '', 'description': ''}, ],
                'education': [{'degree': '', 'school': '', 'dates': '', 'description': ''}, ], 'skills': ['', '', '']}
 
-recruiter_prompt = 'You are a recruiter and a professional resume builder.'
-command_prompt = 'Re-write the input as professionally as possible, adding vital, valuable information and skills.'
+recruiter_prompt = 'You are a professional resume builder and a recruiter.\n'
+command_prompt = 'Re-write the input as professionally as possible, adding vital, valuable information and skills.\n' \
+                 'Enhance the input to showcase the relevant education, experience, and skills in a professional manner to effectively demonstrate value to potential employers.\n' \
+                 f'Do it for every value in your output {str(list(data_format))}.  '
 user_request_prompt = f'{prompt_placeholder}'
 
 output_commands_prompts = dict()
@@ -29,4 +31,4 @@ def get_prompt(input_data, user_request='', output_type='all'):
     template = '\n'.join(
         [recruiter_prompt, command_prompt, user_request_prompt.replace(prompt_placeholder, user_request),
          input_prompt.replace(prompt_placeholder, input_data), output_commands_prompts[output_type], command_prompt])
-    return template
+    return template

src/css/main.css

@@ -24,6 +24,8 @@ html
 
 body
 {
+  	font-family:Open Sans, sans-serif;
+
     top: 0px;
 	-webkit-font-smoothing:antialiased;
 	font-family:Lora, serif;
@@ -44,7 +46,7 @@ body
 
 h1
 {
-	color:rgba(0,0,0,1);
+color: #145DA0;
 }
 
 .wrapper
@@ -57,6 +59,7 @@ h1
 
 .left
 {
+    	font-family:Open Sans, sans-serif;
 
 left: 0px;
 	height:100%;
@@ -68,11 +71,13 @@ left: 0px;
 
 .right
 {
+    	font-family:Open Sans, sans-serif;
+
 right: 0px;
 position: absolute;
 	text-align:center;
-	background-color:rgba(200,0,0,.025);
-	border-left:1px solid rgba(230,0,0,.05);
+	background-color:rgba(0, 0, 0,0.1);
+	border-left:1px solid rgba(0,0,0,.1);
 
 	height:100%;
 	width:30%;
@@ -81,11 +86,35 @@ position: absolute;
 
 }
 
+.right h1
+{
+
+color: #145DA0;
+}
+
+.right h2
+{
+
+color: #145DA0;
+}
+
+
+
+.right p
+{
+color: #145DA0;
+}
+
+.left p
+{
+color: #145DA0;
+}
 
 
 .name-hero
 {
 	background:rgba(0,0,0,.001);
+	margin:0px 1px 0 0;
 
 	width:85%;
 }
@@ -108,9 +137,34 @@ position: absolute;
 .name-hero h3
 {
 	font-family:Open Sans, sans-serif;
+
 	font-size:1.5em;
 	text-align:center;
 	margin:0px auto;
+   padding-top: -50 px;
+}
+
+
+.title h3
+{
+	font-family:Open Sans, sans-serif;
+
+	font-size:1.5em;
+	text-align:center;
+	margin:0px auto;
+   color:rgba(128, 128, 128,0.5);
+   padding-top: -50 px;
+}
+
+.name-hero title
+{
+	font-family:Open Sans, sans-serif;
+
+	font-size:1.5em;
+	text-align:center;
+	margin:0px auto;
+   color:rgba(128, 128, 128,0.5);
+   padding-top: -0 px;
 }
 
 .name-hero h1 em
@@ -122,6 +176,7 @@ position: absolute;
 
 .name-hero p
 {
+    font-family:Open Sans, sans-serif;
 	color:rgba(0,0,0,1);
 	font-size:.75em;
 	line-height:1.5;
@@ -133,10 +188,13 @@ position: absolute;
 {
     font-size:1.5em;
 	margin:0 auto;
+	color: #145DA0;
 
 	width:85%;
 }
 
+
+
 .inner
 {
 	margin:0 auto;
@@ -156,7 +214,7 @@ position: absolute;
 
 .inner p em
 {
-	color:rgba(0,0,0,1);
+color: #145DA0;
 	font-style:normal;
 }
 
@@ -175,6 +233,7 @@ ul
 
 .skill-set
 {
+  	font-family:Open Sans, sans-serif;
 
 	color:rgba(0,0,0,1);
 	list-style:none;

src/data_handler.py

@@ -0,0 +1,89 @@
+import pdfkit
+import streamlit as st
+
+from src.exceptions import PDFSizeException
+from src.pdf_handler import build_html_resume, parse_pdf
+from src.utils import count_entries
+
+
+def init_resume(uploaded_file):
+    resume_data, num_pages = parse_pdf(uploaded_file)
+    if num_pages > 3:
+        raise PDFSizeException
+    st.session_state['file_id'] = uploaded_file.id
+    return resume_data
+
+
+def update_resume_data(text_input, section_name, item_id=0):
+    if section_name in ['workExperience', 'education']:
+        key = 'description'
+        st.session_state['resume_data'][section_name][item_id][key] = text_input
+    elif section_name == 'summary':
+        section_key = f'{section_name}'
+        st.session_state['resume_data'][section_key] = text_input
+
+
+@st.cache
+def download_pdf():
+    if st.session_state.get('name'):
+        resume_data = format_resume_data()
+    else:
+        resume_data = st.session_state['resume_data']
+    html_resume = build_html_resume(resume_data)
+    options = {'page-size': 'A4', 'margin-top': '0.75in', 'margin-right': '0.75in', 'margin-bottom': '0.75in',
+               'margin-left': '0.75in', 'encoding': "UTF-8", 'no-outline': None}
+    return pdfkit.from_string(html_resume, options=options, css='src/css/main.css')
+
+
+def improve_resume(resume_data=None):
+    if resume_data is not None:
+        st.session_state['resume_data'] = st.session_state['chatbot'].improve_resume(resume_data)
+    else:
+        st.session_state['resume_data'] = st.session_state['chatbot'].improve_resume(st.session_state['resume_data'])
+
+
+def format_resume_data():
+    current_state = st.session_state
+    resume_data = {}
+    contact_info = {}
+    work_experience = []
+    education = []
+    skills = []
+
+    resume_data['name'] = current_state.get('name', '')
+    resume_data['title'] = current_state.get('title', '')
+
+    contact_info_keys = ['linkedin', 'github', 'email', 'address']
+    for key in contact_info_keys:
+        contact_info[key] = current_state.get(f'contactInfo_{key}', '')
+
+    resume_data['contactInfo'] = contact_info
+
+    resume_data['summary'] = current_state.get('summary', '')
+
+    work_experience_keys = ['workExperience_{}_title', 'workExperience_{}_company', 'workExperience_{}_dates',
+                            'workExperience_{}_description']
+    education_keys = ['education_{}_degree', 'education_{}_school', 'education_{}_dates', 'education_{}_description']
+
+    total_work_experience = count_entries(st.session_state, 'workExperience')
+    total_education = count_entries(st.session_state, 'education')
+
+    for i in range(total_work_experience):
+        work_experience.append(
+            {key.split('_')[2]: current_state.get(key.format(i), '') for key in work_experience_keys})
+
+    for i in range(total_education):
+        education.append({key.split('_')[2]: current_state.get(key.format(i), '') for key in education_keys})
+
+    resume_data['workExperience'] = work_experience
+    resume_data['education'] = education
+
+    total_skills = count_entries(st.session_state, 'skills')
+
+    for i in range(total_skills):
+        skill_key = f'skills_{i}'
+
+        skills.append(current_state.get(skill_key, ''))
+    resume_data['skills'] = skills
+
+    return resume_data

src/exceptions.py

@@ -0,0 +1,13 @@
+class PDFSizeException(Exception):
+    """Raised when the input value is less than 3"""
+    pass
+
+
+class ChatbotInitException(Exception):
+    """Raised when there's a problem with chabot init"""
+    pass
+
+
+class ChatbotAPIException(Exception):
+    """Raised when there's a problem with openai api"""
+    pass

src/pdf_handler.py

@@ -0,0 +1,39 @@
+import PyPDF2
+from jinja2 import FileSystemLoader, Environment
+
+
+def parse_pdf(pdf_file):
+    if pdf_file is isinstance(pdf_file, str):
+        with open(pdf_file, "rb") as file:
+            return _parse(file)
+    else:
+        return _parse(pdf_file)
+
+
+def _parse(file):
+    reader = PyPDF2.PdfReader(file)
+    pdf_text = []
+    num_pages = len(reader.pages)
+    # Iterate over each page
+    for page_number in range(num_pages):
+        # Get the current page
+        page = reader.pages[page_number]
+
+        # Extract the text from the page
+        page_text = page.extract_text()
+
+        pdf_text.append(page_text)
+    pdf_text = '\n'.join(pdf_text)
+    return pdf_text, num_pages
+
+
+def build_html_resume(data):
+    env = Environment(loader=FileSystemLoader('src/templates'))
+    template = env.get_template('resume.html')
+    html_resume = template.render(data)
+    return html_resume
+
+
+def export_html(html_resume, output_path):
+    with open(output_path, 'w', encoding='utf8') as f:
+        f.write(html_resume)

src/templates/resume.html

@@ -6,10 +6,13 @@
         <div class="inner">
             <div class="name-hero">
                 <div class="name-text">
-                    <h1>{{name}} </h1>
-                    <h3>{{title}} </h3>
+                    <h3>{{name}} </h3>
+
                 </div>
 
+				<div class="title">
+                    <h3> {{title}} </h3>
+                 </div>
             </div>
 
 
@@ -37,7 +40,7 @@
         </div>
     </div>
     <div class="right">
-            <h1>Contact Information</h1>
+            <h2>Contact Information</h2>
             <p>Email: {{email}}</p>
             <p>Phone: {{phone}}</p>
             <p>Address: {{address}}</p>

src/ui.py

@@ -0,0 +1,185 @@
+import streamlit as st
+
+from src.chatbot.chatgpt import openai_key_info, Chatgpt
+from src.data_handler import improve_resume, init_resume, download_pdf, update_resume_data, PDFSizeException
+from src.exceptions import ChatbotInitException
+from src.utils import is_new_file, is_data_loaded, key_to_tab_name, get_item_key, init_user_info
+
+section_examples = {'summary': 'I have passion for new tech',
+                    'workExperience': 'Tell about my ability to lead projects',
+                    'education': 'Describe my degree type in more details'}
+
+
+def title():
+    st.title("ChatCV - AI Resume Builder")
+
+
+def resume_header():
+    st.text_input('name', st.session_state.resume_data['name'], key="name")
+    st.text_input('title', st.session_state.resume_data['title'], key="title")
+
+
+def unknown_error():
+    st.session_state['user_info'] = init_user_info(error_info, "It's just a glitch in the matrix."
+                                                               " Try hitting refresh, and if that doesn't work, just imagine yourself in a peaceful place.")
+    user_info()
+
+
+def user_info():
+    if not st.session_state.get('user_info'):
+        upload_resume_header()
+
+    message_type = st.session_state['user_info']['message_type']
+    message = st.session_state['user_info']['message']
+    message_type(message)
+
+
+def upload_resume_header():
+    st.session_state['user_info'] = init_user_info(st.success, "Upload PDF Resume - Let the magic begin. \n\n"
+                                                               "This may take a bit... Grub a warm cup of coffee while we working :)")
+
+
+def upload(uploaded_file):
+    try:
+        resume_data = init_resume(uploaded_file)
+        st.session_state['user_info'] = init_user_info(success_info, "Working on it...")
+        improve_resume(resume_data)
+
+    except PDFSizeException:
+        st.session_state['user_info'] = init_user_info(error_info, "PDF size max size is 4, try upload again...")
+
+    except Exception:
+        st.session_state['user_info'] = init_user_info(error_info, "PDF upload, try upload again...")
+
+
+def sidebar():
+    with st.sidebar:
+        uploaded_file = st.file_uploader('Upload PDF Resume', type=["PDF"])
+        if uploaded_file and is_new_file(uploaded_file):
+            upload(uploaded_file)
+            st.experimental_rerun()
+
+        if is_data_loaded():
+            st.button("Improve More", on_click=improve_resume)
+            st.download_button('Download PDF', file_name='out.pdf', mime="application/json", data=download_pdf())
+
+
+def body():
+    section_dict = {'contactInfo': contact_info_section, 'summary': summary_section, 'workExperience': list_section,
+                    'education': list_section, 'skills': skills_section}
+    tabs_names = [key_to_tab_name(key) for key in section_dict.keys()]
+    tabs = st.tabs(tabs_names)
+    for tab, key in zip(tabs, section_dict):
+        section_func = section_dict[key]
+        with tab:
+            section_func(key, st.session_state['resume_data'][key])
+
+
+def init_chatbot():
+    cols = st.columns([6, 1, 1])
+    api_key = cols[0].text_input("Enter OpenAI API key")
+    cols[1].markdown("#")
+    api_submit = cols[1].button("Submit")
+
+    cols[2].markdown("#")
+    get_info = cols[2].button("Get key")
+    if get_info:
+        st.info(f"Get your key at: {openai_key_info}")
+    if api_submit:
+        if Chatgpt.validate_api(api_key):
+            try:
+                st.session_state['chatbot'] = Chatgpt(api_key)
+            except ChatbotInitException:
+                st.session_state['user_info'] = init_user_info(error_info,
+                                                               "Error with Chatbot loadin, please refresh...")
+
+            st.experimental_rerun()
+
+        else:
+            st.error("Not valid API key - try again...")
+
+
+def summary_section(section_name, summary_data):
+    st.text_area(section_name, summary_data, key=f'{section_name}', label_visibility='hidden')
+    recruiter_subsection(section_name, section_examples[section_name])
+
+
+def list_section(section_name, section_data):
+    description_key = 'description'
+
+    item_keys = list(section_data[0].keys())
+    item_keys.remove(description_key)
+    for item_id, section_item in enumerate(section_data):
+        cols = st.columns(len(item_keys))
+        for col, key in zip(cols, item_keys):
+            col.text_input(key, section_item[key], key=f'{section_name}_{item_id}_{key}')
+        st.text_area(description_key, section_item[description_key], key=f'{section_name}_{item_id}_{description_key}')
+
+        recruiter_subsection(section_name, section_example=section_examples[section_name], item_id=item_id)
+        st.markdown('***')
+
+
+def skills_section(section_name, skills_data):
+    num_columns = 3
+    for skills_row in range(0, len(skills_data), num_columns):
+        cols = st.columns([3, 1] * num_columns)
+        skills_row_names = skills_data[skills_row: skills_row + num_columns]
+        for item_id, skill in enumerate(skills_row_names):
+            skill_id = skills_row + item_id
+            cols[item_id * 2].text_input(' ', value=skill, key=f'{section_name}_{skill_id}', label_visibility='hidden')
+            cols[item_id * 2 + 1].markdown('## ')
+            if cols[item_id * 2 + 1].button('x', key=f'{section_name}_{skill_id}remove_from_list'):
+                del skills_data[skill_id]
+                st.experimental_rerun()
+
+    skill_subsection(section_name)
+    st.markdown('***')
+
+
+def contact_info_section(section_name, info_data):
+    for key, value in info_data.items():
+        if value:
+            st.text_input(key.title(), value, key=f'{section_name}_{key}')
+    st.markdown('***')
+
+
+def skill_subsection(section_name, item_id=0):
+    key = f'{section_name}_{item_id}_add_skill'
+    cols = st.columns([12, 1])
+    new_skill = cols[0].text_input("Add skill", key=key)
+    cols[1].markdown('##')
+    clicked = cols[1].button("\+")
+    if clicked and new_skill:
+        st.write(new_skill)
+        st.session_state['resume_data'][section_name].append(new_skill)
+        st.write(st.session_state['resume_data'][section_name])
+        st.experimental_rerun()
+
+
+def recruiter_subsection(section_name, section_example, item_id=0):
+    with st.container():
+        cols = st.columns([3, 10], gap='small')
+        cols[0].write('\n')
+        cols[0].write('\n')
+        button_clicked = cols[0].button("Auto Section Improve", key=f'{section_name}_{item_id}_improve_auto')
+        trigger_key = 'Add a special request'
+        user_request_template = f"{trigger_key} to the bot here... e.g. {section_example}."
+
+        user_request = cols[1].text_input("section_example", value=user_request_template,
+                                          key=f'{section_name}_{item_id}_improve_manual', label_visibility='hidden')
+        if button_clicked:
+            user_request = '' if trigger_key in user_request else user_request
+            section_key = get_item_key(section_name, item_id)
+            section_text = st.session_state[section_key]
+            new_section_text = st.session_state['chatbot'].improve_section(section_text, user_request)
+
+            update_resume_data(new_section_text, section_name, item_id)
+            st.experimental_rerun()
+
+
+def success_info(message):
+    st.success(message)
+
+
+def error_info(message):
+    st.error(message)

src/utils.py

@@ -1,38 +1,42 @@
-import PyPDF2
-from jinja2 import FileSystemLoader, Environment
+import re
 
+import streamlit as st
 
-def parse_pdf(pdf_file):
-    if pdf_file is isinstance(pdf_file, str):
-        with open(pdf_file, "rb") as file:
-            return _parse(file)
-    else:
-        return _parse(pdf_file)
 
+def is_chatbot_loaded():
+    return st.session_state.get('chatbot')
 
-def _parse(file):
-    reader = PyPDF2.PdfReader(file)
-    pdf_text = []
-    # Iterate over each page
-    for page_number in range(len(reader.pages)):
-        # Get the current page
-        page = reader.pages[page_number]
 
-        # Extract the text from the page
-        page_text = page.extract_text()
+def is_new_file(uploaded_file):
+    return uploaded_file.id != st.session_state.get('file_id', '')
 
-        pdf_text.append(page_text)
-    pdf_text = '\n'.join(pdf_text)
-    return pdf_text
 
+def is_data_loaded():
+    return st.session_state.get('resume_data')
 
-def build_html_resume(data):
-    env = Environment(loader=FileSystemLoader('src/templates'))
-    template = env.get_template('resume.html')
-    html_resume = template.render(data)
-    return html_resume
 
+def key_to_tab_name(input_string):
+    return re.sub(r'([A-Z])', r' \1', input_string).strip().title()
 
-def export_html(html_resume, output_path):
-    with open(output_path, 'w', encoding='utf8') as f:
-        f.write(html_resume)
+
+def count_entries(input_dict, entry_type):
+    max_index = max([int(key.split("_")[1]) for key in input_dict.keys() if key.startswith(f"{entry_type}_")],
+                    default=0)
+    return max_index + 1
+
+
+def get_item_key(section_name, item_id=0):
+    section_key = ''
+    if section_name in ['workExperience', 'education']:
+        key = 'description'
+        section_key = f'{section_name}_{item_id}_{key}'
+    elif section_name == 'summary':
+        section_key = f'{section_name}'
+    return section_key
+
+
+def init_user_info(message_type, message):
+    return {
+        'message_type': message_type,
+        'message': message
+    }