Update app.py

app.py

@@ -798,14 +798,14 @@ class NutrientCalculator:
         self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
         self.total_ec = 0.0
 
-        # Веса для компенсации
+        # Инициализация весов компенсации
         self.compensation_weights = {
-            "POTASSIUM_NITRATE": {"weight": 0.5, "fert": "Калий азотнокислый", "main_element": "K"},
-            "CALCIUM_NITRATE": {"weight": 0.3, "fert": "Кальциевая селитра", "main_element": "Ca"},
-            "POTASSIUM_SULFATE": {"weight": 0.2, "fert": "Калий сернокислый", "main_element": "K"}
+            "Калий азотнокислый": {"weight": 0.5, "main_elements": ["K", "N (NO3-)"]},
+            "Кальциевая селитра": {"weight": 0.3, "main_elements": ["Ca", "N (NO3-)"]},
+            "Калий сернокислый": {"weight": 0.2, "main_elements": ["K", "S"]}
         }
 
-        # Расчёт азота (оставляем без изменений)
+        # Расчёт азота
         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)
@@ -815,7 +815,6 @@ class NutrientCalculator:
         }
 
     def _label(self, element):
-        """Форматирование названий элементов для вывода"""
         labels = {
             'N (NO3-)': 'NO3',
             'N (NH4+)': 'NH4'
@@ -823,7 +822,6 @@ class NutrientCalculator:
         return labels.get(element, element)
 
     def _apply(self, fert_name, main_element, required_ppm):
-        """Внесение удобрения"""
         if required_ppm <= 0:
             return
             
@@ -842,7 +840,7 @@ class NutrientCalculator:
                 self.results[fert_name] = result
                 
             self.results[fert_name]['граммы'] += grams
-            self.results[fert_name]['миллиграммы'] = int(self.results[fert_name]['граммы'] * 1000)
+            self.results[fert_name]['миллиграммы'] = int(grams * 1000)
             
             fert_ec = 0.0
             for element, percent in self.fertilizers[fert_name].items():
@@ -857,61 +855,59 @@ class NutrientCalculator:
             print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
             raise
 
-    def _compensate_element(self, element: str, needed_ppm: float):
-        """Компенсация дефицита элемента с учетом весов"""
-        suitable_ferts = []
-        for fert_key, fert_data in self.compensation_weights.items():
-            fert_name = fert_data['fert']
-            if fert_name in self.fertilizers and element in self.fertilizers[fert_name]:
-                suitable_ferts.append({
+    def _compensate_element(self, element):
+        """Улучшенная компенсация с приоритетами"""
+        needed = self.target_profile[element] - self.actual_profile[element]
+        if needed <= 0:
+            return
+
+        # Находим все удобрения, содержащие нужный элемент
+        candidates = []
+        for fert_name, fert_data in self.fertilizers.items():
+            if element in fert_data:
+                weight = next((w['weight'] for w in self.compensation_weights.values() 
+                             if w['fert'] == fert_name), 0.1)
+                candidates.append({
                     'name': fert_name,
-                    'weight': fert_data['weight'],
-                    'content': self.fertilizers[fert_name][element]
+                    'weight': weight,
+                    'content': fert_data[element]
                 })
-        
-        if not suitable_ferts:
-            raise ValueError(f"Нет подходящих удобрений для элемента {element}")
 
-        total_weight = sum(f['weight'] for f in suitable_ferts)
-        for fert in suitable_ferts:
-            share = fert['weight'] / total_weight
-            ppm_to_apply = needed_ppm * share
-            self._apply(fert['name'], element, ppm_to_apply)
+        if not candidates:
+            raise ValueError(f"Нет удобрений для элемента {element}")
 
-    def _apply_k_sulfate(self):
-        """Внесение калия сернокислого с учетом весов"""
-        s_def = self.target_profile['S'] - self.actual_profile['S']
-        if s_def > 0.1:
-            self._compensate_element("S", s_def)
+        # Сортируем по весу (убывание)
+        candidates.sort(key=lambda x: x['weight'], reverse=True)
+        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 calculate(self):
-        """Основной расчет"""
         try:
             # 1. Вносим магний
             self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
             
-            # 2. Вносим кальций
-            self._apply("Кальциевая селитра", "Ca", self.target_profile['Ca'])
+            # 2. Вносим кальций через компенсацию
+            self._compensate_element("Ca")
             
             # 3. Вносим фосфор
             self._apply("Монофосфат калия", "P", self.target_profile['P'])
             
-            # 4. Вносим аммонийный азот (логика не меняется)
+            # 4. Вносим аммонийный азот
             self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
             
-            # 5. Компенсируем нитратный азот (логика не меняется)
-            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)
+            # 5. Компенсируем нитратный азот
+            self._compensate_element("N (NO3-)")
             
-            # 6. Вносим серу и калий с учетом весов
-            self._apply_k_sulfate()
+            # 6. Компенсируем серу
+            self._compensate_element("S")
             
-            # 7. Компенсируем остаточный дефицит калия с учетом весов
-            k_deficit = self.target_profile['K'] - self.actual_profile['K']
-            if k_deficit > 0.1:
-                self._compensate_element("K", k_deficit)
+            # 7. Компенсируем калий
+            self._compensate_element("K")
             
             return self.results
         except Exception as e:
@@ -919,11 +915,9 @@ class NutrientCalculator:
             raise
 
     def calculate_ec(self):
-        """Расчет общей электропроводности"""
         return round(self.total_ec, 2)
 
     def print_report(self):
-        """Вывод отчета"""
         try:
             print("\n" + "="*60)
             print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
@@ -971,11 +965,33 @@ class NutrientCalculator:
             raise
 
 if __name__ == "__main__":
-    # Убедитесь, что эти константы определены
-    TOTAL_NITROGEN = 155.0  # Примерное значение, замените на своё
-    NO3_RATIO = 9.0         # Примерное значение, замените на своё
-    NH4_RATIO = 1.0         # Примерное значение, замените на своё
-    VOLUME_LITERS = 1.0     # Примерное значение, замените на своё
+    # Константы (замените на свои значения)
+    TOTAL_NITROGEN = 125.0
+    NO3_RATIO = 8.25
+    NH4_RATIO = 1.0
+    VOLUME_LITERS = 100
+    
+    # Убедитесь, что эти словари определены
+    BASE_PROFILE = {
+        'P': 31.0, 'K': 210.0, 'Mg': 24.0,
+        'Ca': 84.0, 'S': 56.439,
+        'N (NO3-)': 0.0, 'N (NH4+)': 0.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}
+    }
+    
+    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
+    }
 
     try:
         calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)