Update app.py

app.py

@@ -680,7 +680,6 @@ def view_image():
 
 
 
-
 from tabulate import tabulate
 
 # Константы для расчёта
@@ -717,11 +716,16 @@ class NutrientCalculator:
         self.profile = BASE_PROFILE.copy()
         self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
         self.fertilizers = FERTILIZERS
+        self.final_profile = BASE_PROFILE.copy()
         
         # Рассчитываем целевые значения азота
         total_parts = NO3_RATIO + NH4_RATIO
         self.target_no3 = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
         self.target_nh4 = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
+        self.initial_n_profile = {
+            "NO3-": self.target_no3,
+            "NH4+": self.target_nh4
+        }
 
     def calculate(self):
         # 1. Вносим сульфат магния (Mg)
@@ -745,6 +749,9 @@ class NutrientCalculator:
         # Пересчитываем фактическое содержание азота
         self._recalculate_nitrogen()
         
+        # Обновляем финальный профиль
+        self._update_final_profile()
+        
         return self.results
 
     def _apply_fertilizer(self, name, ppm_needed, main_element, additions):
@@ -755,12 +762,18 @@ class NutrientCalculator:
         for element, label in additions.items():
             added_amount = grams * self.fertilizers[name][element] * 1000 / self.volume
             added[label] = round(added_amount, 1)
+            # Уменьшаем дефицит по этому элементу
+            if element in self.final_profile:
+                self.final_profile[element] -= added_amount
         
         self.results[name] = {
             'граммы': round(grams, 3),
             'миллиграммы': int(grams * 1000),
             **added
         }
+        
+        # Уменьшаем дефицит по основному элементу
+        self.final_profile[main_element] -= ppm_needed
 
     def _apply_potassium_sulfate(self):
         # Рассчитываем необходимое количество для покрытия дефицита
@@ -781,6 +794,10 @@ class NutrientCalculator:
             'внесет K': round(added_k, 1),
             'внесет S': round(added_s, 1)
         }
+        
+        # Обновляем финальный профиль
+        self.final_profile['K'] -= added_k
+        self.final_profile['S'] -= added_s
 
     def _apply_potassium_nitrate(self):
         # Рассчитываем остаточную потребность в калии
@@ -797,6 +814,10 @@ class NutrientCalculator:
             'внесет K': round(added_k, 1),
             'внесет NO3': round(added_n, 1)
         }
+        
+        # Обновляем финальный профиль
+        self.final_profile['K'] -= added_k
+        self.final_profile['N (NO3-)'] += added_n
 
     def _recalculate_nitrogen(self):
         # Суммируем весь внесенный азот
@@ -807,30 +828,41 @@ class NutrientCalculator:
         self.profile['N (NO3-)'] = total_no3
         self.profile['N (NH4+)'] = total_nh4
 
+    def _update_final_profile(self):
+        # Обновляем финальный профиль с учетом внесенных элементов
+        for fert, data in self.results.items():
+            for key, value in data.items():
+                if key.startswith('внесет'):
+                    element = key.split(': ')[0].replace('внесет ', '')
+                    if element == 'NO3':
+                        element = 'N (NO3-)'
+                    elif element == 'NH4':
+                        element = 'N (NH4+)'
+                    
+                    if element in self.final_profile:
+                        self.final_profile[element] -= value
+
     def calculate_ec(self):
         return round(self.total_ppm / 700, 2)
 
     def print_report(self):
         print("\n" + "="*60)
-        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА:")
+        print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
         print("="*60)
-        
-        # Выводим базовые значения + пересчитанный азот
-        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()]
+        table = [[el, round(val, 1)] for el, val in self.profile.items()]
         print(tabulate(table, headers=["Элемент", "ppm"]))
-        
+
+        print("\nИсходный расчёт азота:")
+        for form, value in self.initial_n_profile.items():
+            print(f"  {form}: {value} ppm")
+
         print("\n" + "="*60)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         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 штук):")
+        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
         fert_table = []
         for fert, data in self.results.items():
             details = [f"+{k}: {v} ppm" for k, v in data.items() if k.startswith('внесет')]
@@ -842,6 +874,14 @@ class NutrientCalculator:
             ])
         print(tabulate(fert_table, headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
 
+        print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
+        deficit = {k: round(abs(v), 1) for k, v in self.final_profile.items() if abs(v) > 0.1}
+        if deficit:
+            for el, val in deficit.items():
+                print(f"  {el}: {val} ppm")
+        else:
+            print("  Все элементы покрыты полностью")
+
 if __name__ == "__main__":
     calculator = NutrientCalculator(volume_liters=VOLUME_LITERS)
     calculator.calculate()