Update app.py

app.py

@@ -727,12 +727,12 @@ EC_COEFFICIENTS = {
 }
 
 class NutrientCalculator:
-    def __init__(self, volume_liters=1.0, profile=BASE_PROFILE):
+    def __init__(self, volume_liters=1.0, profile=None):
         self.volume = volume_liters
         self.results = {}
-        self.target_profile = profile.copy()
+        self.target_profile = profile.copy() if profile else BASE_PROFILE.copy()
         self.actual_profile = {k: 0.0 for k in self.target_profile}
-        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
+        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS
         self.total_ec = 0.0
 
         # Расчёт азота
@@ -752,7 +752,7 @@ class NutrientCalculator:
             "K": {"weight": 0.2, "fert": "Калий азотнокислый", "main_element": "K"},
             "Mg": {"weight": 0.2, "fert": "Сульфат магния", "main_element": "Mg"},
             "P": {"weight": 0.1, "fert": "Монофосфат калия", "main_element": "P"},
-            "S": {"weight": 0.1, "fert": "Калий сернокислый", "main_element": "S"},
+            "S": {"weight": 0.1, "fert": "Сульфат кальция", "main_element": "S"},
             "N (NO3-)": {"weight": 0.05, "fert": "Калий азотнокислый", "main_element": "N (NO3-)"},
             "N (NH4+)": {"weight": 0.05, "fert": "Аммоний азотнокислый", "main_element": "N (NH4+)"}
         }
@@ -767,23 +767,45 @@ class NutrientCalculator:
 
     def calculate(self):
         try:
-            # Первый проход: компенсация основных элементов
-            self._compensate_main_elements()
+            max_iterations = 10  # Максимальное количество итераций для корректировки
+            iteration = 0
+            while iteration < max_iterations:
+                self._reset_actual_profile()
+                self._compensate_main_elements()
+                self._compensate_nitrogen()
+                self._compensate_secondary_elements()
 
-            # Второй проход: компенсация азота
-            self._compensate_nitrogen()
+                # Проверка результатов
+                if self.validate_results():
+                    break
 
-            # Третий проход: компенсация второстепенных элементов
-            self._compensate_secondary_elements()
+                # Корректировка перебора
+                self._adjust_overages()
+                iteration += 1
 
-            # Четвертый проход: корректировка перебора
-            self._adjust_overages()
+            if iteration == max_iterations:
+                raise ValueError("Не удалось достичь целевых значений после максимального количества итераций.")
 
             return self.results
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
+    def validate_results(self):
+        """Проверка соответствия целевым значениям"""
+        for element, target_value in self.target_profile.items():
+            actual_value = self.actual_profile[element]
+            if abs(actual_value - target_value) > 0.1:  # Допустимая погрешность
+                print(f"Несоответствие: {element} (цель: {target_value:.2f}, фактически: {actual_value:.2f})")
+                return False
+        return True
+
+    def _reset_actual_profile(self):
+        """Сброс фактического профиля перед новой итерацией"""
+        self.actual_profile = {k: 0.0 for k in self.target_profile}
+        self.total_ec = 0.0
+        self.results = {}
+
     def _compensate_main_elements(self):
         """Компенсация основных элементов (Ca, Mg, P)"""
         for element, weight_data in self.compensation_weights.items():