Upload 14 files

src/chatbot/chatgpt.py

@@ -18,7 +18,6 @@ openai_key_info = 'https://platform.openai.com/account/api-keys'
 
 class Chatgpt:
     def __init__(self, api_key):
-        self.validate_api(api_key)
         self.chatbot = Chatbot(api_key)
         logging.info("API key loaded successfully")
 

src/chatbot/prompts.py

@@ -2,9 +2,8 @@ prompt_placeholder = '[$$$]'
 
 data_format = {'name': '', 'title': '',
                'contactInfo': {'linkedin': '', 'github': '', 'email': '', 'address': '', 'phone': ''}, 'summary': '',
-               'workExperience': [{'title': '', 'company': '', 'dates': '', 'description': ''},
-                                  {'title': '', 'company': '', 'dates': '', 'description': ''}, ],
-               'education': [{'degree': '', 'school': '', 'dates': '', 'description': ''}, ], 'skills': ['', '', '']}
+               'workExperience': [{'title': '', 'company': '', 'dates': '', 'description': ''}, ],
+               'education': [{'degree': '', 'school': '', 'dates': '', 'description': ''}, ], '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' \
@@ -31,4 +30,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/data_handler.py

@@ -23,7 +23,6 @@ def update_resume_data(text_input, section_name, item_id=0):
         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()
@@ -85,5 +84,4 @@ def format_resume_data():
 
         skills.append(current_state.get(skill_key, ''))
     resume_data['skills'] = skills
-
     return resume_data

src/templates/resume.html

@@ -41,11 +41,11 @@
     </div>
     <div class="right">
             <h2>Contact Information</h2>
-            <p>Email: {{email}}</p>
-            <p>Phone: {{phone}}</p>
-            <p>Address: {{address}}</p>
-            <p>LinkedIn: <a href="{{linkedin_url}}">LinkedIn Profile</a></p>
-            <p>Github: <a href="{{github_url}}">GitHub Profile</a></p>
+            <p>Email: {{contactInfo.email}}</p>
+            <p>Phone: {{contactInfo.phone}}</p>
+            <p>Address: {{contactInfo.address}}</p>
+            <p>LinkedIn: <a href="{{contactInfo.linkedin}}">LinkedIn Profile</a></p>
+            <p>Github: <a href="{{contactInfo.github}}">GitHub Profile</a></p>
 
         <section>
             <h1>Skills</h1>

src/ui.py

@@ -1,13 +1,15 @@
 import streamlit as st
 
 from src.chatbot.chatgpt import openai_key_info, Chatgpt
+from src.chatbot.prompts import data_format
 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'}
+                    'education': 'Describe my degree type in more details',
+                    'contactInfo': 'phone, Linkedin, etc.'}
 
 
 def title():
@@ -110,16 +112,63 @@ def list_section(section_name, section_data):
     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)
+        edit_list_subsection(section_name, section_data, item_id)
+
         st.markdown('***')
 
 
+def edit_list_subsection(section_name, section_data, item_id):
+    with st.container():
+        st.markdown(
+            """<style>
+                .element-container:nth-of-type(1) button {
+                    width: 100%;
+                }
+                </style>""",
+            unsafe_allow_html=True,
+        )
+
+        remove_col, add_col = st.columns(2)
+        if remove_col.button('Delete', key=f'{section_name}_{item_id}_remove_from_list') and len(section_data) > 1:
+            del section_data[item_id]
+            st.experimental_rerun()
+
+        if add_col.button('Add', key=f'{section_name}_{item_id}_add_to_list') and len(section_data) < 10:
+            section_data.append(data_format[section_name][0])
+            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 skills_section(section_name, skills_data):
+    [skills_data.remove(skill) for skill in skills_data if not skill]
+
     num_columns = 3
     for skills_row in range(0, len(skills_data), num_columns):
         cols = st.columns([3, 1] * num_columns)
@@ -128,7 +177,7 @@ def skills_section(section_name, skills_data):
             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'):
+            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()
 
@@ -136,13 +185,6 @@ def skills_section(section_name, skills_data):
     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])
@@ -150,31 +192,45 @@ def skill_subsection(section_name, item_id=0):
     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}."
+def contact_info_section(section_name, info_data):
+    keys = sorted(info_data.keys())
+    for key in keys:
+        value = info_data[key]
+        cols = st.columns([12, 1])
+        cols[0].text_input(key.title(), value, key=f'{section_name}_{key}')
+        cols[1].markdown('##')
+        clicked = cols[1].button('\-', key=f'{section_name}_{key}_remove')
+        if clicked:
+            del info_data[key]
+            st.experimental_rerun()
 
-        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)
+    add_contact_subsection(section_name, info_data)
+    st.markdown('***')
 
-            update_resume_data(new_section_text, section_name, item_id)
+
+def add_contact_subsection(section_name, info_data):
+    st.markdown('***')
+
+    with st.container():
+        cols = st.columns([12, 1])
+        new_key = cols[0].text_input('Add new details', value=f"e.g, {section_examples[section_name]}")
+        cols[1].markdown('##')
+        clicked = cols[1].button('\+', key=f'{section_name}_add_details')
+
+        if clicked and new_key:
+            info_data[new_key] = ''
             st.experimental_rerun()
+    # if remove_col.button('Delete', key=f'{section_name}_{item_id}_remove_from_list') and len(section_data) > 1:
+    #     del section_data[item_id]
+    #     st.experimental_rerun()
+    #
+    # if add_col.button('Add', key=f'{section_name}_{item_id}_add_to_list') and len(section_data) < 10:
+    #     section_data.append(data_format[section_name][0])
+    #     st.experimental_rerun()
 
 
 def success_info(message):