Update app.py

app.py

@@ -16,21 +16,38 @@ def calculate_correlations(file_obj):
                 except pd.errors.ParserError as e:
                     return {"error": f"Ошибка парсинга файла: {e}"}
 
+        # НОВЫЕ НАЗВАНИЯ СТОЛБЦОВ
+        if 'оценка' not in df.columns:
+            return {"error": "Столбец 'оценка' не найден"}
 
-        # Проверяем, что столбец 'Soon ' существует
-        if 'Soon ' not in df.columns:
-          return {"error": "Столбец 'Soon ' не найден в файле."}
+        if 'коментарий' not in df.columns:
+            return {"error": "Столбец 'коментарий' не найден"}
 
+        if 'статус' not in df.columns:
+            return {"error": "Столбец 'статус' не найден"}
 
-        # Преобразование 'Soon ' с обработкой ошибок
+
+        # Преобразование 'оценка'
         try:
-            df['Soon '] = df['Soon '].astype(str).str.replace(' из 5', '')
-            df['Soon '] = pd.to_numeric(df['Soon '].astype(str).str.replace(',', '.'), errors='coerce')
+            df['оценка'] = df['оценка'].astype(str).str.replace(' из 5', '')
+            df['оценка'] = pd.to_numeric(df['оценка'].astype(str).str.replace(',', '.'), errors='coerce')
         except AttributeError:
-            return {"error": "Ошибка преобразования столбца 'Soon '. Проверьте формат данных."}
+            return {"error": "Ошибка преобразования столбца 'оценка'. Проверьте формат данных."}
+
+        # 1. Оценка и наличие отзыва (коментарий)
+        df['has_review'] = df['коментарий'].notna().astype(int)
+        correlation_review = df['оценка'].corr(df['has_review'])
 
-        # ... (остальной код для расчета корреляций без изменений) ...
+        # 2. Оценка и тип размещения (приблизительно)
+        accommodation_types = ['кемпер', 'палатка', 'бунгало', 'мобильный дом']
+        correlation_accommodation = {}
+        for acc_type in accommodation_types:
+            df[acc_type] = df['коментарий'].apply(lambda x: 1 if isinstance(x, str) and re.search(acc_type, x, re.IGNORECASE) else 0)
+            correlation_accommodation[acc_type] = df['оценка'].corr(df[acc_type])
 
+        # 3. Оценка и статус эксперта
+        df['is_expert'] = df['статус'].apply(lambda x: 1 if isinstance(x, str) and 'местный эксперт' in x.lower() else 0)
+        correlation_expert = df['оценка'].corr(df['is_expert'])
 
         return {
             "correlation_review": correlation_review,
@@ -38,11 +55,8 @@ def calculate_correlations(file_obj):
             "correlation_expert": correlation_expert
         }
 
-
     except Exception as e:
-        return {"error": f"Неизвестная ошибка: {e}"} # Общее сообщение об ошибке
-
-
+        return {"error": f"Неизвестная ошибка: {e}"}
 
 iface = gr.Interface(
     fn=calculate_correlations,