Update app.py

app.py

@@ -712,11 +712,14 @@ EC_COEFFICIENTS = {
     '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
+    "P": 60,
+    "K": 300,
+    "Mg": 70,
+    "Ca": 150,
+    "S": 120,
+    "N (NO3-)": 175,
+    "N (NH4+)": 15
 }
 
 NUTRIENT_CONTENT_IN_FERTILIZERS = {
@@ -726,9 +729,20 @@ NUTRIENT_CONTENT_IN_FERTILIZERS = {
     "Аммоний азотнокислый": {"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}  # Новый элемент
+    "Сульфат кальция": {"Ca": 0.23, "S": 0.186}
 }
 
+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
+}
+
+# Глобальные константы для азота
+TOTAL_NITROGEN = 190  # Общее количество азота
+NO3_RATIO = 7         # Соотношение NO3:NH4
+NH4_RATIO = 1
+
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
@@ -749,6 +763,17 @@ class NutrientCalculator:
             "NH4+": self.target_profile['N (NH4+)']
         }
 
+        # Веса компенсации
+        self.compensation_weights = {
+            "Ca": {"weight": 0.3, "fert": "Сульфат кальция", "main_element": "Ca"},
+            "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"},
+            "N (NO3-)": {"weight": 0.05, "fert": "Калий азотнокислый", "main_element": "N (NO3-)"},
+            "N (NH4+)": {"weight": 0.05, "fert": "Аммоний азотнокислый", "main_element": "N (NH4+)"}
+        }
+
     def _label(self, element):
         """Форматирование названий элементов для вывода"""
         labels = {
@@ -759,43 +784,48 @@ class NutrientCalculator:
 
     def calculate(self):
         try:
-            # 1. Вносим сульфат кальция для Ca
-            ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
-            if ca_needed > 0.1:
-                self._apply("Сульфат кальция", "Ca", ca_needed)
-
-            # 2. Вносим кальциевую селитру для Ca и NO3
-            ca_needed_after_sulfate = self.target_profile['Ca'] - self.actual_profile['Ca']
-            if ca_needed_after_sulfate > 0.1:
-                self._apply("Кальциевая селитра", "Ca", ca_needed_after_sulfate)
-
-            # 3. Вносим остальные элементы
-            self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
-            self._apply("Монофосфат калия", "P", self.target_profile['P'])
-            self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
-
-            # 4. Компенсируем NO3
-            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_main_elements()
 
-            # 5. Компенсируем K через сульфат калия
-            k_deficit = self.target_profile['K'] - self.actual_profile['K']
-            if k_deficit > 0.1:
-                self._apply("Калий сернокислый", "K", k_deficit)
+            # Второй проход: компенсация азота
+            self._compensate_nitrogen()
 
-            # 6. Компенсируем S через сульфат калия
-            s_deficit = self.target_profile['S'] - self.actual_profile['S']
-            if s_deficit > 0.1:
-                self._apply("Калий сернокислый", "S", s_deficit)
+            # Третий проход: компенсация второстепенных элементов
+            self._compensate_secondary_elements()
 
             return self.results
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
+    def _compensate_main_elements(self):
+        """Компенсация основных элементов (Ca, Mg, P)"""
+        for element, weight_data in self.compensation_weights.items():
+            if element in ["Ca", "Mg", "P"]:
+                fert_name = weight_data["fert"]
+                main_element = weight_data["main_element"]
+                required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
+                if required_ppm > 0.1:
+                    self._apply(fert_name, main_element, required_ppm)
+
+    def _compensate_nitrogen(self):
+        """Компенсация азота (NO3-, NH4+)"""
+        for nitrogen_type in ["N (NO3-)", "N (NH4+)"]:
+            required_ppm = self.target_profile[nitrogen_type] - self.actual_profile[nitrogen_type]
+            if required_ppm > 0.1:
+                fert_name = self.compensation_weights[nitrogen_type]["fert"]
+                self._apply(fert_name, nitrogen_type, required_ppm)
+
+    def _compensate_secondary_elements(self):
+        """Компенсация второстепенных элементов (K, S)"""
+        for element, weight_data in self.compensation_weights.items():
+            if element in ["K", "S"]:
+                fert_name = weight_data["fert"]
+                main_element = weight_data["main_element"]
+                required_ppm = self.target_profile[main_element] - self.actual_profile[main_element]
+                if required_ppm > 0.1:
+                    self._apply(fert_name, main_element, required_ppm)
+
     def _apply(self, fert_name, main_element, required_ppm):
         if required_ppm <= 0:
             return
@@ -833,6 +863,15 @@ class NutrientCalculator:
     def calculate_ec(self):
         return round(self.total_ec, 2)
 
+    def print_initial_nitrogen_report(self):
+        try:
+            print("Исходный расчёт азота:")
+            print(f"  NO3-: {self.initial_n_profile['NO3-']} ppm")
+            print(f"  NH4+: {self.initial_n_profile['NH4+']} ppm")
+        except Exception as e:
+            print(f"Ошибка при выводе отчёта: {str(e)}")
+            raise
+
     def print_initial_nitrogen_report(self):
         try:
             print("Исходный расчёт азота:")