Update app.py

app.py

@@ -809,11 +809,11 @@ class NutrientCalculator:
             'N (NO3-)': 0.0017, 'N (NH4+)': 0.0019
         }
 
-        # Веса для компенсации (приоритеты)
+        # Новая структура весов для компенсации
         self.compensation_weights = {
-            'Калий азотнокислый': {'weight': 0.8, 'main_elements': ['K', 'N (NO3-)']},
-            'Кальциевая селитра': {'weight': 0.1, 'main_elements': ['Ca', 'N (NO3-)']},
-            'Калий сернокислый': {'weight': 0.1, 'main_elements': ['K', 'S']}
+            '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'}
         }
 
     def _apply_fertilizer(self, fert_name: str, element: str, ppm: float):
@@ -856,44 +856,42 @@ class NutrientCalculator:
         return round(self.total_ec, 2)
 
     def _compensate_element(self, element: str, needed_ppm: float):
-        """Исправленная компенсация с правильным учетом весов"""
-        # Находим все подходящие удобрения
+        """Исправленная компенсация с учетом новых весов и приоритетов азота"""
+        # Находим подходящие удобрения для элемента
         suitable_ferts = []
-        for fert_name, fert_data in self.fertilizers.items():
-            if element in fert_data:
-                weight = self.compensation_weights.get(fert_name, {}).get('weight', 0.1)
+        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({
                     'name': fert_name,
-                    'weight': weight,
-                    'content': fert_data[element]
+                    'weight': fert_data['weight'],
+                    'content': self.fertilizers[fert_name][element]
                 })
         
         if not suitable_ferts:
             raise ValueError(f"Нет удобрений для элемента {element}")
 
-        # Сортируем по весу (убывание)
-        suitable_ferts.sort(key=lambda x: x['weight'], reverse=True)
-
         # Рассчитываем общий вес для нормализации
         total_weight = sum(f['weight'] for f in suitable_ferts)
         
+        # Распределяем дефицит пропорционально весам
         remaining_ppm = needed_ppm
         for fert in suitable_ferts:
             if remaining_ppm <= 0:
                 break
                 
-            # Рассчитываем долю этого удобрения
+            # Вычисляем долю удобрения
             share = fert['weight'] / total_weight
             ppm_to_apply = remaining_ppm * share
             
-            # Вносим рассчитанную долю
+            # Вносим удобрение
             self._apply_fertilizer(fert['name'], element, ppm_to_apply)
             
-            # Обновляем оставшийся дефицит
+            # Обновляем остаток дефицита
             remaining_ppm = self.target_profile[element] - self.actual_profile[element]
 
     def calculate(self):
-        """Основной расчет с исправленной компенсацией"""
+        """Основной расчет с учетом новых приоритетов"""
         try:
             # 1. Вносим магний (сульфат магния)
             self._apply_fertilizer("Сульфат магния", "Mg", self.target_profile['Mg'])
@@ -901,14 +899,16 @@ class NutrientCalculator:
             # 2. Вносим аммонийный азот
             self._apply_fertilizer("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
             
-            # 3. Компенсируем нитратный азот с учетом весов
-            self._compensate_element("N (NO3-)", self.target_profile['N (NO3-)'])
+            # 3. Компенсируем нитратный азот с учетом новых весов
+            if self.target_profile['N (NO3-)'] > 0:
+                self._compensate_element("N (NO3-)", self.target_profile['N (NO3-)'])
             
             # 4. Вносим фосфор
             self._apply_fertilizer("Монофосфат калия", "P", self.target_profile['P'])
             
-            # 5. Компенсируем калий с учетом весов
-            self._compensate_element("K", self.target_profile['K'])
+            # 5. Компенсируем калий с учетом новых весов
+            if self.target_profile['K'] > self.actual_profile['K']:
+                self._compensate_element("K", self.target_profile['K'] - self.actual_profile['K'])
             
             # 6. Компенсируем кальций (если нужно)
             ca_needed = self.target_profile['Ca'] - self.actual_profile['Ca']
@@ -920,7 +920,7 @@ class NutrientCalculator:
         except Exception as e:
             raise RuntimeError(f"Ошибка расчета: {str(e)}")
 
-    def _prepare_results(self) -> Dict[str, Any]:
+    def _prepare_results(self) -> dict:
         """Подготовка результатов для API"""
         return {
             'actual_profile': {k: round(v, 2) for k, v in self.actual_profile.items()},
@@ -943,8 +943,6 @@ class NutrientCalculator:
         }
 
 
-
-
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try: