Update app.py

app.py

@@ -711,8 +711,32 @@ INPUT_DATA = {
 
 
 
+# Входные данные (пример)
+INPUT_DATA = {
+    "fertilizerConstants": {
+        "Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca": 0.16972},
+        "Калий азотнокислый": {"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}
+    },
+    "profileSettings": {
+        "P": 60, "K": 194, "Mg": 48.5, "Ca": 89.25, "S": 79.445, "CaCl": 38.5,
+        "NO3_RAT": 1.5, "TOTAL_NITROG": 138.57, "liters": 100,
+        "activation_cacl": 5,  # Добавлено
+        "enhancement_cacl": 0.1,  # Добавлено
+        "use_magnesium_sulfate": 1,  # Флаг для использования "Сульфата магния"
+        "magnesium_nitrate_ratio": 0.2  # Коэффициент для "Магния азотнокислого"
+    }
+}
+
+from tabulate import tabulate
+
 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_sulfate, magnesium_nitrate_ratio):
         self.fertilizers = fertilizer_constants
         self.profile = profile_settings
         self.volume = liters
@@ -721,6 +745,8 @@ class NutrientCalculator:
         # Новые параметры
         self.activation_cacl = float(activation_cacl)  # Активация CaCl
         self.enhancement_cacl = float(enhancement_cacl)  # Усиление CaCl
+        self.use_magnesium_sulfate = int(use_magnesium_sulfate)  # Флаг для использования "Сульфата магния"
+        self.magnesium_nitrate_ratio = float(magnesium_nitrate_ratio)  # Коэффициент для "Магния азотнокислого"
         
         total_parts = self.profile["NO3_RAT"] + 1
         self.target = {
@@ -737,14 +763,35 @@ 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_sulfate == 1:
+            mg_needed = self.target["Mg"] - self.actual["Mg"]
+            if mg_needed > 0:
+                self._apply_fertilizer("Сульфат магния", "Mg", mg_needed * (1 - self.magnesium_nitrate_ratio))
+        
+        # Добавляем магний через "Магний азотнокислый", если коэффициент > 0
+        if self.magnesium_nitrate_ratio > 0:
+            mg_remaining = self.target["Mg"] - self.actual["Mg"]
+            if mg_remaining > 0:
+                self._apply_fertilizer("Магний азотнокислый", "Mg", mg_remaining * self.magnesium_nitrate_ratio)
+        
+        # Балансируем калий и серу
         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):
@@ -847,6 +894,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_sulfate = int(profile_settings.get("use_magnesium_sulfate", 1))  # Флаг для "Сульфата магния"
+magnesium_nitrate_ratio = float(profile_settings.get("magnesium_nitrate_ratio", 0.2))  # Коэффициент для "Магния азотнокислого"
 
 # Создаем калькулятор
 calculator = NutrientCalculator(
@@ -855,7 +904,9 @@ calculator = NutrientCalculator(
     liters=liters,
     rounding_precision=rounding_precision,
     activation_cacl=activation_cacl,
-    enhancement_cacl=enhancement_cacl
+    enhancement_cacl=enhancement_cacl,
+    use_magnesium_sulfate=use_magnesium_sulfate,
+    magnesium_nitrate_ratio=magnesium_nitrate_ratio
 )
 
 # Запуск расчета