Add common categories

app.py

@@ -112,6 +112,11 @@ def show_heading():
     ''', unsafe_allow_html=True)
     st.write('This app lets you search and sort talent by job title or relevant job descriptions from ecommurz talent list in real-time.')
 
+def get_specific_category(model, data, category, corpus_embeddings):
+    """Get specific category with confidence score > 0.45"""
+    data = get_similarity_score(model, data, category, corpus_embeddings)
+    return data[data['score'] > 0.45].shape[0]
+    
 def main():
     """Main Function"""
     show_heading()
@@ -121,11 +126,44 @@ 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(10)
+
+    with col1:
+        data_count = get_specific_category(model, data, 'data', corpus_embeddings)
+        data_bt = st.button(f'Data ({data_count})')
+    with col2:
+        finance_count = get_specific_category(model, data, 'finance', corpus_embeddings)
+        finance_bt = st.button(f'Finance ({finance_count})')
+    with col3:
+        marketing_count = get_specific_category(model, data, 'marketing', corpus_embeddings)
+        marketing_bt = st.button(f'Marketing ({marketing_count})')
+    with col4:
+        social_media_count = get_specific_category(model, data, 'social media', corpus_embeddings)
+        social_media_bt = st.button(f'Social Media ({social_media_count})')
+    with col5:
+        arts_design_count = get_specific_category(model, data, 'design and creative', corpus_embeddings)
+        arts_design_bt = st.button(f'Arts & Design ({arts_design_count})')
+    with col6:
+        computer_count = get_specific_category(model, data, 'engineer', corpus_embeddings)
+        computer_bt = st.button(f'Computer Science ({computer_count})')
 
     job_title = st.text_input('Insert the job title below:', '')
     submitted = st.button('Submit')
 
-    if submitted:
+    if data_bt:
+        job_title = 'data'
+    if finance_bt:
+        job_title = 'finance'
+    if marketing_bt:
+        job_title = 'marketing'
+    if social_media_bt:
+        job_title = 'social media'
+    if arts_design_bt:
+        job_title = 'design and creative'
+    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:
         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)