Update app.py

app.py

@@ -702,7 +702,7 @@ BASE_PROFILE = {
     'N (NH4+)': 0
 }
 
-# Полная база всех удобрений (6 удобрений, сульфат калия и калий сернокислый - одно и то же)
+# Полная база всех удобрений (7 удобрений)
 ALL_FERTILIZERS = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118, "Ca": 0.169
@@ -721,6 +721,9 @@ ALL_FERTILIZERS = {
     },
     "Калий фосфорнокислый": {
         "P": 0.228, "K": 0.287
+    },
+    "Монофосфат калия": {
+        "P": 0.228, "K": 0.287
     }
 }
 
@@ -731,28 +734,27 @@ class NutrientCalculator:
         self.final_profile = BASE_PROFILE.copy()
         self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
         self.fertilizers_db = ALL_FERTILIZERS
+        self.target_nitrogen = {
+            'NO3': TOTAL_NITROGEN * (NO3_RATIO / (NO3_RATIO + NH4_RATIO)),
+            'NH4': TOTAL_NITROGEN * (NH4_RATIO / (NO3_RATIO + NH4_RATIO))
+        }
 
-    def calculate(self, total_n, no3_ratio, nh4_ratio):
-        # 1. Рассчитываем баланс азота
-        total_parts = no3_ratio + nh4_ratio
-        no3 = total_n * (no3_ratio / total_parts)
-        nh4 = total_n * (nh4_ratio / total_parts)
-        
-        # Записываем азот в профиль
-        self.final_profile['N (NO3-)'] = no3
-        self.final_profile['N (NH4+)'] = nh4
-
-        # 2. Вносим удобрения в строгом порядке
+    def calculate(self):
+        # 1. Вносим удобрения в строгом порядке
         self._apply_magnesium_sulfate()
         self._apply_calcium_nitrate()
         self._apply_potassium_phosphate()
         self._apply_ammonium_nitrate()
         self._apply_potassium_sulfate()
         self._apply_potassium_nitrate()
+        self._apply_monopotassium_phosphate()
         
-        # 3. Корректируем отрицательные значения
+        # 2. Корректируем отрицательные значения
         self._adjust_negative_values()
         
+        # 3. Пересчитываем фактическое содержание азота
+        self._recalculate_actual_nitrogen()
+        
         return self.results
 
     def _apply_fertilizer(self, fert_name, grams, additions):
@@ -767,6 +769,20 @@ class NutrientCalculator:
             if self.final_profile[element] < 0:
                 self.final_profile[element] = 0
 
+    def _recalculate_actual_nitrogen(self):
+        # Пересчитываем фактическое содержание азота на основе внесенных удобрений
+        no3_added = 0
+        nh4_added = 0
+        
+        for fert in self.results.values():
+            if 'внесет NO3' in fert:
+                no3_added += fert['внесет NO3']
+            if 'внесет NH4' in fert:
+                nh4_added += fert['внесет NH4']
+        
+        self.final_profile['N (NO3-)'] = no3_added
+        self.final_profile['N (NH4+)'] = nh4_added
+
     def _apply_magnesium_sulfate(self):
         mg_need = self.final_profile['Mg']
         if mg_need <= 0:
@@ -789,7 +805,6 @@ class NutrientCalculator:
         grams = (ca_need * self.volume) / (ca_content * 1000)
         added_n = grams * self.fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
 
-        self.final_profile['N (NO3-)'] -= added_n
         self._apply_fertilizer("Кальциевая селитра", grams, {'внесет NO3': round(added_n, 1)})
         self.final_profile['Ca'] = 0
 
@@ -807,7 +822,7 @@ class NutrientCalculator:
         self.final_profile['P'] = 0
 
     def _apply_ammonium_nitrate(self):
-        nh4_need = self.final_profile['N (NH4+)']
+        nh4_need = self.target_nitrogen['NH4']
         if nh4_need <= 0:
             return
             
@@ -815,9 +830,10 @@ class NutrientCalculator:
         grams = (nh4_need * self.volume) / (nh4_content * 1000)
         added_n = grams * self.fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
-        self.final_profile['N (NO3-)'] -= added_n
-        self._apply_fertilizer("Аммоний азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
-        self.final_profile['N (NH4+)'] = 0
+        self._apply_fertilizer("Аммоний азотнокислый", grams, {
+            'внесет NO3': round(added_n, 1),
+            'внесет NH4': round(nh4_need, 1)
+        })
 
     def _apply_potassium_sulfate(self):
         k_need = self.final_profile['K']
@@ -829,10 +845,10 @@ class NutrientCalculator:
         k_content = self.fertilizers_db["Калий сернокислый"]["K"]
         s_content = self.fertilizers_db["Калий сернокислый"]["S"]
         
-        # Рассчитываем сколько можем внести по калию и сере
-        grams_by_k = (k_need * self.volume) / (k_content * 1000)
-        grams_by_s = (s_need * self.volume) / (s_content * 1000)
-        grams = min(grams_by_k, grams_by_s)
+        grams = min(
+            (k_need * self.volume) / (k_content * 1000),
+            (s_need * self.volume) / (s_content * 1000)
+        )
         
         added_k = grams * k_content * 1000 / self.volume
         added_s = grams * s_content * 1000 / self.volume
@@ -854,18 +870,38 @@ class NutrientCalculator:
         added_n = grams * self.fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
 
         self.final_profile['K'] = 0
-        self.final_profile['N (NO3-)'] -= added_n
         self._apply_fertilizer("Калий азотнокислый", grams, {'внесет NO3': round(added_n, 2)})
 
+    def _apply_monopotassium_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
+
+        self.final_profile['K'] -= added_k
+        self._apply_fertilizer("Монофосфат калия", grams, {
+            'внесет P': round(p_need, 1),
+            'внесет K': round(added_k, 1)
+        })
+        self.final_profile['P'] = 0
+
     def calculate_ec(self):
         return round(self.total_ppm / 700, 2)
 
     def print_report(self):
         print("\n" + "="*60)
-        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
+        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ПЕРЕСЧИТАННЫЙ):")
         print("="*60)
         table = [[el, round(val, 1)] for el, val in self.final_profile.items()]
-        print(tabulate(table, headers=["Элемент", "Остаток ppm"]))
+        print(tabulate(table, headers=["Элемент", "ppm"]))
+        
+        print("\nЦелевые значения азота:")
+        print(f"  N (NO3-): {round(self.target_nitrogen['NO3'], 1)} ppm")
+        print(f"  N (NH4+): {round(self.target_nitrogen['NH4'], 1)} ppm")
 
         print("\n" + "="*60)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
@@ -893,10 +929,9 @@ class NutrientCalculator:
         else:
             print("  Все элементы покрыты полностью")
 
-# Пример использования
 if __name__ == "__main__":
     calculator = NutrientCalculator(volume_liters=10)
-    calculator.calculate(TOTAL_NITROGEN, NO3_RATIO, NH4_RATIO)
+    calculator.calculate()
     calculator.print_report()