Update app.py

app.py

@@ -683,7 +683,6 @@ def view_image():
 
 
 
-# Профиль питательного раствора (ppm = мг/литр)
 TOMATO_PROFILE = {
     'N (NO3-)': 200,
     'N (NH4+)': 20,
@@ -694,154 +693,113 @@ TOMATO_PROFILE = {
     'S': 100
 }
 
-# База данных удобрений (монокомпоненты)
 fertilizers_db = {
     "Кальциевая селитра": {
-        "N (NO3-)": 0.118,  # 11.8% N
-        "Ca": 0.169         # 16.9% Ca
+        "N (NO3-)": 0.118,
+        "Ca": 0.169,
+        "EC_factor": 1.2  # мСм/см на 1 г/л
     },
     "Калий азотнокислый": {
-        "N (NO3-)": 0.138,  # 13.8% N
-        "K": 0.387          # 38.7% K
+        "N (NO3-)": 0.138,
+        "K": 0.387,
+        "EC_factor": 1.4
     },
     "Аммоний азотнокислый": {
-        "N (NO3-)": 0.175,  # 17.5% N
-        "N (NH4+)": 0.175   # 17.5% N
+        "N (NO3-)": 0.175,
+        "N (NH4+)": 0.175,
+        "EC_factor": 1.5
     },
     "Сульфат магния": {
-        "Mg": 0.098,  # 9.8% Mg
-        "S": 0.13     # 13% S
+        "Mg": 0.098,
+        "S": 0.13,
+        "EC_factor": 1.0
     },
     "Монофосфат калия": {
-        "P": 0.227,  # 22.7% P
-        "K": 0.287   # 28.7% K
+        "P": 0.227,
+        "K": 0.287,
+        "EC_factor": 0.7
     }
 }
 
-# Приблизительный вклад элементов в электропроводимость, мСм/см на 100 ppm
-# Источник: усреднённые данные по литературе (можно адаптировать)
-EC_CONTRIBUTION = {
-    'N (NO3-)': 0.6,
-    'N (NH4+)': 0.7,
-    'P': 0.2,
-    'K': 0.8,
-    'Mg': 0.5,
-    'Ca': 0.4,
-    'S': 0.4
-}
-
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
-        self.volume = volume_liters  # Объем раствора в литрах
+        self.volume = volume_liters
         self.results = {}
-        self.total_ppm = {}  # накопленные ppm по каждому элементу
-    
+        self.ec_total = 0.0
+        
     def calculate(self, profile):
-        """Основной расчет"""
-        self.total_ppm = {key: 0 for key in profile}  # начальная инициализация
         self._calc_single_component(profile)
         self._calc_complex(profile)
         return self.results
     
-    def _add_to_total(self, element, ppm):
-        """Добавляет ppm вещества к общему количеству для расчета EC"""
-        if element in self.total_ppm:
-            self.total_ppm[element] += ppm
+    def _add_ec(self, fert_name, grams):
+        ec_factor = fertilizers_db[fert_name].get("EC_factor", 0)
+        ec_contrib = ec_factor * (grams / self.volume)
+        self.ec_total += ec_contrib
     
     def _calc_single_component(self, profile):
         if 'Mg' in profile and 'S' in profile:
             mg_needed = profile['Mg']
-            s_needed = profile['S']
-            
             mg_grams = (mg_needed * self.volume) / (fertilizers_db["Сульфат магния"]["Mg"] * 1000)
             s_from_mg = mg_grams * fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
-            
             self.results["Сульфат магния"] = {
                 'граммы': round(mg_grams, 3),
                 'миллиграммы': int(mg_grams * 1000),
                 'внесет S': round(s_from_mg, 1)
             }
             profile['S'] -= s_from_mg
-            self._add_to_total('Mg', mg_needed)
-            self._add_to_total('S', s_from_mg)
+            self._add_ec("Сульфат магния", mg_grams)
     
     def _calc_complex(self, profile):
         if 'Ca' in profile and 'N (NO3-)' in profile:
             ca_grams = (profile['Ca'] * self.volume) / (fertilizers_db["Кальциевая селитра"]["Ca"] * 1000)
             no3_from_ca = ca_grams * fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
-            
             self.results["Кальциевая селитра"] = {
                 'граммы': round(ca_grams, 3),
                 'миллиграммы': int(ca_grams * 1000),
                 'внесет NO3': round(no3_from_ca, 1)
             }
             profile['N (NO3-)'] -= no3_from_ca
-            self._add_to_total('Ca', profile['Ca'])
-            self._add_to_total('N (NO3-)', no3_from_ca)
+            self._add_ec("Кальциевая селитра", ca_grams)
         
         if 'P' in profile and 'K' in profile:
             p_grams = (profile['P'] * self.volume) / (fertilizers_db["Монофосфат калия"]["P"] * 1000)
             k_from_p = p_grams * fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
-            
             self.results["Монофосфат калия"] = {
                 'граммы': round(p_grams, 3),
                 'миллиграммы': int(p_grams * 1000),
                 'вне��ет K': round(k_from_p, 1)
             }
             profile['K'] -= k_from_p
-            self._add_to_total('P', profile['P'])
-            self._add_to_total('K', k_from_p)
+            self._add_ec("Монофосфат калия", p_grams)
         
         if 'K' in profile and profile['K'] > 0 and 'N (NO3-)' in profile:
             k_grams = (profile['K'] * self.volume) / (fertilizers_db["Калий азотнокислый"]["K"] * 1000)
             no3_from_k = k_grams * fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
-            
             self.results["Калий азотнокислый"] = {
                 'граммы': round(k_grams, 3),
                 'миллиграммы': int(k_grams * 1000),
                 'внесет NO3': round(no3_from_k, 1)
             }
             profile['N (NO3-)'] -= no3_from_k
-            self._add_to_total('K', profile['K'])
-            self._add_to_total('N (NO3-)', no3_from_k)
+            self._add_ec("Калий азотнокислый", k_grams)
         
         if 'N (NH4+)' in profile and profile['N (NH4+)'] > 0:
             nh4_grams = (profile['N (NH4+)'] * self.volume) / (fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"] * 1000)
             no3_from_nh4 = nh4_grams * fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
-            
             self.results["Аммоний азотнокислый"] = {
                 'граммы': round(nh4_grams, 3),
                 'миллиграммы': int(nh4_grams * 1000),
                 'внесет NO3': round(no3_from_nh4, 1)
             }
             profile['N (NO3-)'] -= no3_from_nh4
-            self._add_to_total('N (NH4+)', profile['N (NH4+)'])
-            self._add_to_total('N (NO3-)', no3_from_nh4)
-
-    def calculate_ec(self):
-        ppm = self.calculate_ppm()
-        ec = 0.0
-        EC_COEFFICIENTS = {
-            'N (NO3-)': 0.6,
-            'N (NH4+)': 0.4,
-            'K': 0.7,
-            'Ca': 0.75,
-            'Mg': 0.6,
-            'S': 0.3,
-            'P': 0.1,
-        }
-        for element, value in ppm.items():
-            if element in EC_COEFFICIENTS:
-                ec += value * EC_COEFFICIENTS[element]
-        return ec / 100
-        
+            self._add_ec("Аммоний азотнокислый", nh4_grams)
 
     def print_report(self, profile):
-        """Красивый вывод результатов"""
         print("\n" + "="*50)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*50)
-
+        
         print("\nЦЕЛЕВОЙ ПРОФИЛЬ (ppm):")
         print(tabulate([(k, v) for k, v in profile.items()], 
                       headers=["Элемент", "Концентрация"]))
@@ -855,19 +813,16 @@ class NutrientCalculator:
                 f"{data['миллиграммы']} мг",
                 f"+{list(data.keys())[2]} {data[list(data.keys())[2]]} ppm"
             ])
-        print(tabulate(table, 
-                      headers=["Удобрение", "Граммы", "Миллиграммы", "Дополнительно"]))
+        print(tabulate(table, headers=["Удобрение", "Граммы", "Миллиграммы", "Дополнительно"]))
+
+        print(f"\n🌊 Примерная электропроводимость (EC): {round(self.ec_total, 2)} мСм/см")
 
         print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
         print(tabulate([(k, v) for k, v in profile.items() if v > 0.1], 
                       headers=["Элемент", "Недостача (ppm)"]))
 
-        # Электропроводимость
-        ec_value = self._calculate_ec()
-        print(f"\nПРИБЛИЗИТЕЛЬНАЯ ЭЛЕКТРОПРОВОДИМОСТЬ (EC): {ec_value} мСм/см")
-
 # Пример использования
-calc = NutrientCalculator(volume_liters=10)  # Для 10 литров раствора
+calc = NutrientCalculator(volume_liters=10)
 results = calc.calculate(TOMATO_PROFILE.copy())
 calc.print_report(TOMATO_PROFILE.copy())