Update app.py

app.py

@@ -707,79 +707,73 @@ def nutri_call():
 
 
 
-
 
 
 from tabulate import tabulate
-import numpy as np
 
-# Константы
-TOTAL_NITROGEN = 120.0
-NO3_RATIO = 8.35
-NH4_RATIO = 1.0
-VOLUME_LITERS = 100
-
-# Целевой профиль (ppm)
-BASE_PROFILE = {
-    'P': 31.0, 'K': 210.0, 'Mg': 24.0,
-    'Ca': 84.0, 'S': 56.439,
-    'N (NO3-)': 0, 'N (NH4+)': 0  # Будет рассчитано
-}
-
-# Состав удобрений (доли)
-NUTRIENT_CONTENT = {
-    "Кальциевая селитра": {"N (NO3-)": 0.11863, "Ca": 0.16972},
-    "Калий азотнокислый": {"N (NO3-)": 0.136, "K": 0.382},
-    "Калий сернокислый": {"K": 0.44874, "S": 0.18401},
-    "Аммоний азотнокислый": {"N (NO3-)": 0.17499, "N (NH4+)": 0.17499},
-    "Сульфат магния": {"Mg": 0.09861, "S": 0.13010},
-    "Монофосфат калия": {"P": 0.218, "K": 0.275},
-    "Сульфат кальция": {"Ca": 0.23, "S": 0.186}
+# Входные данные
+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}
+    },
+    "profileSettings": {
+        "P": 60, "K": 194, "Mg": 48.5, "Ca": 121.25, "S": 79.445,
+        "NO3_RAT": 9.5, "TOTAL_NITROG": 138.57, "liters": 100
+    }
 }
 
-class PrecisionNutrientCalculator:
-    def __init__(self):
-        self.volume = VOLUME_LITERS
-        self.target = self._calculate_nitrogen_profile()
-        self.fertilizers = NUTRIENT_CONTENT
-        self.results = {}
-        self.actual = {k: 0.0 for k in self.target}
+class NutrientCalculator:
+    def __init__(self, input_data):
+        self.fertilizers = input_data["fertilizerConstants"]
+        self.profile = input_data["profileSettings"]
+        self.volume = self.profile["liters"]
+        
+        # Рассчитываем азотные компоненты
+        total_parts = self.profile["NO3_RAT"] + 1
+        self.target = {
+            'P': self.profile["P"],
+            'K': self.profile["K"],
+            'Mg': self.profile["Mg"],
+            'Ca': self.profile["Ca"],
+            'S': self.profile["S"],
+            'N (NO3-)': self.profile["TOTAL_NITROG"] * (self.profile["NO3_RAT"] / total_parts),
+            'N (NH4+)': self.profile["TOTAL_NITROG"] * (1 / total_parts)
+        }
         
-    def _calculate_nitrogen_profile(self):
-        """Расчет азотных компонентов"""
-        profile = BASE_PROFILE.copy()
-        total = NO3_RATIO + NH4_RATIO
-        profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total)
-        profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total)
-        return profile
+        self.actual = {k: 0.0 for k in self.target}
+        self.results = {}
 
     def calculate(self):
-        """Основной метод расчета с проверкой"""
         # 1. Вносим кальциевую селитру (Ca + NO3)
-        self._apply_fertilizer("Кальциевая селитра", "Ca", self.target['Ca'])
+        self._apply_fertilizer("Кальциевая селитра", "Ca", self.target["Ca"])
         
         # 2. Вносим аммонийный азот
-        self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target['N (NH4+)'])
+        self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target["N (NH4+)"])
         
         # 3. Компенсируем NO3 калийной селитрой
-        no3_needed = self.target['N (NO3-)'] - self.actual['N (NO3-)']
+        no3_needed = self.target["N (NO3-)"] - self.actual["N (NO3-)"]
         if no3_needed > 0:
             self._apply_fertilizer("Калий азотнокислый", "N (NO3-)", no3_needed)
         
         # 4. Вносим фосфор
-        self._apply_fertilizer("Монофосфат калия", "P", self.target['P'])
+        self._apply_fertilizer("Монофосфат калия", "P", self.target["P"])
         
         # 5. Вносим магний
-        self._apply_fertilizer("Сульфат магния", "Mg", self.target['Mg'])
+        self._apply_fertilizer("Сульфат магния", "Mg", self.target["Mg"])
         
         # 6. Балансируем калий и серу
         self._balance_k_s()
         
         return self._verify_results()
 
-    def _apply_fertilizer(self, name, main_element, target_ppm):
-        """Точное внесение удобрения для конкретного элемента"""
-        content = self.fertilizers[name][main_element]
+    def _apply_fertilizer(self, name, element, target_ppm):
+        """Применение удобрения с точным расчетом"""
+        content = self.fertilizers[name][element]
         grams = (target_ppm * self.volume) / (content * 1000)
         
         if name not in self.results:
@@ -787,68 +781,78 @@ class PrecisionNutrientCalculator:
         
         self.results[name]['граммы'] += grams
         
-        # Учитываем все элементы из удобрения
-        for element, percent in self.fertilizers[name].items():
-            added_ppm = (grams * percent * 1000) / self.volume
-            self.actual[element] += added_ppm
+        for el, val in self.fertilizers[name].items():
+            added_ppm = (grams * val * 1000) / self.volume
+            self.actual[el] += added_ppm
 
     def _balance_k_s(self):
-        """Балансировка калия и серы"""
-        k_needed = self.target['K'] - self.actual['K']
-        s_needed = self.target['S'] - self.actual['S']
+        """Интеллектуальная балансировка K и S"""
+        k_needed = self.target["K"] - self.actual["K"]
+        s_needed = self.target["S"] - self.actual["S"]
         
-        # Используем K2SO4 если нужна и сера и калий
+        # Используем K2SO4 если нужны оба элемента
         if k_needed > 0 and s_needed > 0:
             k_from_k2so4 = min(k_needed, s_needed * 0.44874/0.18401)
             self._apply_fertilizer("Калий сернокислый", "K", k_from_k2so4)
         
         # Добираем остаток калия KNO3
-        remaining_k = self.target['K'] - self.actual['K']
+        remaining_k = self.target["K"] - self.actual["K"]
         if remaining_k > 0:
             self._apply_fertilizer("Калий азотнокислый", "K", remaining_k)
 
     def _verify_results(self):
-        """Проверка соответствия целевому профилю"""
+        """Проверка точности соответствия"""
         deficits = {}
-        for element in self.target:
-            diff = self.target[element] - self.actual[element]
+        for el in self.target:
+            diff = self.target[el] - self.actual[el]
             if abs(diff) > 0.1:  # Допустимая погрешность 0.1 ppm
-                deficits[element] = round(diff, 3)
+                deficits[el] = round(diff, 3)
         
         return {
             'fertilizers': {k: round(v['граммы'], 3) for k, v in self.results.items()},
             'actual_profile': {k: round(v, 3) for k, v in self.actual.items()},
             'deficits': deficits,
-            'success': len(deficits) == 0
+            'total_ppm': round(sum(self.actual.values()), 3)
         }
 
     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()):
+            element_table.append([
+                el,
+                f"{self.target[el]} ppm",
+                f"{results['actual_profile'][el]} ppm",
+                f"{results['actual_profile'][el] - self.target[el]:+.2f} ppm"
+            ])
+        
         report = "РЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:\n"
-        for fert, data in results['fertilizers'].items():
-            report += f"{fert}: {data} г\n"
+        report += tabulate(fert_table, headers=["Удобрение", "Количество"], tablefmt="grid")
         
-        report += "\nФАКТИЧЕСКИЙ ПРОФИЛЬ (ppm):\n"
-        for element, value in results['actual_profile'].items():
-            report += f"{element}: {value} (цель: {self.target[element]})\n"
+        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 element, diff in results['deficits'].items():
-                report += f"{element}: не хватает {diff} ppm\n"
-        else:
-            report += "\nВсе элементы сбалансированы идеально!"
+            report += "\n\nВНИМАНИЕ: Обнаружены небольшие отклонения:"
+            for el, diff in results['deficits'].items():
+                report += f"\n- {el}: не хватает {abs(diff)} ppm"
         
         return report
 
-# Пример использования
-if __name__ == "__main__":
-    calculator = PrecisionNutrientCalculator()
-    results = calculator.calculate()
-    print(calculator.generate_report(results))
-
-
-
+# Запуск расчета
+calculator = NutrientCalculator(INPUT_DATA)
+results = calculator.calculate()
+print(calculator.generate_report(results))
 
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():