Update app.py

app.py

@@ -756,23 +756,35 @@ class NutrientCalculator:
 
     def calculate(self):
         try:
+            # 1. Вносим основные элементы
             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+)'])
-            
+
+            # 2. Компенсируем NO3
             current_no3 = self.actual_profile['N (NO3-)']
             no3_needed = self.target_profile['N (NO3-)'] - current_no3
             
             if no3_needed > 0.1:
+                # Ограничиваем внесение NO3, чтобы не превысить общее количество азота
+                max_nitrogen = self.target_profile['N (NO3-)'] + self.target_profile['N (NH4+)']
+                actual_nitrogen = self.actual_profile['N (NO3-)'] + self.actual_profile['N (NH4+)']
+                if actual_nitrogen + no3_needed > max_nitrogen:
+                    no3_needed = max_nitrogen - actual_nitrogen
                 self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
-            
+
+            # 3. Компенсируем K и S через сульфат калия
             self._apply_k_sulfate()
-            
+
+            # 4. Дополнительная компенсация K, если необходимо
             k_deficit = self.target_profile['K'] - self.actual_profile['K']
             if k_deficit > 0.1:
                 self._apply("Калий азотнокислый", "K", k_deficit)
-            
+
+            # 5. Проверяем соотношение NH4:NO3
+            self._adjust_nh4_no3_ratio()
+
             return self.results
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
@@ -838,9 +850,29 @@ class NutrientCalculator:
             print(f"Ошибка при расчёте сульфата калия: {str(e)}")
             raise
 
+    def _adjust_nh4_no3_ratio(self):
+        """Корректировка соотношения NH4:NO3"""
+        nh4_actual = self.actual_profile['N (NH4+)']
+        no3_actual = self.actual_profile['N (NO3-)']
+        total_n = nh4_actual + no3_actual
+        
+        if total_n > 0:
+            nh4_ratio = nh4_actual / total_n
+            target_nh4_ratio = NH4_RATIO / (NH4_RATIO + NO3_RATIO)
+            
+            if abs(nh4_ratio - target_nh4_ratio) > 0.05:  # Допустимая погрешность 5%
+                if nh4_ratio < target_nh4_ratio:
+                    # Увеличиваем NH4+
+                    nh4_needed = total_n * target_nh4_ratio - nh4_actual
+                    self._apply("Аммоний азотнокислый", "N (NH4+)", nh4_needed)
+                else:
+                    # Уменьшаем NH4+, увеличиваем NO3-
+                    nh4_excess = nh4_actual - total_n * target_nh4_ratio
+                    self._apply("Аммоний азотнокислый", "N (NH4+)", -nh4_excess)
+                    self._apply("Калий азотнокислый", "N (NO3-)", nh4_excess)
+
     def calculate_ec(self):
         return round(self.total_ec, 2)
-
     def print_report(self):
         try:
             print("\n" + "="*60)