Update app.py

app.py

@@ -678,16 +678,15 @@ def view_image():
 
 
 
-
 from tabulate import tabulate
 
-# Константы для расчёта
-TOTAL_NITROGEN = 125  # ppm
-NO3_RATIO = 8.00      # Доля нитратного азота 
-NH4_RATIO = 1.00      # Доля аммонийного азота
-VOLUME_LITERS = 100   # Объём раствора (можно менять)
+# Константы
+TOTAL_NITROGEN = 125
+NO3_RATIO = 8.00
+NH4_RATIO = 1.00
+VOLUME_LITERS = 100
 
-# Базовый профиль (7 элементов)
+# Целевые значения
 BASE_PROFILE = {
     'P': 31.000,
     'K': 210.000,
@@ -698,7 +697,6 @@ BASE_PROFILE = {
     'N (NH4+)': 0
 }
 
-# База удобрений (6 удобрений)
 FERTILIZERS = {
     "Кальциевая селитра": {"N (NO3-)": 0.118, "Ca": 0.169},
     "Калий азотнокислый": {"N (NO3-)": 0.138, "K": 0.387},
@@ -712,169 +710,126 @@ class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
-        self.profile = BASE_PROFILE.copy()
-        self.total_ppm = sum(BASE_PROFILE.values()) + TOTAL_NITROGEN
+        self.target_profile = BASE_PROFILE.copy()
+        self.actual_profile = {k: 0.0 for k in BASE_PROFILE}
         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.target_profile['N (NO3-)'] = TOTAL_NITROGEN * (NO3_RATIO / total_parts)
+        self.target_profile['N (NH4+)'] = TOTAL_NITROGEN * (NH4_RATIO / total_parts)
         self.initial_n_profile = {
-            "NO3-": self.target_no3,
-            "NH4+": self.target_nh4
+            "NO3-": self.target_profile['N (NO3-)'],
+            "NH4+": self.target_profile['N (NH4+)']
         }
 
     def calculate(self):
-        # 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"})
-        
-        # 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()
-        
-        # Пересчитываем фактическое содержание азота
-        self._recalculate_nitrogen()
-        
-        # Обновляем финальный профиль
-        self._update_final_profile()
-        
+        self._apply("Сульфат магния", "Mg", 24.0)
+        self._apply("Кальциевая селитра", "Ca", 84.0)
+        self._apply("Калий фосфорнокислый", "P", 31.0)
+        self._apply("Аммоний азотнокислый", "N (NH4+)", self.target_profile['N (NH4+)'])
+        self._apply_k_sulfate()
+        self._apply_k_nitrate()
         return self.results
 
-    def _apply_fertilizer(self, name, ppm_needed, main_element, additions):
-        content = self.fertilizers[name][main_element]
-        grams = (ppm_needed * self.volume) / (content * 1000)
-        
-        added = {}
-        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] = {
+    def _apply(self, fert_name, main_element, required_ppm):
+        content = self.fertilizers[fert_name][main_element]
+        grams = required_ppm * self.volume / (content * 1000)
+        result = {
             'граммы': round(grams, 3),
-            'миллиграммы': int(grams * 1000),
-            **added
+            'миллиграммы': int(grams * 1000)
         }
-        
-        # Уменьшаем дефицит по основному элементу
-        self.final_profile[main_element] -= ppm_needed
 
-    def _apply_potassium_sulfate(self):
-        # Рассчитываем необходимое количество для покрытия дефицита
-        k_content = self.fertilizers["Калий сернокислый"]["K"]
-        s_content = self.fertilizers["Калий сернокислый"]["S"]
-        
-        # Рассчитываем граммы для покрытия дефицита
-        grams_k = (self.profile['K'] * self.volume) / (k_content * 1000)
-        grams_s = (self.profile['S'] * self.volume) / (s_content * 1000)
-        grams = max(grams_k, grams_s)  # Берем максимальное значение
-        
-        added_k = grams * k_content * 1000 / self.volume
-        added_s = grams * s_content * 1000 / self.volume
+        for element, percent in self.fertilizers[fert_name].items():
+            added_ppm = grams * percent * 1000 / self.volume
+            result[f'внесет {self._label(element)}'] = round(added_ppm, 1)
+            self.actual_profile[element] += added_ppm
 
-        self.results["Калий сернокислый"] = {
+        self.results[fert_name] = result
+
+    def _apply_k_sulfate(self):
+        fert = "Калий сернокислый"
+        k_def = self.target_profile['K'] - self.actual_profile['K']
+        s_def = self.target_profile['S'] - self.actual_profile['S']
+        k_content = self.fertilizers[fert]["K"]
+        s_content = self.fertilizers[fert]["S"]
+
+        grams_k = k_def * self.volume / (k_content * 1000) if k_content else 0
+        grams_s = s_def * self.volume / (s_content * 1000) if s_content else 0
+        grams = max(grams_k, grams_s)
+
+        result = {
             'граммы': round(grams, 3),
-            'миллиграммы': int(grams * 1000),
-            'внесет K': round(added_k, 1),
-            'внесет S': round(added_s, 1)
+            'миллиграммы': int(grams * 1000)
         }
-        
-        # Обновляем финальный профиль
-        self.final_profile['K'] -= added_k
-        self.final_profile['S'] -= added_s
-
-    def _apply_potassium_nitrate(self):
-        # Рассчитываем остаточную потребность в калии
-        k_content = self.fertilizers["Калий азотнокислый"]["K"]
-        n_content = self.fertilizers["Калий азотнокислый"]["N (NO3-)"]
-        
-        grams = (self.profile['K'] * self.volume) / (k_content * 1000)
-        added_k = grams * k_content * 1000 / self.volume
-        added_n = grams * n_content * 1000 / self.volume
 
-        self.results["Калий азотнокислый"] = {
+        for el in ["K", "S"]:
+            ppm = grams * self.fertilizers[fert][el] * 1000 / self.volume
+            result[f"внесет {el}"] = round(ppm, 1)
+            self.actual_profile[el] += ppm
+
+        self.results[fert] = result
+
+    def _apply_k_nitrate(self):
+        fert = "Калий азотнокислый"
+        k_def = self.target_profile['K'] - self.actual_profile['K']
+        k_content = self.fertilizers[fert]["K"]
+        grams = k_def * self.volume / (k_content * 1000)
+
+        result = {
             'граммы': round(grams, 3),
-            'миллиграммы': int(grams * 1000),
-            'внесет K': round(added_k, 1),
-            'внесет NO3': round(added_n, 1)
+            'миллиграммы': int(grams * 1000)
         }
-        
-        # Обновляем финальный профиль
-        self.final_profile['K'] -= added_k
-        self.final_profile['N (NO3-)'] += added_n
-
-    def _recalculate_nitrogen(self):
-        # Суммируем весь внесенный азот
-        total_no3 = sum(f.get('внесет NO3', 0) for f in self.results.values())
-        total_nh4 = sum(f.get('внесет NH4', 0) for f in self.results.values())
-        
-        # Обновляем профиль
-        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
+        for el in ["K", "N (NO3-)"]:
+            ppm = grams * self.fertilizers[fert][el] * 1000 / self.volume
+            result[f"внесет {self._label(el)}"] = round(ppm, 1)
+            self.actual_profile[el] += ppm
+
+        self.results[fert] = result
+
+    def _label(self, el):
+        return "NO3" if el == "N (NO3-)" else "NH4" if el == "N (NH4+)" else el
 
     def calculate_ec(self):
-        return round(self.total_ppm / 700, 2)
+        total_ppm = sum(self.target_profile.values())
+        return round(total_ppm / 700, 2)
 
     def print_report(self):
         print("\n" + "="*60)
         print("ПРОФИЛЬ ПИТАТЕЛЬНОГО РАСТВОРА (ИТОГО):")
         print("="*60)
-        table = [[el, round(val, 1)] for el, val in self.profile.items()]
+        table = [[el, round(self.actual_profile[el], 1)] for el in self.actual_profile]
         print(tabulate(table, headers=["Элемент", "ppm"]))
 
         print("\nИсходный расчёт азота:")
-        for form, value in self.initial_n_profile.items():
-            print(f"  {form}: {value} ppm")
+        for form, val in self.initial_n_profile.items():
+            print(f"  {form}: {round(val, 1)} ppm")
 
         print("\n" + "="*60)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*60)
-        print(f"Общая концентрация: {round(self.total_ppm, 1)} ppm")
+        print(f"Общая концентрация: {round(sum(self.target_profile.values()), 1)} ppm")
         print(f"EC: {self.calculate_ec()} mS/cm")
 
         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('внесет')]
+            adds = [f"+{k}: {v} ppm" for k, v in data.items() if k.startswith('внесет')]
             fert_table.append([
                 fert,
-                f"{data['граммы']} г",
-                f"{data['миллиграммы']} мг",
-                "\n".join(details)
+                data['граммы'],
+                data['миллиграммы'],
+                "\n".join(adds)
             ])
         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}
+        deficit = {
+            k: round(self.target_profile[k] - self.actual_profile[k], 1)
+            for k in self.target_profile
+            if abs(self.target_profile[k] - self.actual_profile[k]) > 0.1
+        }
         if deficit:
             for el, val in deficit.items():
                 print(f"  {el}: {val} ppm")
@@ -888,5 +843,6 @@ if __name__ == "__main__":
 
 
 
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))