Update app.py

app.py

@@ -833,51 +833,39 @@ class NutrientCalculator:
         self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
         self.total_ec = 0.0
         
-        # Веса компенсации с добавлением кальциевой селитры для NO3
+        # Веса компенсации (только для элементов, которые компенсируются)
         self.element_compensation_weights = {
             "POTASSIUM_SULFATE": {"weight": 0.18, "fert": "Калий сернокислый", "main_element": "K"},
             "MAGNESIUM_SULFATE": {"weight": 0.1, "fert": "Сульфат магния", "main_element": "Mg"},
-            "MONOPOTASSIUM_PHOSPHATE": {"weight": 0.14, "fert": "Монофосфат калия", "main_element": "P"},
-            "CALCIUM_NITRATE": {"weight": 0.1, "fert": "Кальциевая селитра", "main_element": "Ca"},
-            "CALCIUM_NITRATE_NO3": {"weight": 0.1, "fert": "Кальциевая селитра", "main_element": "N (NO3-)"}
+            "MONOPOTASSIUM_PHOSPHATE": {"weight": 0.14, "fert": "Монофосфат калия", "main_element": "P"}
         }
+        
+        # Коэффициенты распределения для кальциевой селитры
+        self.calcium_nitrate_ratios = {
+            "Ca": 0.7,  # 70% дефицита кальция покрываем кальциевой селитрой
+            "NO3": 0.3  # 30% дефицита NO3 покрываем кальциевой селитрой
+        }
+
         # Расчёт азота
         total_parts = NO3_RATIO + NH4_RATIO
         self.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
         self.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
 
-    def _label(self, element):
-        """Форматирование названий элементов для вывода"""
-        labels = {
-            'N (NO3-)': 'NO3',
-            'N (NH4+)': 'NH4'
-        }
-        return labels.get(element, element)
-
     def calculate(self):
         try:
-            # Вносим магний через компенсацию
+            # 1. Вносим кальциевую селитру для Ca и частично для NO3
+            self._apply_calcium_nitrate()
+            
+            # 2. Вносим аммонийный азот
+            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+            
+            # 3. Компенсируем остаток NO3 калийной селитрой
+            self._compensate_no3()
+            
+            # 4. Вносим остальные элементы через компенсацию
             self._compensate_element("Mg")
-
-            # Вносим кальций напрямую
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
-
-            # Вносим фосфор через компенсацию
             self._compensate_element("P")
-
-            # Вносим аммонийный азот напрямую
-            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
-
-            # Компенсируем нитратный азот напрямую
-            current_no3 = self.actual_profile['N (NO3-)']
-            no3_needed = self.target_profile['N (NO3-)'] - current_no3
-            if no3_needed > 0.1:
-                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
-
-            # Компенсируем серу через компенсацию
             self._compensate_element("S")
-
-            # Компенсируем калий через компенсацию
             self._compensate_element("K")
 
             return self.results
@@ -885,6 +873,53 @@ class NutrientCalculator:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
+    def _apply_calcium_nitrate(self):
+        """Специальный метод для внесения кальциевой селитры"""
+        # Вносим кальций
+        ca_needed = self.target_profile['Ca'] - self.actual_profile.get('Ca', 0)
+        if ca_needed > 0.1:
+            ca_share = self.calcium_nitrate_ratios['Ca']
+            ca_to_apply = ca_needed * ca_share
+            self._apply("Кальциевая селитра", "Ca", ca_to_apply)
+        
+        # Вносим часть NO3 из кальциевой селитры
+        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
+        if no3_needed > 0.1:
+            no3_share = self.calcium_nitrate_ratios['NO3']
+            no3_to_apply = no3_needed * no3_share
+            self._apply("Кальциевая селитра", "N (NO3-)", no3_to_apply)
+
+    def _compensate_no3(self):
+        """Компенсация остатка NO3 после внесения ка��ьциевой селитры"""
+        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
+        if no3_needed > 0.1:
+            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+
+    def _compensate_element(self, element):
+        """Общий метод компенсации для других элементов"""
+        needed = self.target_profile[element] - self.actual_profile.get(element, 0)
+        if abs(needed) < 0.1:
+            return
+
+        candidates = []
+        for weight_key, weight_data in self.element_compensation_weights.items():
+            fert_name = weight_data["fert"]
+            if element in self.fertilizers.get(fert_name, {}):
+                candidates.append({
+                    'name': fert_name,
+                    'weight': weight_data["weight"],
+                    'content': self.fertilizers[fert_name][element]
+                })
+
+        if not candidates:
+            raise ValueError(f"Нет удобрений для элемента {element}")
+
+        total_weight = sum(c['weight'] for c in candidates)
+        for candidate in candidates:
+            share = candidate['weight'] / total_weight
+            ppm_to_apply = needed * share
+            self._apply(candidate['name'], element, ppm_to_apply)
+
     def _apply(self, fert_name, main_element, required_ppm):
         if required_ppm <= 0:
             return