Update app.py

app.py

@@ -691,16 +691,18 @@ NO3_RATIO = 8.00      # Доля нитратного азота
 NH4_RATIO = 1.00      # Доля аммонийного азота
 VOLUME_LITERS = 10    # Объём раствора
 
-# Базовый профиль (без азота)
+# Базовый профиль (7 элементов)
 BASE_PROFILE = {
     'P': 31.000,
     'K': 210.694,
     'Mg': 24.000,
     'Ca': 84.000,
-    'S': 137.397
+    'S': 137.397,
+    'N (NO3-)': 0,
+    'N (NH4+)': 0
 }
 
-# Полная база всех возможных удобрений
+# Полная база всех возможных удобрений (7 удобрений)
 ALL_FERTILIZERS = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118, "Ca": 0.169
@@ -712,7 +714,7 @@ ALL_FERTILIZERS = {
         "K": 0.448, "S": 0.184
     },
     "Сульфат калия": {
-        "K": 0.448, "S": 0.184  # Альтернативное название калия сернокислого
+        "K": 0.448, "S": 0.184
     },
     "Аммоний азотнокислый": {
         "N (NO3-)": 0.175, "N (NH4+)": 0.175
@@ -721,24 +723,19 @@ ALL_FERTILIZERS = {
         "Mg": 0.098, "S": 0.13
     },
     "Калий фосфорнокислый": {
-        "P": 0.228, "K": 0.287  # KH2PO4 - почти аналогичен монофосфату калия
+        "P": 0.228, "K": 0.287
     }
 }
 
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0, use_k2so4=True, use_kno3=True):
-        """
-        :param volume_liters: Объём раствора
-        :param use_k2so4: Использовать калий сернокислый (True/False)
-        :param use_kno3: Использовать калий азотнокислый (True/False)
-        """
         self.volume = volume_liters
         self.results = {}
-        self.final_profile = {}
-        self.total_ppm = 0
+        self.final_profile = BASE_PROFILE.copy()
+        self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
         self.initial_n_profile = {}
         
-        # Формируем активные удобрения на основе выбора
+        # Формируем активные удобрения
         self.fertilizers_db = {
             "Кальциевая селитра": ALL_FERTILIZERS["Кальциевая селитра"],
             "Аммоний азотнокислый": ALL_FERTILIZERS["Аммоний азотнокислый"],
@@ -752,26 +749,30 @@ class NutrientCalculator:
             self.fertilizers_db["Калий азотнокислый"] = ALL_FERTILIZERS["Калий азотнокислый"]
 
     def calculate(self, total_n, no3_ratio, nh4_ratio):
+        # Рассчитываем баланс азота
         total_parts = no3_ratio + nh4_ratio
         no3 = total_n * (no3_ratio / total_parts)
         nh4 = total_n * (nh4_ratio / total_parts)
-
+        
         self.initial_n_profile = {
             'N (NO3-)': round(no3, 1),
             'N (NH4+)': round(nh4, 1)
         }
-
-        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_potassium_phosphate()
-        self._apply_potassium_fertilizers()
         self._apply_ammonium_nitrate()
+        self._apply_potassium_fertilizers()
+        
+        # Корректируем отрицательные значения
+        self._adjust_negative_values()
+        
         return self.results
 
     def _apply_fertilizer(self, fert_name, grams, additions):
@@ -781,8 +782,17 @@ class NutrientCalculator:
         }
         self.results[fert_name].update(additions)
 
+    def _adjust_negative_values(self):
+        # Корректируем отрицательные значения до 0
+        for element in self.final_profile:
+            if self.final_profile[element] < 0:
+                self.final_profile[element] = 0
+
     def _apply_magnesium_sulfate(self):
         mg_need = self.final_profile['Mg']
+        if mg_need <= 0:
+            return
+            
         mg_content = self.fertilizers_db["Сульфат магния"]["Mg"]
         grams = (mg_need * self.volume) / (mg_content * 1000)
         added_s = grams * self.fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
@@ -793,6 +803,9 @@ class NutrientCalculator:
 
     def _apply_calcium_nitrate(self):
         ca_need = self.final_profile['Ca']
+        if ca_need <= 0:
+            return
+            
         ca_content = self.fertilizers_db["Кальциевая селитра"]["Ca"]
         grams = (ca_need * self.volume) / (ca_content * 1000)
         added_n = grams * self.fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
@@ -803,6 +816,9 @@ class NutrientCalculator:
 
     def _apply_potassium_phosphate(self):
         p_need = self.final_profile['P']
+        if p_need <= 0:
+            return
+            
         p_content = self.fertilizers_db["Калий фосфорнокислый"]["P"]
         grams = (p_need * self.volume) / (p_content * 1000)
         added_k = grams * self.fertilizers_db["Калий фосфорнокислый"]["K"] * 1000 / self.volume
@@ -816,30 +832,22 @@ class NutrientCalculator:
         if k_need <= 0:
             return
 
-        # Вариант 1: Если есть калий сернокислый
+        # Сначала используем калий сернокислый (если доступен)
         if "Калий сернокислый" in self.fertilizers_db:
-            s_deficit = max(0, self.final_profile['S'])
-            if s_deficit > 0:
-                s_content = self.fertilizers_db["Калий сернокислый"]["S"]
-                k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
-                added_k = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
-
-                if added_k > k_need:
-                    k2so4_grams = (k_need * self.volume) / (self.fertilizers_db["Калий сернокислый"]["K"] * 1000)
-                    added_k = k_need
-                    added_s = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
-                else:
-                    added_s = s_deficit
+            s_deficit = self.final_profile['S']
+            k2so4_grams = (k_need * self.volume) / (self.fertilizers_db["Калий сернокислый"]["K"] * 1000)
+            added_s = k2so4_grams * self.fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
 
+            if s_deficit > 0 and added_s <= s_deficit:
                 self._apply_fertilizer("Калий сернокислый", k2so4_grams, {
-                    'внесет K': round(added_k, 2),
+                    'внесет K': round(k_need, 2),
                     'внесет S': round(added_s, 2)
                 })
-                self.final_profile['K'] -= added_k
+                self.final_profile['K'] = 0
                 self.final_profile['S'] -= added_s
-                k_need = self.final_profile['K']
+                return
 
-        # Вариант 2: Если есть только калий азотнокислый
+        # Если остался калий, используем калий азотнокислый
         if k_need > 0 and "Калий азотнокислый" in self.fertilizers_db:
             kno3_grams = (k_need * self.volume) / (self.fertilizers_db["Калий азотнокислый"]["K"] * 1000)
             added_n = kno3_grams * self.fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
@@ -852,14 +860,11 @@ class NutrientCalculator:
             self.final_profile['K'] = 0
             self.final_profile['N (NO3-)'] -= added_n
 
-        # Если остался дефицит и нет подходящих удобрений
-        if k_need > 0:
-            print(f"  Внимание: Не удалось полностью покрыть дефицит K ({k_need:.2f} ppm)")
-
     def _apply_ammonium_nitrate(self):
         nh4_need = self.final_profile['N (NH4+)']
         if nh4_need <= 0:
             return
+            
         nh4_content = self.fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"]
         grams = (nh4_need * self.volume) / (nh4_content * 1000)
         added_n = grams * self.fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
@@ -885,7 +890,7 @@ class NutrientCalculator:
         print("\n" + "="*60)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*60)
-        print(f"Общая концентрация: {self.total_ppm} ppm")
+        print(f"Общая концентрация: {round(self.total_ppm, 1)} ppm")
         print(f"EC: {self.calculate_ec()} mS/cm")
 
         print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
@@ -910,7 +915,7 @@ class NutrientCalculator:
 
 # Пример использования
 if __name__ == "__main__":
-    # Вариант 2: С калием азотнокислым
+    # Вариант с калием азотнокислым
     calc_with_kno3 = NutrientCalculator(volume_liters=10, use_k2so4=False, use_kno3=True)
     calc_with_kno3.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
     calc_with_kno3.print_report()
@@ -918,6 +923,5 @@ if __name__ == "__main__":
 
 
 
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))