Update app.py

app.py

@@ -826,40 +826,24 @@ NUTRIENT_CONTENT_IN_FERTILIZERS = {
 
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
-        self.volume = volume_liters
-        self.results = {}
-        self.target_profile = BASE_PROFILE.copy()
-        self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
-        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
-        self.total_ec = 0.0
-
-        # Веса компенсации для элементов (кроме азотов)
+        # ... остальной код инициализации ...
+        
+        # Веса компенсации с добавлением кальциевой селитры для NO3
         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"}
+            "MONOPOTASSIUM_PHOSPHATE": {"weight": -0.14, "fert": "Монофосфат калия", "main_element": "P"},
+            "CALCIUM_NITRATE": {"weight": 0.25, "fert": "Кальциевая селитра", "main_element": "Ca"},
+            "CALCIUM_NITRATE_NO3": {"weight": 0.15, "fert": "Кальциевая селитра", "main_element": "N (NO3-)"}
         }
 
-        # Расчёт азота
-        total_parts = NO3_RATIO + NH4_RATIO
-        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:
             # Вносим магний через компенсацию
             self._compensate_element("Mg")
 
-            # Вносим кальций напрямую
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+            # Теперь вносим кальций через компенсацию, а не напрямую
+            self._compensate_element("Ca")
 
             # Вносим фосфор через компенсацию
             self._compensate_element("P")
@@ -867,11 +851,8 @@ class NutrientCalculator:
             # Вносим аммонийный азот напрямую
             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("N (NO3-)")
 
             # Компенсируем серу через компенсацию
             self._compensate_element("S")
@@ -884,61 +865,48 @@ class NutrientCalculator:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
-    def _apply(self, fert_name, main_element, required_ppm):
-        if required_ppm <= 0:
-            return
-
-        try:
-            content = self.fertilizers[fert_name][main_element]
-            grams = (required_ppm * self.volume) / (content * 1000)
-
-            if fert_name not in self.results:
-                result = {
-                    'граммы': 0.0,
-                    'миллиграммы': 0,
-                    'вклад в EC': 0.0
-                }
-                for element in self.fertilizers[fert_name]:
-                    result[f'внесет {self._label(element)}'] = 0.0
-                self.results[fert_name] = result
-
-            self.results[fert_name]['граммы'] += grams
-            self.results[fert_name]['миллиграммы'] += int(grams * 1000)
-
-            fert_ec = 0.0
-            for element, percent in self.fertilizers[fert_name].items():
-                added_ppm = (grams * percent * 1000) / self.volume
-                self.results[fert_name][f'внесет {self._label(element)}'] += added_ppm
-                self.actual_profile[element] += added_ppm
-                fert_ec += added_ppm * EC_COEFFICIENTS.get(element, 0.0015)
-
-            self.results[fert_name]['вклад в EC'] += fert_ec
-            self.total_ec += fert_ec
-        except KeyError as e:
-            print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
-            raise
-
     def _compensate_element(self, element):
         needed = self.target_profile[element] - self.actual_profile.get(element, 0)
-        if needed <= 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 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(c['weight'] for c in candidates)
+        total_weight = sum(abs(c['weight']) for c in candidates)
         for candidate in candidates:
-            share = candidate['weight'] / total_weight
+            share = abs(candidate['weight']) / total_weight
             ppm_to_apply = needed * share
             self._apply(candidate['name'], element, ppm_to_apply)