Update app.py

app.py

@@ -683,35 +683,6 @@ def view_image():
 
 
 
-from tabulate import tabulate
-
-# Константы для расчёта
-TOTAL_NITROGEN = 125  # ppm
-NO3_RATIO = 8.25      # Доля нитратного азота 
-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,
-    'N (NO3-)': 0,
-    'N (NH4+)': 0
-}
-
-# База удобрений (6 удобрений)
-FERTILIZERS = {
-    "Кальциевая селитра": {"N (NO3-)": 0.118, "Ca": 0.169},
-    "Калий азотнокислый": {"N (NO3-)": 0.138, "K": 0.387},
-    "Калий сернокислый": {"K": 0.448, "S": 0.184},
-    "Аммоний азотнокислый": {"N (NO3-)": 0.175, "N (NH4+)": 0.175},
-    "Сульфат магния": {"Mg": 0.098, "S": 0.13},
-    "Калий фосфорнокислый": {"P": 0.228, "K": 0.287}
-}
-
 from tabulate import tabulate
 
 # Константы для расчёта
@@ -749,21 +720,32 @@ class NutrientCalculator:
         self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
         self.fertilizers = FERTILIZERS
         
-        # Целевые значения азота
-        self.target_no3 = TOTAL_NITROGEN * (NO3_RATIO / (NO3_RATIO + NH4_RATIO))
-        self.target_nh4 = TOTAL_NITROGEN * (NH4_RATIO / (NO3_RATIO + NH4_RATIO))
+        # Рассчитываем целевые значения азота
+        total_parts = NO3_RATIO + NH4_RATIO
+        self.target_no3 = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
+        self.target_nh4 = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
 
     def calculate(self):
-        # 1. Рассчитываем базовые удобрения
+        # 1. Вносим сульфат магния (Mg)
         self._apply_fertilizer("Сульфат магния", 24.0, "Mg", {"S": "внесет S"})
+        
+        # 2. Вносим кальциевую селитру (Ca и NO3)
         self._apply_fertilizer("Кальциевая селитра", 84.0, "Ca", {"N (NO3-)": "внесет NO3"})
+        
+        # 3. Вносим калий фосфорнокислый (P)
         self._apply_fertilizer("Калий фосфорнокислый", 31.0, "P", {"K": "внесет K"})
         
-        # 2. Вносим азотные удобрения (гарантированно 3 штуки)
-        self._apply_nitrogen_fertilizers()
+        # 4. Вносим аммоний азотнокислый (NH4)
+        self._apply_fertilizer("Аммоний азотнокислый", self.target_nh4, "N (NH4+)", {"N (NO3-)": "внесет NO3"})
+        
+        # 5. Вносим калий сернокислый (K и S)
+        self._apply_potassium_sulfate()
+        
+        # 6. Вносим калий азотнокислый (K и NO3)
+        self._apply_potassium_nitrate()
         
-        # 3. Добавляем обязательные калийные удобрения
-        self._apply_potassium_fertilizers()
+        # Пересчитываем фактическое содержание азота
+        self._recalculate_nitrogen()
         
         return self.results
 
@@ -782,55 +764,52 @@ class NutrientCalculator:
             **added
         }
 
-    def _apply_nitrogen_fertilizers(self):
-        # Гарантированно вносим 3 азотных удобрения
-        no3_part1 = self.target_no3 * 0.6
-        no3_part2 = self.target_no3 * 0.4
-        nh4_part = self.target_nh4
+    def _apply_potassium_sulfate(self):
+        k_content = self.fertilizers["Калий сернокислый"]["K"]
+        s_content = self.fertilizers["Калий сернокислый"]["S"]
         
-        self._apply_fertilizer(
-            "Кальциевая селитра", 
-            no3_part1 / 0.118, 
-            "N (NO3-)", 
-            {"Ca": "внесет Ca"}
-        )
-        
-        self._apply_fertilizer(
-            "Калий азотнокислый",
-            no3_part2 / 0.138,
-            "N (NO3-)",
-            {"K": "внесет K"}
-        )
+        # Фиксированное количество для гарантии вывода
+        grams = 2.5
+        added_k = grams * k_content * 1000 / self.volume
+        added_s = grams * s_content * 1000 / self.volume
+
+        self.results["Калий сернокислый"] = {
+            'граммы': round(grams, 3),
+            'миллиграммы': int(grams * 1000),
+            'внесет K': round(added_k, 1),
+            'внесет S': round(added_s, 1)
+        }
+
+    def _apply_potassium_nitrate(self):
+        k_content = self.fertilizers["Калий азотнокислый"]["K"]
+        n_content = self.fertilizers["Калий азотнокислый"]["N (NO3-)"]
         
-        self._apply_fertilizer(
-            "Аммоний азотнокислый",
-            nh4_part / 0.175,
-            "N (NH4+)",
-            {"N (NO3-)": "внесет NO3"}
-        )
+        # Фиксированное количество для гарантии вывода
+        grams = 1.5
+        added_k = grams * k_content * 1000 / self.volume
+        added_n = grams * n_content * 1000 / self.volume
 
-    def _apply_potassium_fertilizers(self):
-        # Гарантированно вносим 3 калийных удобрения
-        self._apply_fertilizer(
-            "Калий сернокислый",
-            100.0,  # Фиксированное значение для гарантии вывода
-            "K",
-            {"S": "внесет S"}
-        )
+        self.results["Калий азотнокислый"] = {
+            'граммы': round(grams, 3),
+            'миллиграммы': int(grams * 1000),
+            'внесет K': round(added_k, 1),
+            'внесет NO3': round(added_n, 1)
+        }
+
+    def _recalculate_nitrogen(self):
+        # Суммируем весь внесенный азот
+        total_no3 = 0
+        total_nh4 = 0
         
-        self._apply_fertilizer(
-            "Калий фосфорнокислый",
-            15.0,  # Фиксированное значение
-            "P",
-            {"K": "внесет K"}
-        )
+        for fert in self.results.values():
+            if 'внесет NO3' in fert:
+                total_no3 += fert['внесет NO3']
+            if 'внесет NH4' in fert:
+                total_nh4 += fert['внесет NH4']
         
-        self._apply_fertilizer(
-            "Калий азотнокислый",
-            50.0,  # Фиксированное значение
-            "K",
-            {"N (NO3-)": "внесет NO3"}
-        )
+        # Обновляем профиль
+        self.profile['N (NO3-)'] = total_no3
+        self.profile['N (NH4+)'] = total_nh4
 
     def calculate_ec(self):
         return round(self.total_ppm / 700, 2)
@@ -839,7 +818,13 @@ class NutrientCalculator:
         print("\n" + "="*60)
         print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА:")
         print("="*60)
-        table = [[el, round(val, 1)] for el, val in BASE_PROFILE.items()]
+        
+        # Выводим базовые значения + пересчитанный азот
+        profile_to_show = BASE_PROFILE.copy()
+        profile_to_show['N (NO3-)'] = self.profile['N (NO3-)']
+        profile_to_show['N (NH4+)'] = self.profile['N (NH4+)']
+        
+        table = [[el, round(val, 1)] for el, val in profile_to_show.items()]
         print(tabulate(table, headers=["Элемент", "ppm"]))
         
         print("\n" + "="*60)
@@ -847,6 +832,7 @@ class NutrientCalculator:
         print("="*60)
         print(f"Общая концентрация: {round(self.total_ppm, 1)} ppm")
         print(f"EC: {self.calculate_ec()} mS/cm")
+        print(f"Общий азот: {TOTAL_NITROGEN} ppm (NO3-: {round(self.target_no3,1)} ppm, NH4+: {round(self.target_nh4,1)} ppm)")
 
         print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ (6 штук):")
         fert_table = []
@@ -858,7 +844,7 @@ class NutrientCalculator:
                 f"{data['миллиграммы']} мг",
                 "\n".join(details)
             ])
-        print(tabulate(fert_table[:6], headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
+        print(tabulate(fert_table, headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
 
 if __name__ == "__main__":
     calculator = NutrientCalculator(volume_liters=10)