Update app.py

app.py

@@ -984,30 +984,45 @@ def handle_calculation():
     try:
         data = request.json
         
-        # Создаем калькулятор с указанным объемом
+        # 1. СОЗДАЕМ КАЛЬКУЛЯТОР (без изменений)
         calculator = NutrientCalculator(volume_liters=data['profileSettings']['liters'])
         
-        # Полностью переопределяем целевой профиль из полученных данных
-        calculator.target_profile = {
+        # 2. ПОДГОТАВЛИВАЕМ ДАННЫЕ ДЛЯ РАСЧЕТА
+        # Получаем значения из запроса
+        no3_value = float(data['profileSettings']['N (NO3-)'])
+        nh4_value = float(data['profileSettings']['N (NH4+)'])
+        total_n = no3_value + nh4_value
+        
+        # 3. ПОДМЕНЯЕМ РАСЧЕТ АЗОТА В КАЛЬКУЛЯТОРЕ
+        # Сохраняем оригинальные значения
+        original_no3 = calculator.target_profile.get('N (NO3-)', 0)
+        original_nh4 = calculator.target_profile.get('N (NH4+)', 0)
+        
+        # Устанавливаем новые значения
+        calculator.target_profile.update({
             'P': float(data['profileSettings']['P']),
             'K': float(data['profileSettings']['K']),
             'Mg': float(data['profileSettings']['Mg']),
             'Ca': float(data['profileSettings']['Ca']),
             'S': float(data['profileSettings']['S']),
-            'N (NO3-)': float(data['profileSettings']['N (NO3-)']),  # Берем из запроса
-            'N (NH4+)': float(data['profileSettings']['N (NH4+)'])   # Берем из запроса
-        }
+            'N (NO3-)': no3_value,  # Используем переданное значение
+            'N (NH4+)': nh4_value   # Используем переданное значение
+        })
         
-        # Устанавливаем параметры удобрений из запроса
+        # 4. НАСТРАИВАЕМ УДОБРЕНИЯ (без изменений)
         calculator.fertilizers = {
             fert: {el: float(p) for el, p in nutrients.items()}
             for fert, nutrients in data['fertilizerConstants'].items()
         }
-
-        # Выполняем расчет
+        
+        # 5. ВЫПОЛНЯЕМ РАСЧЕТ (без изменений)
         calculator.calculate()
         
-        # Возвращаем результаты
+        # 6. ВОЗВРАЩАЕМ ОРИГИНАЛЬНЫЕ ЗНАЧЕНИЯ (для безопасности)
+        calculator.target_profile['N (NO3-)'] = original_no3
+        calculator.target_profile['N (NH4+)'] = original_nh4
+        
+        # 7. ВОЗВРАЩАЕМ РЕЗУЛЬТАТ
         return jsonify(calculator.get_web_results())
         
     except Exception as e:
@@ -1028,9 +1043,5 @@ def handle_calculation():
 
 
 
-
-
-
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))