Add new category business

app.py

@@ -126,7 +126,7 @@ def main():
     data = load_dataset(columns)
     model = load_model()
     corpus_embeddings = create_embedding(model, data, 'Previous Role')
-    col1, col2, col3, col4, col5, col6, _ = st.columns([1.1, 1.3, 1.6, 1.65, 1.7, 2.1, 9])
+    col1, col2, col3, col4, col5, col6, col7, _ = st.columns([1.1, 1.3, 1.6, 1.65, 1.7, 2.1, 2.25, 9])
     
     with col1:
         data_count = get_specific_category(model, data, 'data', corpus_embeddings)
@@ -146,6 +146,9 @@ def main():
     with col6:
         computer_count = get_specific_category(model, data, 'engineer', corpus_embeddings)
         computer_bt = st.button(f'Computer Science ({computer_count})')
+    with col7:
+        business_count = get_specific_category(model, data, 'business and management', corpus_embeddings)
+        business_bt = st.button(f'Business and Management ({business_count})')
 
     job_title = st.text_input('Insert the job title below:', '')
     submitted = st.button('Submit')
@@ -153,7 +156,9 @@ def main():
     if data_bt:
         job_title = 'data'
     if finance_bt:
-        job_title = 'finance'
+        job_title = 'finance and accounting'
+    if business_bt:
+        job_title = 'business and management'    
     if marketing_bt:
         job_title = 'marketing'
     if social_media_bt:
@@ -163,7 +168,7 @@ def main():
     if computer_bt:
         job_title = 'engineer and developer'
 
-    if submitted or data_bt or finance_bt or marketing_bt or social_media_bt or arts_design_bt or computer_bt:
+    if submitted or data_bt or finance_bt or marketing_bt or social_media_bt or arts_design_bt or computer_bt or business_bt:
         print(job_title + ',' + str(pd.Timestamp.now()))
         st.info(f'Showing most similar results for {job_title}...')
         result = get_similarity_score(model, data, job_title, corpus_embeddings)