Update app.py

app.py

@@ -852,6 +852,7 @@ class NutrientCalculator:
             print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
             raise
 
+
     def _compensate_element(self, element):
         needed = self.target_profile[element] - self.actual_profile.get(element, 0)
         if needed <= 0:
@@ -860,14 +861,17 @@ class NutrientCalculator:
         candidates = []
     
         # Для азотных соединений особый случай
-        if element in ['NH4', 'NO3']:
+        if element in ['N (NH4+)', 'N (NO3-)']:
             # Здесь ваша текущая логика для NH4 и NO3
             pass
         else:
             # Стандартная компенсация для других элементов
             for weight_key, weight_data in self.element_compensation_weights.items():
                 fert_name = weight_data["fert"]
-                if element in self.fertilizers.get(fert_name, {}):
+                main_element = weight_data["main_element"]
+            
+                # Проверяем, что удобрение содержит нужный элемент
+                if main_element == element and element in self.fertilizers.get(fert_name, {}):
                     candidates.append({
                         'name': fert_name,
                         'weight': weight_data["weight"],
@@ -877,11 +881,25 @@ class NutrientCalculator:
         if not candidates:
             raise ValueError(f"Нет удобрений для элемента {element}")
 
-        total_weight = sum(abs(c['weight']) for c in candidates)  # Берем модуль весов
+        # Для отрицательных весов нужно инвертировать знак needed
+        if any(c['weight'] < 0 for c in candidates):
+            needed = -needed
+
+        total_weight = sum(c['weight'] for c in candidates)
         for candidate in candidates:
-            share = abs(candidate['weight']) / total_weight  # Используем модуль веса
+           share = candidate['weight'] / total_weight
             ppm_to_apply = needed * share
             self._apply(candidate['name'], element, ppm_to_apply)
+                    })
+
+            if not candidates:
+                raise ValueError(f"Нет удобрений для элемента {element}")
+
+            total_weight = sum(abs(c['weight']) for c in candidates)  # Берем модуль весов
+            for candidate in candidates:
+                share = abs(candidate['weight']) / total_weight  # Используем модуль веса
+                ppm_to_apply = needed * share
+                self._apply(candidate['name'], element, ppm_to_apply)
 
     def calculate_ec(self):
         return round(self.total_ec, 2)