Update app.py

app.py

@@ -697,6 +697,7 @@ INPUT_DATA = {
         "Калий азотнокислый": {"N (NO3-)": 0.13854, "K": 0.36672},
         "Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499},
         "Сульфат магния": {"Mg": 0.10220, "S": 0.13483},
+        "Магний азотнокислый": {"Mg": 0.0948, "N (NO3-)": 0.109},  # Новое удобрение
         "Монофосфат калия": {"P": 0.22761, "K": 0.28731},
         "Калий сернокислый": {"K": 0.44874, "S": 0.18401},
         "Кальций хлорид": {"Ca": 0.272, "Cl": 0.483}
@@ -705,14 +706,16 @@ INPUT_DATA = {
         "P": 31, "K": 210, "Mg": 24, "Ca": 82, "S": 57.5, "CaCl": 18.5,
         "NO3_RAT": 8.25, "TOTAL_NITROG": 125, "liters": 100,
         "activation_cacl": 5,  # Активация CaCl
-        "enhancement_cacl": 0.1
+        "enhancement_cacl": 0.1,  # Усиление CaCl
+        "use_magnesium_nitrate": 0,  # Флаг для использования "Магния азотнокислого"
+        "magnesium_nitrate_ratio": 0  # Коэффициент для "Магния азотнокислого"
     }
 }
 
 
 
 class NutrientCalculator:
-    def __init__(self, fertilizer_constants, profile_settings, liters, rounding_precision, activation_cacl, enhancement_cacl):
+    def __init__(self, fertilizer_constants, profile_settings, liters, rounding_precision, activation_cacl, enhancement_cacl, use_magnesium_nitrate, magnesium_nitrate_ratio):
         self.fertilizers = fertilizer_constants
         self.profile = profile_settings
         self.volume = liters
@@ -721,6 +724,8 @@ class NutrientCalculator:
         # Новые параметры
         self.activation_cacl = float(activation_cacl)  # Активация CaCl
         self.enhancement_cacl = float(enhancement_cacl)  # Усиление CaCl
+        self.use_magnesium_nitrate = int(use_magnesium_nitrate)  # Флаг для использования "Магния азотнокислого"
+        self.magnesium_nitrate_ratio = float(magnesium_nitrate_ratio)  # Коэффициент для "Магния азотнокислого"
         
         total_parts = self.profile["NO3_RAT"] + 1
         self.target = {
@@ -737,14 +742,39 @@ class NutrientCalculator:
         self.results = {fert: {'граммы': 0.0} for fert in self.fertilizers}
 
     def calculate(self):
+        # Добавляем аммонийный азот
         self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target["N (NH4+)"])
+        
+        # Добавляем фосфор
         self._apply_fertilizer("Монофосфат калия", "P", self.target["P"])
-        self._apply_fertilizer("Сульфат магния", "Mg", self.target["Mg"])
+        
+        # Добавляем магний
+        if self.use_magnesium_nitrate == 0:
+            # Используем только "Сульфат магния"
+            mg_needed = self.target["Mg"] - self.actual["Mg"]
+            if mg_needed > 0:
+                self._apply_fertilizer("Сульфат магния", "Mg", mg_needed)
+        else:
+            # Используем распределение между "Сульфатом магния" и "Магнием азотнокислым"
+            mg_needed = self.target["Mg"] - self.actual["Mg"]
+            if mg_needed > 0:
+                self._apply_fertilizer("Сульфат магния", "Mg", mg_needed * (1 - self.magnesium_nitrate_ratio))
+            
+            mg_remaining = self.target["Mg"] - self.actual["Mg"]
+            if mg_remaining > 0:
+                self._apply_fertilizer("Магний азотнокислый", "Mg", mg_remaining)
+        
+        # Балансируем калий и серу
         self._balance_k_s()
+        
+        # Распределяем кальций
         self._distribute_calcium()
+        
+        # Балансируем нитратный азот
         no3_needed = self.target["N (NO3-)"] - self.actual["N (NO3-)"]
         if no3_needed > 0:
             self._apply_fertilizer("Калий азотнокислый", "N (NO3-)", no3_needed)
+        
         return self._verify_results()
 
     def _apply_fertilizer(self, name, element, target_ppm):
@@ -764,6 +794,8 @@ class NutrientCalculator:
     def _balance_k_s(self):
         k_needed = self.target["K"] - self.actual["K"]
         s_needed = self.target["S"] - self.actual["S"]
+        
+        # Используем "Калий сернокислый" для баланса K и S
         if k_needed > 0 and s_needed > 0:
             k_fraction = self.fertilizers["Калий сернокислый"].get("K", 0)
             s_fraction = self.fertilizers["Калий сернокислый"].get("S", 0)
@@ -772,6 +804,8 @@ class NutrientCalculator:
                 return
             k_from_k2so4 = min(k_needed, s_needed * k_fraction / s_fraction)
             self._apply_fertilizer("Калий сернокисл��й", "K", k_from_k2so4)
+        
+        # Оставшийся калий добавляем через "Калий азотнокислый"
         remaining_k = self.target["K"] - self.actual["K"]
         if remaining_k > 0:
             self._apply_fertilizer("Калий азотнокислый", "K", remaining_k)
@@ -840,6 +874,7 @@ class NutrientCalculator:
                 report += f"\n- {el}: не хватает {abs(diff)} ppm"
         return report
 
+
 # Извлекаем данные из INPUT_DATA
 fertilizer_constants = INPUT_DATA["fertilizerConstants"]
 profile_settings = INPUT_DATA["profileSettings"]
@@ -847,6 +882,8 @@ liters = profile_settings["liters"]
 rounding_precision = 3
 activation_cacl = float(profile_settings.get("activation_cacl", 5))
 enhancement_cacl = float(profile_settings.get("enhancement_cacl", 0.1))
+use_magnesium_nitrate = int(profile_settings.get("use_magnesium_nitrate", 1))  # Флаг для "Магния азотнокислого"
+magnesium_nitrate_ratio = float(profile_settings.get("magnesium_nitrate_ratio", 0.3))  # Коэффициент для "Магния азотнокислого"
 
 # Создаем калькулятор
 calculator = NutrientCalculator(
@@ -855,7 +892,9 @@ calculator = NutrientCalculator(
     liters=liters,
     rounding_precision=rounding_precision,
     activation_cacl=activation_cacl,
-    enhancement_cacl=enhancement_cacl
+    enhancement_cacl=enhancement_cacl,
+    use_magnesium_nitrate=use_magnesium_nitrate,  # Новое название флага
+    magnesium_nitrate_ratio=magnesium_nitrate_ratio
 )
 
 # Запуск расчета