Update app.py

app.py

@@ -896,44 +896,93 @@ if __name__ == "__main__":
 
 
         
-@app.route('/calculate', methods=['POST'])
-def calculate_nutrients():
+@app.route('/calculation', methods=['POST'])
+def handle_calculation():
     try:
-        # Получаем данные из запроса
         data = request.get_json()
         
-        # Обновляем переменные из запроса (кроме NH4+)
-        volume = float(data.get('volume', 100))
+        # 1. Проверка обязательных полей
+        if not data or 'fertilizerConstants' not in data or 'profileSettings' not in data:
+            return jsonify({'error': 'Неверный формат данных'}), 400
+        
+        # 2. Извлекаем данные из запроса
+        fertilizer_data = data['fertilizerConstants']
+        profile_data = data['profileSettings']
+        
+        # 3. Устанавливаем константы из запроса
+        TOTAL_NITROGEN = float(profile_data.get('TOTAL_NITROG', 125.0))
+        NO3_RATIO = float(profile_data.get('NO3_RAT', 8.25))  # Берем как есть, без деления на 100
+        VOLUME_LITERS = float(profile_data.get('liters', 100))
+        NH4_RATIO = 1.00  # Фиксированное значение
+        
+        # 4. Формируем целевой профиль
         target_profile = {
-            'P': float(data.get('P', 31.000)),
-            'K': float(data.get('K', 210.000)),
-            'Mg': float(data.get('Mg', 24.000)),
-            'Ca': float(data.get('Ca', 84.000)),
-            'S': float(data.get('S', 56.439)),
-            'N (NO3-)': float(data.get('NO3', 0)),
-            'N (NH4+)': 1.00  # По умолчанию как в оригинале
+            'P': float(profile_data.get('P', 31.0)),
+            'K': float(profile_data.get('K', 210.0)),
+            'Mg': float(profile_data.get('Mg', 24.0)),
+            'Ca': float(profile_data.get('Ca', 84.0)),
+            'S': float(profile_data.get('S', 56.439)),
+            'N (NO3-)': 0,  # Будет рассчитано в калькуляторе
+            'N (NH4+)': 0   # Будет рассчитано в калькуляторе
+        }
+        
+        # 5. Обновляем константы удобрений
+        NUTRIENT_CONTENT_IN_FERTILIZERS = {
+            "Кальциевая селитра": {
+                "N (NO3-)": float(fertilizer_data["Кальциевая селитра"].get("N (NO3-)", 0.11863)),
+                "Ca": float(fertilizer_data["Кальциевая селитра"].get("Ca", 0.16972))
+            },
+            "Калий азотнокислый": {
+                "N (NO3-)": float(fertilizer_data["Калий азотнокислый"].get("N (NO3-)", 0.13854)),
+                "K": float(fertilizer_data["Калий азотнокислый"].get("K", 0.36672))
+            },
+            "Аммоний азотнокислый": {
+                "N (NO3-)": float(fertilizer_data["Аммоний азотнокислый"].get("N (NO3-)", 0.17499)),
+                "N (NH4+)": float(fertilizer_data["Аммоний азотнокислый"].get("N (NH4+)", 0.17499))
+            },
+            "Сульфат магния": {
+                "Mg": float(fertilizer_data["Сульфат магния"].get("Mg", 0.1022)),
+                "S": float(fertilizer_data["Сульфат магния"].get("S", 0.13483))
+            },
+            "Монофосфат калия": {
+                "P": float(fertilizer_data["Монофосфат калия"].get("P", 0.22761)),
+                "K": float(fertilizer_data["Монофосфат калия"].get("K", 0.28731))
+            },
+            "Калий сернокислый": {
+                "K": float(fertilizer_data["Калий сернокислый"].get("K", 0.44874)),
+                "S": float(fertilizer_data["Калий сернокислый"].get("S", 0.18401))
+            }
         }
         
-        # Создаем и настраиваем калькулятор
-        calculator = NutrientCalculator(volume_liters=volume)
-        calculator.target_profile.update(target_profile)
+        # 6. Создаем и настраиваем калькулятор
+        calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
+        calculator.target_profile = target_profile
+        calculator.fertilizers = NUTRIENT_CONTENT_IN_FERTILIZERS
+        
+        # 7. Устанавливаем параметры азота
+        calculator.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / (NO3_RATIO + NH4_RATIO))
+        calculator.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / (NO3_RATIO + NH4_RATIO))
         
-        # Выполняем расчет
+        # 8. Выполняем расчет
         results = calculator.calculate()
         
-        # Формируем ответ
+        # 9. Формируем ответ
         response = {
-            'profile': {k: round(v, 2) for k, v in calculator.actual_profile.items()},
+            'actual_profile': {k: round(v, 2) for k, v in calculator.actual_profile.items()},
             'fertilizers': results,
             'total_ec': calculator.calculate_ec(),
-            'total_ppm': round(sum(calculator.actual_profile.values()), 1)
+            'total_ppm': round(sum(calculator.actual_profile.values()), 2),
+            'nitrogen_ratios': {
+                'NO3_RATIO': NO3_RATIO,
+                'NH4_RATIO': NH4_RATIO,
+                'TOTAL_NITROGEN': TOTAL_NITROGEN
+            }
         }
         
         return jsonify(response)
         
     except Exception as e:
-        return jsonify({'error': str(e)}), 400
-
+        return jsonify({'error': f'Ошибка при расчете: {str(e)}'}), 500