Update app.py

app.py

@@ -694,38 +694,16 @@ def nutri_call():
 
 
 
-
-from tabulate import tabulate
-
-# Входные данные (пример)
-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},
-        "Монофосфат калия": {"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": 3  # Добавлено
-    }
-}
-
 class NutrientCalculator:
     def __init__(self, fertilizer_constants, profile_settings, liters, rounding_precision, activation_cacl, enhancement_cacl):
         self.fertilizers = fertilizer_constants
         self.profile = profile_settings
         self.volume = liters
-        self.rounding_precision = rounding_precision  # Точность округления
+        self.rounding_precision = rounding_precision
         
         # Новые параметры
-        self.activation_cacl = activation_cacl  # Активация CaCl
-        self.enhancement_cacl = enhancement_cacl  # Усиление CaCl
+        self.activation_cacl = float(activation_cacl)  # Активация CaCl
+        self.enhancement_cacl = float(enhancement_cacl)  # Усиление CaCl
         
         total_parts = self.profile["NO3_RAT"] + 1
         self.target = {
@@ -742,114 +720,80 @@ class NutrientCalculator:
         self.results = {fert: {'граммы': 0.0} for fert in self.fertilizers}
 
     def calculate(self):
-        # 1. Вносим аммонийный азот
         self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target["N (NH4+)"])
-        
-        # 2. Вносим фосфор
         self._apply_fertilizer("Монофосфат калия", "P", self.target["P"])
-        
-        # 3. Вносим магний
         self._apply_fertilizer("Сульфат магния", "Mg", self.target["Mg"])
-        
-        # 4. Балансируем калий и серу
         self._balance_k_s()
-        
-        # 5. Распределяем кальций между селитрой и хлоридом
         self._distribute_calcium()
-        
-        # 6. Компенсируем NO3 калийной селитрой
         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):
         if name not in self.fertilizers:
-            raise KeyError(f"Удобрение '{name}' не найдено в fertilizerConstants!")
-        
-        content = self.fertilizers[name].get(element, 0)  # Получаем содержание элемента
+            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)  # Округляем до заданной точности
-        
+        self.results[name]['граммы'] += round(grams, self.rounding_precision)
         for el, val in self.fertilizers[name].items():
             added_ppm = (grams * val * 1000) / self.volume
             if el in self.actual:
                 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:
-            # Берем динамические значения из fertilizerConstants
             k_fraction = self.fertilizers["Калий сернокислый"].get("K", 0)
             s_fraction = self.fertilizers["Калий сернокислый"].get("S", 0)
-
             if k_fraction == 0 or s_fraction == 0:
-                print("ПРЕДУПРЕЖДЕНИЕ: Удобрение 'Калий сернокислый' содержит нулевые значения для K или S!")
+                print("ПРЕДУПРЕЖДЕНИЕ: Удобрение 'Калий сернокислый' содержит нулевые значения!")
                 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 _distribute_calcium(self):
-        ca_target = self.target["Ca"]  # Общий кальций
+        ca_target = self.target["Ca"]
         no3_ratio = self.profile["NO3_RAT"]
-    
-        # Проверка соотношения NO3/NH4
-        if no3_ratio >= self.activation_cacl:  # Заменили "5" на self.activation_cacl
+        if no3_ratio >= self.activation_cacl:
             print(f"Соотношение NO3/NH4 >= {self.activation_cacl}. Кальций хлористый не добавляется.")
             remaining_ca = ca_target
         else:
-            # Определяем, сколько кальция взять из хлористого кальция
-            cacl_ratio = self.enhancement_cacl  # Заменили "0.1" на self.enhancement_cacl
+            cacl_ratio = self.enhancement_cacl
             cacl_target = ca_target * cacl_ratio
-        
-            # Вносим кальций из хлористого кальция
             if cacl_target > 0:
                 self._apply_fertilizer("Кальций хлорид", "Ca", cacl_target)
-        
-            # Оставшийся кальций берем из кальциевой селитры
             remaining_ca = ca_target - cacl_target
-    
-        # Добавляем оставшийся кальций из кальциевой селитры
         if remaining_ca > 0:
             self._apply_fertilizer("Кальциевая селитра", "Ca", remaining_ca)
 
-        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()},
-                'actual_profile': {k: round(v, self.rounding_precision) for k, v in self.actual.items()},
-                'deficits': deficits,
-                'total_ppm': round(sum(self.actual.values()), self.rounding_precision)
-                 }
+    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()},
+            'actual_profile': {k: round(v, self.rounding_precision) for k, v in self.actual.items()},
+            'deficits': deficits,
+            'total_ppm': round(sum(self.actual.values()), self.rounding_precision)
+        }
 
     def generate_report(self, results):
-        """Генерация текстового отчета"""
         fert_table = []
         for name, data in results['fertilizers'].items():
             fert_table.append([name, f"{data} г"])
-        
         element_table = []
         for el in sorted(self.target.keys()):
-            if el == "CaCl":  # Пропускаем CaCl в отчете
+            if el == "CaCl":
                 continue
             element_table.append([
                 el,
@@ -857,31 +801,26 @@ class NutrientCalculator:
                 f"{results['actual_profile'][el]} ppm",
                 f"{results['actual_profile'][el] - self.target[el]:+.2f} ppm"
             ])
-        
         report = "РЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:\n"
         report += tabulate(fert_table, headers=["Удобрение", "Количество"], tablefmt="grid")
-        
         report += "\n\nБАЛАНС ЭЛЕМЕНТОВ:\n"
         report += tabulate(element_table, 
                          headers=["Элемент", "Цель", "Факт", "Отклонение"],
                          tablefmt="grid")
-        
         report += f"\n\nОбщая концентрация: {results['total_ppm']} ppm"
-        
         if results['deficits']:
             report += "\n\nВНИМАНИЕ: Обнаружены небольшие отклонения:"
             for el, diff in results['deficits'].items():
                 report += f"\n- {el}: не хватает {abs(diff)} ppm"
-        
         return report
 
 # Извлекаем данные из INPUT_DATA
 fertilizer_constants = INPUT_DATA["fertilizerConstants"]
 profile_settings = INPUT_DATA["profileSettings"]
 liters = profile_settings["liters"]
-rounding_precision = 3  # Значение по умолчанию
-activation_cacl = profile_settings.get("activation_cacl", 0)  # По умолчанию 0
-enhancement_cacl = profile_settings.get("enhancement_cacl", 0)  # По умолчанию 0
+rounding_precision = 3
+activation_cacl = float(profile_settings.get("activation_cacl", 5))
+enhancement_cacl = float(profile_settings.get("enhancement_cacl", 0.1))
 
 # Создаем калькулятор
 calculator = NutrientCalculator(
@@ -893,6 +832,10 @@ calculator = NutrientCalculator(
     enhancement_cacl=enhancement_cacl
 )
 
+# Запуск расчета
+results = calculator.calculate()
+print(calculator.generate_report(results))
+
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try: