Summary

app/event_handlers/calculate_practical_tasks.py

@@ -662,37 +662,89 @@ def event_handler_calculate_practical_task_blocks(
     elif (
         practical_subtasks.lower() == "finding a suitable junior colleague"
         or practical_subtasks.lower() == "finding a suitable senior colleague"
+        or practical_subtasks.lower()
+        == "finding a suitable colleague by personality types"
     ):
-        df_correlation_coefficients = read_csv_file(
-            config_data.Links_FINDING_COLLEAGUE, ["ID"]
-        )
+        pt_scores_copy = pt_scores.iloc[:, 1:].copy()
 
-        b5._colleague_ranking(
-            df_files=pt_scores.iloc[:, 1:],
-            correlation_coefficients=df_correlation_coefficients,
-            target_scores=[
-                target_score_ope,
-                target_score_con,
-                target_score_ext,
-                target_score_agr,
-                target_score_nneu,
-            ],
-            colleague=colleague_type(practical_subtasks),
-            equal_coefficients=equal_coefficient,
-            out=False,
-        )
+        preprocess_scores_df(pt_scores_copy, config_data.Dataframes_PT_SCORES[0][0])
+
+        if (
+            practical_subtasks.lower()
+            != "finding a suitable colleague by personality types"
+        ):
+            df_correlation_coefficients = read_csv_file(
+                config_data.Links_FINDING_COLLEAGUE, ["ID"]
+            )
 
-        df = apply_rounding_and_rename_columns(b5.df_files_colleague_)
+            b5._colleague_ranking(
+                df_files=pt_scores_copy,
+                correlation_coefficients=df_correlation_coefficients,
+                target_scores=[
+                    target_score_ope,
+                    target_score_con,
+                    target_score_ext,
+                    target_score_agr,
+                    target_score_nneu,
+                ],
+                colleague=colleague_type(practical_subtasks),
+                equal_coefficients=equal_coefficient,
+                out=False,
+            )
+            df = apply_rounding_and_rename_columns(b5.df_files_colleague_)
 
-        df_hidden = df.drop(columns=config_data.Settings_SHORT_PROFESSIONAL_SKILLS)
+            df_hidden = df.drop(columns=config_data.Settings_SHORT_PROFESSIONAL_SKILLS)
 
-        df_hidden.to_csv(
-            colleague_type(practical_subtasks) + config_data.Filenames_COLLEAGUE_RANKING
-        )
+            df_hidden.to_csv(
+                colleague_type(practical_subtasks)
+                + config_data.Filenames_COLLEAGUE_RANKING
+            )
+        else:
+            b5._colleague_personality_type_match(
+                df_files=pt_scores_copy,
+                correlation_coefficients=None,
+                target_scores=[
+                    target_score_ope,
+                    target_score_con,
+                    target_score_ext,
+                    target_score_agr,
+                    target_score_nneu,
+                ],
+                threshold=equal_coefficient,
+                out=False,
+            )
+            df = b5.df_files_MBTI_colleague_match_.rename(
+                columns={
+                    "MBTI": "Personality Type",
+                    "MBTI_Score": "Personality Type Score",
+                }
+            )
+
+            df_hidden = df[["Path", "Personality Type", "Match"]]
+
+            df_hidden.to_csv(config_data.Filenames_COLLEAGUE_RANKING)
 
         df_hidden.reset_index(inplace=True)
 
-        person_id = int(df_hidden.iloc[0][config_data.Dataframes_PT_SCORES[0][0]]) - 1
+        if (
+            practical_subtasks.lower()
+            != "finding a suitable colleague by personality types"
+        ):
+            person_id = (
+                int(
+                    df_hidden.iloc[
+                        (
+                            0
+                            if practical_subtasks.lower()
+                            != "finding a suitable colleague by personality types"
+                            else 1
+                        )
+                    ][config_data.Dataframes_PT_SCORES[0][0]]
+                )
+                - 1
+            )
+        else:
+            person_id = 0
 
         person_metadata = create_person_metadata(person_id, files, video_metadata)
 

app/event_handlers/practical_subtasks.py

@@ -219,6 +219,8 @@ def event_handler_practical_subtasks(
     elif (
         practical_subtasks.lower() == "finding a suitable junior colleague"
         or practical_subtasks.lower() == "finding a suitable senior colleague"
+        or practical_subtasks.lower()
+        == "finding a suitable colleague by personality types"
     ):
         return (
             practical_subtasks_selected,
@@ -297,7 +299,12 @@ def event_handler_practical_subtasks(
                 minimum=0.0,
                 maximum=1.0,
                 step=0.01,
-                label=config_data.Labels_EQUAL_COEFFICIENT_LABEL,
+                label=(
+                    config_data.Labels_THRESHOLD_TARGET_SCORE_LABEL
+                    if practical_subtasks.lower()
+                    == "finding a suitable colleague by personality types"
+                    else config_data.Labels_EQUAL_COEFFICIENT_LABEL
+                ),
                 info=config_data.InformationMessages_VALUE_FROM_TO_INFO.format(0, 1.0),
                 show_label=True,
                 interactive=True,

config.toml

@@ -1,5 +1,5 @@
 [AppSettings]
-APP_VERSION = "0.10.8"
+APP_VERSION = "0.10.9"
 SERVER_NAME = "127.0.0.1"
 PORT = 7860
 CSS_PATH = "app.css"
@@ -72,6 +72,7 @@ TARGET_SCORE_EXT_LABEL = "Extraversion target score"
 TARGET_SCORE_AGR_LABEL = "Agreeableness target score"
 TARGET_SCORE_NNEU_LABEL = "Non-Neuroticism target score"
 EQUAL_COEFFICIENT_LABEL = "Equal coefficient"
+THRESHOLD_TARGET_SCORE_LABEL = "Polarity traits threshold"
 NUMBER_PRIORITY_LABEL = "Priority number"
 NUMBER_IMPORTANCE_TRAITS_LABEL = "Importance traits number"
 NUMBER_IMPORTANCE_OPE_LABEL = "Openness weight"

practical_tasks_en.yaml

@@ -7,6 +7,7 @@
   subtasks:
     - "Finding a suitable junior colleague"
     - "Finding a suitable senior colleague"
+    - "Finding a suitable colleague by personality types"
 - task: "Predicting consumer preferences for industrial goods"
   subtasks:
     - "Car characteristics"

practical_tasks_ru.yaml

@@ -7,6 +7,7 @@
   subtasks:
     - "Поиск подходящего младшего коллеги"
     - "Поиск подходящего старшего коллеги"
+    - "Поиск подходящего коллеги по типам личности"
 - task: "Прогнозирование потребительских предпочтений в отношении промышленных товаров"
   subtasks:
     - "Характеристики автомобиля"