Update app.py

app.py

@@ -19,7 +19,7 @@ with open("embeddings.npy", 'rb') as f:
 
 def get_recommend(user_input, 
                   top_k_spec = 3,
-                  top_k_services = 5,
+                  top_k_services = 10,
                   treshold = 0.8):
     
     cols_for_top_k = ["Специальность врача",
@@ -41,15 +41,33 @@ def get_recommend(user_input,
     for col in cols_for_top_k:
         result[col] = sorted_df[col].value_counts()[:top_k_spec].index.tolist()
     result['Жалобы'] = sorted_df['Жалобы'].value_counts()[:top_k_services].index.tolist()
-    result['Диагноз МКБ'] = sorted_df['Диагноз МКБ'].value_counts()[:top_k_services].index.tolist()
     
-    lst = []    
+    top_k_mkb = sorted_df['Диагноз МКБ'].value_counts()[:top_k_services].index.tolist()
+    result['Диагноз МКБ'] = top_k_mkb
+    
     categories = ['Инструментальная диагностика', 'Лабораторная диагностика']
+    
+    top_k_services_lst_by_mkb = []
+    for mkb in top_k_mkb:
+        temp_lst = []
+        slice_df = sorted_df[sorted_df['Диагноз МКБ'] == mkb]
+        for category in categories:
+            top_k_services_in_cat_mkb = slice_df[slice_df['service_name_category'] == category]['Рекомендации по обследованию'].value_counts()[:top_k_services].index.tolist()
+            temp_lst.append({category:top_k_services_in_cat_mkb})
+            
+        top_k_services_lst_by_mkb.append({mkb:temp_lst})
+            
+    top_k_services_lst = []    
+    
     for category in categories:
-        list_top_k_services = sorted_df[sorted_df['service_name_category'] == category]['Рекомендации по обследованию'].value_counts()[:top_k_services].index.tolist()
-        lst.append({category:list_top_k_services})
+        slice_df = sorted_df[sorted_df['service_name_category'] == category]
+        list_top_k_services = slice_df['Рекомендации по обследованию'].value_counts()[:top_k_services].index.tolist()
+        top_k_services_lst.append({category:list_top_k_services})
+        
+        
     
-    result['Рекомендации по обследованию'] = lst
+    result['Рекомендации по обследованию'] = top_k_services_lst
+    result['Рекомендации по обследованию по МКБ'] = top_k_services_lst_by_mkb
 
     return result
 #%%