Update app.py

app.py

@@ -726,7 +726,7 @@ class NutrientCalculator:
         self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
         self.total_ec = 0.0
 
-        # Расчёт азота
+        # Расчет азота
         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)
@@ -743,26 +743,66 @@ class NutrientCalculator:
         }
         return labels.get(element, element)
 
+    def _balance_nitrogen_with_k(self):
+        """Новый метод с компенсацией калия"""
+        # 1. Вносим NH4 через аммиачную селитру
+        nh4_needed = self.target_profile['N (NH4+)'] - self.actual_profile['N (NH4+)']
+        if nh4_needed > 0.1:
+            self._apply("Аммоний азотнокислый", "N (NH4+)", nh4_needed)
+        
+        # 2. Рассчитываем сколько NO3 нужно внести
+        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile['N (NO3-)']
+        
+        # 3. Сначала вносим через кальциевую селитру (для Ca)
+        ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
+        if ca_needed > 0.1 and no3_needed > 0.1:
+            max_ca_no3 = (ca_needed / self.fertilizers["Кальциевая селитра"]["Ca"]) * \
+                        self.fertilizers["Кальциевая селитра"]["N (NO3-)"]
+            ca_no3_part = min(no3_needed, max_ca_no3)
+            self._apply("Кальциевая селитра", "N (NO3-)", ca_no3_part)
+            no3_needed -= ca_no3_part
+        
+        # 4. Остаток NO3 вносим через калийную селитру
+        if no3_needed > 0.1:
+            # Рассчитываем сколько K внесется с этим NO3
+            k_from_kno3 = (no3_needed / self.fertilizers["Калий азотнокислый"]["N (NO3-)"]) * \
+                         self.fertilizers["Калий азотнокислый"]["K"]
+            
+            # Проверяем, не превысит ли это целевое значение K
+            k_remaining = self.target_profile['K'] - self.actual_profile['K']
+            if k_from_kno3 > k_remaining:
+                # Если превысит - уменьшаем количество KNO3
+                no3_needed = (k_remaining / self.fertilizers["Калий азотнокислый"]["K"]) * \
+                            self.fertilizers["Калий азотнокислый"]["N (NO3-)"]
+            
+            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+
     def calculate(self):
+        """Основной метод расчета с новой логикой"""
         try:
+            # 1. Вносим Mg и S
             self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
-            self._apply("Монофосфат калия", "P", self.target_profile['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
+            # 2. Балансируем азот с учетом калия
+            self._balance_nitrogen_with_k()
             
-            if no3_needed > 0.1:
-                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+            # 3. Вносим Ca (остаток)
+            ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
+            if ca_needed > 0.1:
+                self._apply("Кальциевая селитра", "Ca", ca_needed)
             
-            self._apply_k_sulfate()
+            # 4. Вносим P
+            p_needed = self.target_profile['P'] - self.actual_profile['P']
+            if p_needed > 0.1:
+                self._apply("Монофосфат калия", "P", p_needed)
             
-            k_deficit = self.target_profile['K'] - self.actual_profile['K']
-            if k_deficit > 0.1:
-                self._apply("Калий азотнокислый", "K", k_deficit)
+            # 5. Корректируем K через сульфат калия (если остался дефицит)
+            k_needed = self.target_profile['K'] - self.actual_profile['K']
+            if k_needed > 0.1:
+                self._apply("Калий сернокислый", "K", k_needed)
             
             return self.results
+            
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise