Update app.py

app.py

@@ -735,6 +735,13 @@ class NutrientCalculator:
             "NH4+": self.target_profile['N (NH4+)']
         }
 
+        # Настройки компенсации по умолчанию
+        self.compensation_weights = {
+            'KNO3': 0.5,  # Вес калийной селитры (0-1)
+            'CaNO3': 0.3, # Вес кальциевой селитры (0-1)
+            'K2SO4': 0.2  # Вес сульфата калия (0-1)
+        }
+
     def _label(self, element):
         """Форматирование названий элементов для вывода"""
         labels = {
@@ -743,39 +750,44 @@ class NutrientCalculator:
         }
         return labels.get(element, element)
 
-    def _balance_nitrogen_with_k(self):
-        """Новый метод с компенсацией калия"""
-        # 1. Вносим NH4 через аммиачную селитру
-        nh4_needed = self.target_profile['N (NH4+)'] - self.actual_profile['N (NH4+)']
-        if nh4_needed > 0.1:
-            self._apply("Аммоний азотнокислый", "N (NH4+)", nh4_needed)
-        
-        # 2. Рассчитываем сколько NO3 нужно внести
-        no3_needed = self.target_profile['N (NO3-)'] - self.actual_profile['N (NO3-)']
-        
-        # 3. Сначала вносим через кальциевую селитру (для Ca)
-        ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
-        if ca_needed > 0.1 and no3_needed > 0.1:
-            max_ca_no3 = (ca_needed / self.fertilizers["Кальциевая селитра"]["Ca"]) * \
-                        self.fertilizers["Кальциевая селитра"]["N (NO3-)"]
-            ca_no3_part = min(no3_needed, max_ca_no3)
-            self._apply("Кальциевая селитра", "N (NO3-)", ca_no3_part)
-            no3_needed -= ca_no3_part
-        
-        # 4. Остаток NO3 вносим через калийную селитру
-        if no3_needed > 0.1:
-            # Рассчитываем сколько K внесется с этим NO3
-            k_from_kno3 = (no3_needed / self.fertilizers["Калий азотнокислый"]["N (NO3-)"]) * \
-                         self.fertilizers["Калий азотнокислый"]["K"]
+    def set_compensation_weights(self, kno3_weight, cano3_weight, k2so4_weight):
+        """Установка весов для компенсации элементов"""
+        total = kno3_weight + cano3_weight + k2so4_weight
+        self.compensation_weights = {
+            'KNO3': kno3_weight / total,
+            'CaNO3': cano3_weight / total,
+            'K2SO4': k2so4_weight / total
+        }
+
+    def calculate(self):
+        """Основной метод расчета с новой логикой"""
+        try:
+            # 1. Вносим Mg и S
+            self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
+            
+            # 2. Балансируем азот с учетом компенсации
+            self._balance_nitrogen_with_compensation()
+            
+            # 3. Вносим Ca (остаток)
+            ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
+            if ca_needed > 0.1:
+                self._apply("Кальциевая селитра", "Ca", ca_needed)
+            
+            # 4. Вносим P
+            p_needed = self.target_profile['P'] - self.actual_profile['P']
+            if p_needed > 0.1:
+                self._apply("Монофосфат калия", "P", p_needed)
+            
+            # 5. Корректируем K через сульфат калия (если остался дефицит)
+            k_needed = self.target_profile['K'] - self.actual_profile['K']
+            if k_needed > 0.1:
+                self._apply("Калий сернокислый", "K", k_needed)
             
-            # Проверяем, не превысит ли это целевое значение K
-            k_remaining = self.target_profile['K'] - self.actual_profile['K']
-            if k_from_kno3 > k_remaining:
-                # Если превысит - уменьшаем количество KNO3
-                no3_needed = (k_remaining / self.fertilizers["Калий азотнокислый"]["K"]) * \
-                            self.fertilizers["Калий азотнокислый"]["N (NO3-)"]
+            return self.results
             
-            self._apply("Калий азотнокислый", "N (NO3-)", no3_needed)
+        except Exception as e:
+            print(f"Ошибка при расчёте: {str(e)}")
+            raise
 
     def calculate(self):
         """Основной метод расчета с новой логикой"""