Update app.py

app.py

@@ -683,12 +683,12 @@ def view_image():
 
 
 
-from tabulate import tabulate
+# Общий азот и соотношение NO3:NH4
+TOTAL_NITROGEN = 220  # ppm
+N_RATIO = (10, 1)      # Пример: 10:1 (NO3:NH4)
 
 # Профиль питательного раствора для томатов (ppm)
 TOMATO_PROFILE = {
-    'N (NO3-)': 200,
-    'N (NH4+)': 20,
     'P': 50,
     'K': 350,
     'Mg': 50,
@@ -696,7 +696,7 @@ TOMATO_PROFILE = {
     'S': 100
 }
 
-# Обновленная база данных удобрений с добавлением сульфата калия
+# База данных удобрений
 fertilizers_db = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118,
@@ -706,9 +706,9 @@ fertilizers_db = {
         "N (NO3-)": 0.138,
         "K": 0.387,
     },
-    "Калий сернокислый": {  # Добавлен новый компонент
-        "K": 0.448,  # 44.8% K
-        "S": 0.184,   # 18.4% S
+    "Калий сернокислый": {
+        "K": 0.448,
+        "S": 0.184,
     },
     "Аммоний азотнокислый": {
         "N (NO3-)": 0.175,
@@ -724,19 +724,31 @@ fertilizers_db = {
     }
 }
 
+
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
         self.final_profile = {}
-        self.total_ppm = sum(TOMATO_PROFILE.values())
+        self.total_ppm = 0
+
+    def calculate(self, base_profile, total_n, n_ratio):
+        # Расчёт NO3 и NH4 на основе общего азота и соотношения
+        total_parts = n_ratio[0] + n_ratio[1]
+        no3 = total_n * (n_ratio[0] / total_parts)
+        nh4 = total_n * (n_ratio[1] / total_parts)
 
-    def calculate(self, profile):
-        self.final_profile = profile.copy()
+        # Формирование полного профиля
+        self.final_profile = base_profile.copy()
+        self.final_profile['N (NO3-)'] = no3
+        self.final_profile['N (NH4+)'] = nh4
+        self.total_ppm = total_n + sum(base_profile.values())
+
+        # Последовательное внесение удобрений
         self._apply_magnesium_sulfate()
         self._apply_calcium_nitrate()
         self._apply_mkp()
-        self._apply_potassium_fertilizers()  # Измененный метод для работы с двумя калийными удобрениями
+        self._apply_potassium_fertilizers()
         self._apply_ammonium_nitrate()
         return self.results
 
@@ -778,51 +790,36 @@ class NutrientCalculator:
         self.final_profile['P'] = 0
 
     def _apply_potassium_fertilizers(self):
-        """Новый метод для распределения калия между двумя удобрениями"""
         k_need = self.final_profile['K']
         if k_need <= 0:
             return
-            
-        # Сначала вносим часть калия через сернокислый калий для компенсации серы
+
         s_deficit = max(0, self.final_profile['S'])
         if s_deficit > 0:
-            # Рассчитываем сколько сернокислого калия нужно для компенсации дефицита серы
             s_content = fertilizers_db["Калий сернокислый"]["S"]
             k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
-            
-            # Сколько калия это добавит
             added_k = k2so4_grams * fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
-            
-            # Если этим удобрением мы покрываем больше калия, чем нужно
+
             if added_k > k_need:
                 k2so4_grams = (k_need * self.volume) / (fertilizers_db["Калий сернокислый"]["K"] * 1000)
                 added_k = k_need
                 added_s = k2so4_grams * fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
             else:
                 added_s = s_deficit
-                
-            self._apply_fertilizer(
-                "Калий сернокислый", 
-                k2so4_grams, 
-                {
-                    'внесет K': round(added_k, 1),
-                    'внесет S': round(added_s, 1)
-                }
-            )
+
+            self._apply_fertilizer("Калий сернокислый", k2so4_grams, {
+                'внесет K': round(added_k, 1),
+                'внесет S': round(added_s, 1)
+            })
             self.final_profile['K'] -= added_k
             self.final_profile['S'] -= added_s
             k_need = self.final_profile['K']
-        
-        # Остаток калия вносим азотнокислым калием
+
         if k_need > 0:
             kno3_grams = (k_need * self.volume) / (fertilizers_db["Калий азотнокислый"]["K"] * 1000)
             added_n = kno3_grams * fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
-            
-            self._apply_fertilizer(
-                "Калий азотнокислый", 
-                kno3_grams, 
-                {'внесет NO3': round(added_n, 1)}
-            )
+
+            self._apply_fertilizer("Калий азотнокислый", kno3_grams, {'внесет NO3': round(added_n, 1)})
             self.final_profile['K'] = 0
             self.final_profile['N (NO3-)'] -= added_n
 
@@ -839,29 +836,24 @@ class NutrientCalculator:
         self.final_profile['N (NH4+)'] = 0
 
     def calculate_ec(self):
-        """Расчет электропроводимости (700 ppm ≈ 1 mS/cm)"""
         return round(self.total_ppm / 700, 2)
 
     def print_report(self):
-        # Сначала выводим заданный профиль
         print("\n" + "="*50)
         print("ЗАДАННЫЙ ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ppm):")
         print("="*50)
         profile_table = []
-        for element, value in TOMATO_PROFILE.items():
-            profile_table.append([element, value])
+        for element, value in self.final_profile.items():
+            profile_table.append([element, round(value, 1)])
         print(tabulate(profile_table, headers=["Элемент", "Концентрация (ppm)"]))
-        
-        # Затем выводим результаты расчета
+
         ec = self.calculate_ec()
-        
         print("\n" + "="*50)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*50)
-        
         print(f"\nОБЩАЯ КОНЦЕНТРАЦИЯ: {self.total_ppm} ppm")
         print(f"ЭЛЕКТРОПРОВОДИМОСТЬ (EC): {ec} mS/cm (при 25°C)")
-        
+
         print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
         table = []
         for fert, data in self.results.items():
@@ -873,7 +865,7 @@ class NutrientCalculator:
                 "\n".join(additions)
             ])
         print(tabulate(table, headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
-        
+
         print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
         deficit = {k: v for k, v in self.final_profile.items() if v > 0.1}
         if deficit:
@@ -882,10 +874,11 @@ class NutrientCalculator:
         else:
             print("  Все элементы полностью покрыты")
 
+
 # Пример использования
 if __name__ == "__main__":
     calc = NutrientCalculator(volume_liters=10)
-    results = calc.calculate(TOMATO_PROFILE.copy())
+    results = calc.calculate(TOMATO_PROFILE.copy(), total_n=TOTAL_NITROGEN, n_ratio=N_RATIO)
     calc.print_report()
 
 
@@ -897,7 +890,5 @@ if __name__ == "__main__":
 
 
 
-
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))