Update app.py

app.py

@@ -793,25 +793,16 @@ class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
-        self.target_profile = BASE_PROFILE.copy()
-        self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
-        self.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS 
+        self.target_profile = {}
+        self.actual_profile = {}
+        self.fertilizers = {}
         self.total_ec = 0.0
 
         # Инициализация весов компенсации
         self.compensation_weights = {
-            "Калий азотнокислый": {"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)
-        self.initial_n_profile = {
-            "NO3-": self.target_profile['N (NO3-)'],
-            "NH4+": self.target_profile['N (NH4+)']
+            "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 _label(self, element):
@@ -831,7 +822,7 @@ class NutrientCalculator:
                 raise KeyError(f"Элемент {main_element} отсутствует в удобрении {fert_name}")
 
             grams = (required_ppm * self.volume) / (content * 1000)
-            
+
             if fert_name not in self.results:
                 result = {
                     'граммы': 0.0,
@@ -841,17 +832,17 @@ class NutrientCalculator:
                 for element in self.fertilizers[fert_name]:
                     result[f'внесет {self._label(element)}'] = 0.0
                 self.results[fert_name] = result
-                
+
             self.results[fert_name]['граммы'] += grams
             self.results[fert_name]['миллиграммы'] = int(grams * 1000)
-            
+
             fert_ec = 0.0
             for element, percent in self.fertilizers[fert_name].items():
                 added_ppm = (grams * percent * 1000) / self.volume
                 self.results[fert_name][f'внесет {self._label(element)}'] += added_ppm
-                self.actual_profile[element] += added_ppm
-                fert_ec += added_ppm * EC_COEFFICIENTS.get(element, 0.0015)
-                
+                self.actual_profile[element] = self.actual_profile.get(element, 0) + added_ppm
+                fert_ec += added_ppm * 0.0015  # Примерный коэффициент EC
+
             self.results[fert_name]['вклад в EC'] += fert_ec
             self.total_ec += fert_ec
         except KeyError as e:
@@ -859,30 +850,26 @@ class NutrientCalculator:
             raise
 
     def _compensate_element(self, element):
-        """Улучшенная компенсация с приоритетами"""
-        needed = self.target_profile[element] - self.actual_profile[element]
+        needed = self.target_profile[element] - self.actual_profile.get(element, 0)
         if needed <= 0:
             return
 
-        # Находим все удобрения, содержащие нужный элемент
         candidates = []
-        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 weight_key, weight_data in self.compensation_weights.items():
+            fert_name = weight_data["fert"]
+            main_element = weight_data["main_element"]
+
+            if element in self.fertilizers.get(fert_name, {}):
                 candidates.append({
                     'name': fert_name,
-                    'weight': weight,
-                    'content': fert_data[element]
+                    'weight': weight_data["weight"],
+                    'content': self.fertilizers[fert_name][element]
                 })
 
         if not candidates:
             raise ValueError(f"Нет удобрений для элемента {element}")
 
-        # Сортируем по весу (убывание)
-        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
@@ -890,130 +877,46 @@ class NutrientCalculator:
 
     def calculate(self):
         try:
-            # 1. Вносим магний
+            # Вносим магний
             self._apply("Сульфат магния", "Mg", self.target_profile['Mg'])
-            
-            # 2. Вносим кальций через компенсацию
+
+            # Вносим кальций через компенсацию
             self._compensate_element("Ca")
-            
-            # 3. Вносим фосфор
+
+            # Вносим фосфор
             self._apply("Монофосфат калия", "P", self.target_profile['P'])
-            
-            # 4. Вносим аммонийный азот
+
+            # Вносим аммонийный азот
             self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
-            
-            # 5. Компенсируем нитратный азот
+
+            # Компенсируем нитратный азот
             self._compensate_element("N (NO3-)")
-            
-            # 6. Компенсируем серу
+
+            # Компенсируем серу
             self._compensate_element("S")
-            
-            # 7. Компенсируем калий
+
+            # Компенсируем калий
             self._compensate_element("K")
-            
+
             return self.results
         except Exception as e:
             print(f"Ошибка при расчёте: {str(e)}")
             raise
 
-    def calculate_ec(self):
-        return round(self.total_ec, 2)
-
-    def print_report(self):
-        try:
-            print("\n" + "="*60)
-            print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
-            print("="*60)
-            table = [[el, round(self.actual_profile[el], 1)] for el in self.actual_profile]
-            print(tabulate(table, headers=["Элемент", "ppm"]))
-
-            print("\nИсходный расчёт азота:")
-            for form, val in self.initial_n_profile.items():
-                print(f"  {form}: {round(val, 1)} ppm")
-
-            print("\n" + "="*60)
-            print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
-            print("="*60)
-            print(f"Общая концентрация: {round(sum(self.actual_profile.values()), 1)} ppm")
-            print(f"EC: {self.calculate_ec()} mS/cm")
-
-            print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
-            fert_table = []
-            for fert, data in self.results.items():
-                adds = [f"+{k}: {v:.1f} ppm" for k, v in data.items() if k.startswith('внесет')]
-                fert_table.append([
-                    fert,
-                    round(data['граммы'], 3),
-                    data['миллиграммы'],
-                    round(data['вклад в EC'], 3),
-                    "\n".join(adds)
-                ])
-            print(tabulate(fert_table, 
-                         headers=["Удобрение", "Граммы", "Миллиграммы", "EC (мСм/см)", "Добавит"]))
-
-            print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
-            deficit = {
-                k: round(self.target_profile[k] - self.actual_profile[k], 1)
-                for k in self.target_profile
-                if abs(self.target_profile[k] - self.actual_profile[k]) > 0.1
-            }
-            if deficit:
-                for el, val in deficit.items():
-                    print(f"  {el}: {val} ppm")
-            else:
-                print("  Все элементы покрыты полностью")
-        except Exception as e:
-            print(f"Ошибка при выводе отчёта: {str(e)}")
-            raise
-
-if __name__ == "__main__":
-    # Константы (замените на свои значения)
-    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)
-        calculator.calculate()
-        calculator.print_report()
-    except Exception as e:
-        print(f"Критическая ошибка: {str(e)}")
 
+# Роут для обработки расчетов
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try:
         data = request.get_json()
-        
+
         # Проверка обязательных полей
         if not data or 'fertilizerConstants' not in data or 'profileSettings' not in data:
             return jsonify({'error': 'Неверный формат данных'}), 400
-        
+
         # Инициализация калькулятора
         calculator = NutrientCalculator(volume_liters=float(data['profileSettings'].get('liters', 100)))
-        
+
         # Настройка удобрений из запроса
         calculator.fertilizers = {
             "Кальциевая селитра": {
@@ -1041,7 +944,7 @@ def handle_calculation():
                 "S": float(data['fertilizerConstants']["Калий сернокислый"].get("S", 0.18401))
             }
         }
-        
+
         # Настройка целевого профиля
         calculator.target_profile = {
             'P': float(data['profileSettings'].get('P', 31.0)),
@@ -1052,19 +955,19 @@ def handle_calculation():
             'N (NO3-)': 0.0,
             'N (NH4+)': 0.0
         }
-        
+
         # Установка параметров азота
         total_n = float(data['profileSettings'].get('TOTAL_NITROGEN', 125.0))
         no3_ratio = float(data['profileSettings'].get('NO3_RATIO', 8.25))
         nh4_ratio = 1.0
-        
+
         calculator.target_profile['N (NO3-)'] = total_n * (no3_ratio / (no3_ratio + nh4_ratio))
         calculator.target_profile['N (NH4+)'] = total_n * (nh4_ratio / (no3_ratio + nh4_ratio))
-        
+
         # Выполнение расчета
         results = calculator.calculate()
         return jsonify(results)
-        
+
     except Exception as e:
         return jsonify({'error': str(e)}), 500