Update app.py

app.py

@@ -758,6 +758,41 @@ NUTRIENT_CONTENT_IN_FERTILIZERS = {
 
 
 
+from tabulate import tabulate
+
+# Константы
+TOTAL_NITROGEN = 125.000
+NO3_RATIO = 8.25
+NH4_RATIO = 1.00
+VOLUME_LITERS = 100
+
+# Коэффициенты электропроводности
+EC_COEFFICIENTS = {
+    'P': 0.0012, 'K': 0.0018, 'Mg': 0.0015, 
+    'Ca': 0.0016, 'S': 0.0014, 
+    'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
+}
+
+# Целевые значения
+BASE_PROFILE = {
+    'P': 31.000, 'K': 210.000, 'Mg': 24.000,
+    'Ca': 84.000, 'S': 56.439,
+    'N (NO3-)': 0, 'N (NH4+)': 0
+}
+
+NUTRIENT_CONTENT_IN_FERTILIZERS = {
+    "Кальциевая селитра": {"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}
+}
+
+
+
+
+
 from tabulate import tabulate
 
 # Константы
@@ -832,18 +867,12 @@ class NutrientCalculator:
         self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
         self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
         self.total_ec = 0.0
-        
-        # Веса компенсации (только для элементов, которые компенсируются)
+
+        # Веса компенсации для элементов (кроме азотов)
         self.element_compensation_weights = {
-            "POTASSIUM_SULFATE": {"weight": 0.18, "fert": "Калий сернокислый", "main_element": "K"},
-            "MAGNESIUM_SULFATE": {"weight": 0.1, "fert": "Сульфат магния", "main_element": "Mg"},
-            "MONOPOTASSIUM_PHOSPHATE": {"weight": 0.14, "fert": "Монофосфат калия", "main_element": "P"}
-        }
-        
-        # Коэффициенты распределения для кальциевой селитры
-        self.calcium_nitrate_ratios = {
-            "Ca": 0.7,  # 70% дефицита кальция покрываем кальциевой селитрой
-            "NO3": 0.3  # 30% дефицита NO3 покрываем кальциевой селитрой
+            "POTASSIUM_SULFATE": {"weight": -0.18, "fert": "Калий сернокислый", "main_element": "K"},
+            "MAGNESIUM_SULFATE": {"weight": -0.1, "fert": "Сульфат магния", "main_element": "Mg"},
+            "MONOPOTASSIUM_PHOSPHATE": {"weight": -0.14, "fert": "Монофосфат калия", "main_element": "P"}
         }
 
         # Расчёт азота
@@ -851,21 +880,38 @@ class NutrientCalculator:
         self.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
         self.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
 
+    def _label(self, element):
+        """Форматирование названий элементов для вывода"""
+        labels = {
+            'N (NO3-)': 'NO3',
+            'N (NH4+)': 'NH4'
+        }
+        return labels.get(element, element)
+
     def calculate(self):
         try:
-            # 1. Вносим кальциевую селитру для Ca и частично для NO3
-            self._apply_calcium_nitrate()
-            
-            # 2. Вносим аммонийный азот
-            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
-            
-            # 3. Компенсируем остаток NO3 калийной селитрой
-            self._compensate_no3()
-            
-            # 4. Вносим остальные элементы через компенсацию
+            # Вносим магний через компенсацию
             self._compensate_element("Mg")
+
+            # Вносим кальций напрямую
+            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+
+            # Вносим фосфор через компенсацию
             self._compensate_element("P")
+
+            # Вносим аммонийный азот напрямую
+            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+
+            # Компенсируем нитратный азот напрямую
+            current_no3 = self.actual_profile['N (NO3-)']
+            no3_needed = self.target_profile['N (NO3-)'] - current_no3
+            if no3_needed > 0.1:
+                self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+
+            # Компенсируем серу через компенсацию
             self._compensate_element("S")
+
+            # Компенсируем калий через компенсацию
             self._compensate_element("K")
 
             return self.results
@@ -873,53 +919,6 @@ class NutrientCalculator:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
-    def _apply_calcium_nitrate(self):
-        """Специальный метод для внесения кальциевой селитры"""
-        # Вносим кальций
-        ca_needed = self.target_profile['Ca'] - self.actual_profile.get('Ca', 0)
-        if ca_needed > 0.1:
-            ca_share = self.calcium_nitrate_ratios['Ca']
-            ca_to_apply = ca_needed * ca_share
-            self._apply("Кальциевая селитра", "Ca", ca_to_apply)
-        
-        # Вносим часть NO3 из кальциевой селитры
-        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
-        if no3_needed > 0.1:
-            no3_share = self.calcium_nitrate_ratios['NO3']
-            no3_to_apply = no3_needed * no3_share
-            self._apply("Кальциевая селитра", "N (NO3-)", no3_to_apply)
-
-    def _compensate_no3(self):
-        """Компенсация остатка NO3 после внесения кальциевой селитры"""
-        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile.get('N (NO3-)', 0)
-        if no3_needed > 0.1:
-            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
-
-    def _compensate_element(self, element):
-        """Общий метод компенсации для других элементов"""
-        needed = self.target_profile[element] - self.actual_profile.get(element, 0)
-        if abs(needed) < 0.1:
-            return
-
-        candidates = []
-        for weight_key, weight_data in self.element_compensation_weights.items():
-            fert_name = weight_data["fert"]
-            if element in self.fertilizers.get(fert_name, {}):
-                candidates.append({
-                    'name': fert_name,
-                    'weight': weight_data["weight"],
-                    'content': self.fertilizers[fert_name][element]
-                })
-
-        if not candidates:
-            raise ValueError(f"Нет удобрений для элемента {element}")
-
-        total_weight = sum(c['weight'] for c in candidates)
-        for candidate in candidates:
-            share = candidate['weight'] / total_weight
-            ppm_to_apply = needed * share
-            self._apply(candidate['name'], element, ppm_to_apply)
-
     def _apply(self, fert_name, main_element, required_ppm):
         if required_ppm <= 0:
             return
@@ -956,56 +955,28 @@ class NutrientCalculator:
 
     def _compensate_element(self, element):
         needed = self.target_profile[element] - self.actual_profile.get(element, 0)
-        if abs(needed) < 0.1:  # Игнорируем незначительные отклонения
+        if needed <= 0:
             return
 
         candidates = []
+        for weight_key, weight_data in self.element_compensation_weights.items():
+            fert_name = weight_data["fert"]
+            if element in self.fertilizers.get(fert_name, {}):
+                candidates.append({
+                    'name': fert_name,
+                    'weight': weight_data["weight"],
+                    'content': self.fertilizers[fert_name][element]
+                })
 
-        # Для кальциевой селитры особый случай
-        if element == "Ca":
-            candidates.append({
-                'name': "Кальциевая селитра",
-                'weight': 0.7,  # Основной вес для кальция
-                'content': self.fertilizers["Кальциевая селитра"]["Ca"]
-            })
-        elif element == "N (NO3-)":
-            candidates.append({
-                'name': "Кальциевая селитра",
-                'weight': 0.3,  # Дополнительный вес для NO3 из кальциевой селитры
-                'content': self.fertilizers["Кальциевая селитра"]["N (NO3-)"]
-            })
-            candidates.append({
-                'name': "Калий азотнокислый",
-                'weight': 0.7,  # Основной вес для NO3
-                'content': self.fertilizers["Калий азотнокислый"]["N (NO3-)"]
-            })
-        else:
-            # Стандартная компенсация для других элементов
-            for weight_key, weight_data in self.element_compensation_weights.items():
-                fert_name = weight_data["fert"]
-                if element in self.fertilizers.get(fert_name, {}):
-                    candidates.append({
-                        'name': fert_name,
-                        'weight': weight_data["weight"],
-                        'content': self.fertilizers[fert_name][element]
-                    })
-
-        # Проверяем, есть ли кандидаты
         if not candidates:
             raise ValueError(f"Нет удобрений для элемента {element}")
 
-        # Суммируем только положительные веса
-        total_weight = sum(max(c['weight'], 0) for c in candidates)
-        if total_weight == 0:
-            raise ValueError(f"Общий вес для элемента {element} равен нулю")
-
-        # Распределяем доли
+        total_weight = sum(c['weight'] for c in candidates)
         for candidate in candidates:
-            share = max(candidate['weight'], 0) / total_weight
+            share = candidate['weight'] / total_weight
             ppm_to_apply = needed * share
             self._apply(candidate['name'], element, ppm_to_apply)
 
-
     def calculate_ec(self):
         return round(self.total_ec, 2)