Update app.py

app.py

@@ -1,7 +1,7 @@
 import pandas as pd
 import streamlit as st
 import numpy as np
-from scipy.stats import pearsonr
+from scipy.spatial.distance import hamming
 
 # Загрузка данных из Excel файлов
 vacancy = 'vacancy_new_new.xlsx'
@@ -47,17 +47,19 @@ if st.button("Найти совпадения"):
         # Преобразование в array для поиска
         selected_skills_array = np.array(skills_ids, dtype=int)
 
-        # Поиск совпадений по методу correlation
+        # Поиск совпадений по методу Hamming Distance
         matching_results = []
         for index, row in df_to_search.iterrows():
             if 'arr' in row and isinstance(row['arr'], str):
                 arr_values = np.array([int(x) for x in row['arr'].split(',')], dtype=int)
-                correlation, _ = pearsonr(selected_skills_array, arr_values)
-                if correlation > 0.5:
-                    matching_results.append({
-                        'Name': row['name'],
-                        'Correlation': correlation
-                    })
+                if len(arr_values) == len(selected_skills_array):
+                    hamming_distance = hamming(selected_skills_array, arr_values)
+                    similarity = 1 - hamming_distance
+                    if similarity > 0.5:
+                        matching_results.append({
+                            'Name': row['name'],
+                            'Similarity': similarity
+                        })
 
         # Вывод результатов, где совпадение больше 50%
         st.write("Результаты с совпадением более 50%:")
@@ -67,7 +69,7 @@ if st.button("Найти совпадения"):
                     f"""
                     <div style="border: 1px solid #ddd; padding: 10px; border-radius: 5px; margin-bottom: 10px;">
                         <h4 style="margin: 0;">{result['Name']}</h4>
-                        <p><strong>Коэффициент совпадения:</strong> {result['Correlation']:.2f}</p>
+                        <p><strong>Коэффициент совпадения:</strong> {result['Similarity']:.2f}</p>
                     </div>
                     """,
                     unsafe_allow_html=True