Update app.py

app.py

@@ -693,8 +693,8 @@ from tabulate import tabulate
 
 INPUT_DATA = {
     "fertilizerConstants": {
-        "Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca_nitrate": 0.16972},  # Только нитратный кальций
-        "Хелат кальция": {"Ca_chelate": 0.10},  # Только хелатный кальций
+        "Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca_nitrate": 0.16972},
+        "Хелат кальция": {"Ca_chelate": 0.10},
         "Калий азотнокислый": {"N (NO3-)": 0.13854, "K": 0.36672},
         "Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499},
         "Сульфат магния": {"Mg": 0.10220, "S": 0.13483},
@@ -703,8 +703,8 @@ INPUT_DATA = {
     },
     "profileSettings": {
         "P": 31, "K": 210, "Mg": 24, "S": 57.5,
-        "Ca_nitrate_target": 62,  # Чётко 62 ppm из селитры
-        "Ca_chelate_target": 20,  # Чётко 20 ppm из хелата
+        "Ca_nitrate_target": 62,
+        "Ca_chelate_target": 20,
         "NO3_RAT": 8.25, "TOTAL_NITROG": 125, "liters": 100
     }
 }
@@ -722,8 +722,8 @@ class NutrientCalculator:
             'K': self.profile["K"],
             'Mg': self.profile["Mg"],
             'S': self.profile["S"],
-            'Ca_nitrate': self.profile["Ca_nitrate_target"],  # 62 ppm
-            'Ca_chelate': self.profile["Ca_chelate_target"],  # 20 ppm
+            'Ca_nitrate': self.profile["Ca_nitrate_target"],
+            'Ca_chelate': self.profile["Ca_chelate_target"],
             'N (NO3-)': self.profile["TOTAL_NITROG"] * (self.profile["NO3_RAT"] / total_parts),
             'N (NH4+)': self.profile["TOTAL_NITROG"] * (1 / total_parts)
         }
@@ -735,6 +735,48 @@ class NutrientCalculator:
         }
         self.results = {fert: {'граммы': 0.0} for fert in self.fertilizers}
 
+    def _apply_fertilizer(self, name, element, target_ppm):
+        if name not in self.fertilizers:
+            raise KeyError(f"Удобрение '{name}' не найдено!")
+        content = self.fertilizers[name].get(element, 0)
+        if content == 0:
+            print(f"ПРЕДУПРЕЖДЕНИЕ: Удобрение '{name}' не содержит элемент '{element}'")
+            return
+        grams = (target_ppm * self.volume) / (content * 1000)
+        self.results[name]['граммы'] += round(grams, self.rounding_precision)
+        for el, val in self.fertilizers[name].items():
+            if el in self.actual:
+                added_ppm = (grams * val * 1000) / self.volume
+                self.actual[el] += round(added_ppm, self.rounding_precision)
+
+    def _balance_k_s(self):
+        k_needed = self.target["K"] - self.actual["K"]
+        s_needed = self.target["S"] - self.actual["S"]
+        
+        if k_needed > 0 and s_needed > 0:
+            k_fraction = self.fertilizers["Калий сернокислый"].get("K", 0)
+            s_fraction = self.fertilizers["Калий сернокислый"].get("S", 0)
+            if k_fraction == 0 or s_fraction == 0:
+                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)
+
+    def _verify_results(self):
+        deficits = {}
+        for el in self.target:
+            diff = self.target[el] - self.actual[el]
+            if abs(diff) > 0.1:
+                deficits[el] = round(diff, self.rounding_precision)
+        return {
+            'fertilizers': {k: round(v['граммы'], self.rounding_precision) for k, v in self.results.items() if v['граммы'] > 0},
+            'actual_profile': {k: round(v, self.rounding_precision) for k, v in self.actual.items()},
+            'deficits': deficits
+        }
+
     def calculate(self):
         # 1. Вносим хелатный кальций - строго 20 ppm
         self._apply_fertilizer("Хелат кальция", "Ca_chelate", self.target["Ca_chelate"])
@@ -742,28 +784,24 @@ class NutrientCalculator:
         # 2. Вносим нитратный кальций - строго 62 ppm
         self._apply_fertilizer("Кальциевая селитра", "Ca_nitrate", self.target["Ca_nitrate"])
         
-        # 3. Остальные элементы (как было)
+        # 3. Остальные элементы
         self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target["N (NH4+)"])
         self._apply_fertilizer("Монофосфат калия", "P", self.target["P"])
         self._apply_fertilizer("Сульфат магния", "Mg", self.target["Mg"])
         self._balance_k_s()
         
-        # 4. Балансируем нитратный азот (учитываем уже внесённый из кальц��евой селитры)
+        # 4. Балансируем нитратный азот
         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 generate_report(self, results):
         fert_table = []
         for name, data in results['fertilizers'].items():
-            if data['граммы'] > 0:  # Показываем только используемые удобрения
-                fert_table.append([name, f"{data['граммы']} г"])
+            fert_table.append([name, f"{data} г"])
         
-        # Добавляем строку с суммарным кальцием
         total_ca = self.actual['Ca_nitrate'] + self.actual['Ca_chelate']
         ca_diff = total_ca - (self.target['Ca_nitrate'] + self.target['Ca_chelate'])
         
@@ -785,9 +823,13 @@ class NutrientCalculator:
         report += tabulate(element_table, 
                          headers=["Элемент", "Цель", "Факт", "Отклонение"],
                          tablefmt="grid")
+        if results['deficits']:
+            report += "\n\nВНИМАНИЕ: Обнаружены небольшие отклонения:"
+            for el, diff in results['deficits'].items():
+                report += f"\n- {el}: не хватает {abs(diff)} ppm"
         return report
 
-# Использование
+# Запуск расчета
 calculator = NutrientCalculator(
     fertilizer_constants=INPUT_DATA["fertilizerConstants"],
     profile_settings=INPUT_DATA["profileSettings"],