Update app.py

app.py

@@ -14,6 +14,8 @@ from io import BytesIO
 
 
 
+import numpy as np
+from tabulate import tabulate
 
 
 
@@ -679,166 +681,154 @@ def view_image():
 
 
 
+
+# Профиль питательного раствора (ppm = мг/литр)
+TOMATO_PROFILE = {
+    'N (NO3-)': 200,
+    'N (NH4+)': 20,
+    'P': 50,
+    'K': 350,
+    'Mg': 50,
+    'Ca': 200,
+    'S': 100
+}
+
+# База данных удобрений (монокомпоненты)
 fertilizers_db = {
-    # Кальциевая селитра (Ca(NO3)2 · 4H2O - тетрагидрат)
     "Кальциевая селитра": {
-        "N (NO3-)": 0.118,  # 11.8% азота (правильно)
-        "Ca": 0.169         # 16.9% кальция (правильно)
+        "N (NO3-)": 0.118,  # 11.8% N
+        "Ca": 0.169         # 16.9% Ca
     },
-    
-    # Калий азотнокислый (KNO3)
     "Калий азотнокислый": {
-        "N (NO3-)": 0.138,  # 13.8% азота (правильно)
-        "K": 0.387          # 38.7% калия (правильно)
+        "N (NO3-)": 0.138,  # 13.8% N
+        "K": 0.387          # 38.7% K
     },
-    
-    # Аммоний азотнокислый (NH4NO3)
     "Аммоний азотнокислый": {
-        "N (NO3-)": 0.175,  # 17.5% нитратного азота
-        "N (NH4+)": 0.175   # 17.5% аммонийного азота
+        "N (NO3-)": 0.175,  # 17.5% N
+        "N (NH4+)": 0.175   # 17.5% N
     },
-    
-    # Сульфат магния (MgSO4 · 7H2O - гептагидрат)
     "Сульфат магния": {
-        "Mg": 0.098,  # 9.8% магния (правильно)
-        "S": 0.13     # 13% серы (правильно)
+        "Mg": 0.098,  # 9.8% Mg
+        "S": 0.13     # 13% S
     },
-    
-    # Монофосфат калия (KH2PO4)
     "Монофосфат калия": {
-        "P": 0.227,  # 22.7% фосфора (правильно)
-        "K": 0.287   # 28.7% калия (правильно)
+        "P": 0.227,  # 22.7% P
+        "K": 0.287   # 28.7% K
     }
 }
 
-class HydroCalculator:
+class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
-        self.volume = volume_liters
-        self.fertilizers = []
-
-    def add_fertilizer(self, name, grams):
-        if name not in fertilizers_db:
-            raise ValueError(f"Удобрение '{name}' не найдено в базе!")
-        self.fertilizers.append((name, grams))
-
-    def calculate_ppm(self):
-        ppm = {}
-        for name, grams in self.fertilizers:
-            for element, ratio in fertilizers_db[name].items():
-                ppm[element] = ppm.get(element, 0) + (grams * ratio * 1000) / (self.volume * 10)
-        return ppm
-
-    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
-
-    def calculate_ec_exact(self):
-        MOLAR_MASSES = {
-            'NO3-': 62.0,
-            'NH4+': 18.0,
-            'K+': 39.1,
-            'Ca+2': 40.1,
-            'Mg+2': 24.3,
-            'SO4-2': 96.1,
-            'H2PO4-': 97.0,
-        }
-        ION_CONDUCTIVITY = {
-            'NO3-': 71.4,
-            'NH4+': 73.5,
-            'K+': 73.5,
-            'Ca+2': 119.0,
-            'Mg+2': 106.0,
-            'SO4-2': 160.0,
-            'H2PO4-': 33.0,
-        }
-        ELEMENT_TO_ION = {
-            'N (NO3-)': 'NO3-',
-            'N (NH4+)': 'NH4+',
-            'K': 'K+',
-            'Ca': 'Ca+2',
-            'Mg': 'Mg+2',
-            'S': 'SO4-2',
-            'P': 'H2PO4-',
-        }
-
-        ppm = self.calculate_ppm()
-        ec = 0.0  # µS/cm
-
-        for element, value in ppm.items():
-            if element in ELEMENT_TO_ION:
-                ion = ELEMENT_TO_ION[element]
-                molar_mass = MOLAR_MASSES[ion]
-                conductivity = ION_CONDUCTIVITY[ion]
-                # ppm = mg/L → mmol/L = (ppm / molar_mass)
-                mmol_per_liter = value / molar_mass
-                # µS/cm = mmol/L * µS·cm²/mmol
-                ec += mmol_per_liter * conductivity
-
-        return ec / 100  # переводим в mS/cm
-
-    def print_report(self):
-        ppm = self.calculate_ppm()
-        ec_simple = self.calculate_ec()
-        ec_exact = self.calculate_ec_exact()
-
-         # Вывод основной таблицы элементов
-        print("\n" + "="*50)
-        print("ТАБЛИЦА КОНЦЕНТРАЦИЙ ЭЛЕМЕНТОВ")
-        print("="*50)
-        print("{:<15} {:>15} {:>15}".format("Элемент", "Конц. (ppm)", "Конц. (мг/л)"))
-        print("-"*50)
-        for element, value in sorted(ppm.items()):
-            print("{:<15} {:>15.2f} {:>15.2f}".format(element, value, value))
-        print("="*50)
-
-        # Вывод данных по азоту
-        print("\n" + "="*50)
-        print("АНАЛИЗ АЗОТНЫХ СОЕДИНЕНИЙ")
-        print("="*50)
-        print(f"Общий азот (N): {ppm.get('N (NO3-)', 0) + ppm.get('N (NH4+)', 0):.2f} ppm")
-        print(f"Нитратный азот (NO3-): {ppm.get('N (NO3-)', 0):.2f} ppm")
-        print(f"Аммонийный азот (NH4+): {ppm.get('N (NH4+)', 0):.2f} ppm")
-        print("="*50)
-
-        # Вывод электропроводности
+        self.volume = volume_liters  # Объем раствора в литрах
+        self.results = {}
+        
+    def calculate(self, profile):
+        """Основной расчет"""
+        # Сначала считаем удобрения, содержащие только один нужный элемент
+        self._calc_single_component(profile)
+        # Затем комплексные удобрения
+        self._calc_complex(profile)
+        return self.results
+    
+    def _calc_single_component(self, profile):
+        """Расчет для простых солей"""
+        # Сульфат магния (Mg + S)
+        if 'Mg' in profile and 'S' in profile:
+            mg_needed = profile['Mg']
+            s_needed = profile['S']
+            
+            # Вычисляем через Mg
+            mg_grams = (mg_needed * self.volume) / (fertilizers_db["Сульфат магния"]["Mg"] * 1000)
+            # Пересчитываем сколько S мы получим
+            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
+    
+    def _calc_complex(self, profile):
+        """Расчет для комплексных удобрений"""
+        # Кальциевая селитра (Ca + NO3)
+        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
+        
+        # Монофосфат калия (P + K)
+        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
+        
+        # Калий азотнокислый (K + NO3)
+        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
+        
+        # Аммоний азотнокислый (NH4 + NO3)
+        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
+
+    def print_report(self, profile):
+        """Красивый вывод результатов"""
         print("\n" + "="*50)
-        print("ЭЛЕКТРОПРОВОДНОСТЬ РАСТВОРА")
+        print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
         print("="*50)
-        print(f"EC: {ec_simple:.2f} mS/cm")
-        print("="*50)      
-
         
-
-# Пример использования с правильными названиями удобрений
-calc = HydroCalculator(10)  # 10 литров раствора
-
-# Добавляем удобрения с правильными названиями и адекватными количествами
-calc.add_fertilizer("Кальциевая селитра", 5)  # 10 г
-calc.add_fertilizer("Калий азотнокислый", 5)   # 5 г
-calc.add_fertilizer("Аммоний азотнокислый", 5) # 3 г
-
-# Выводим полный отчёт
-calc.print_report()
-
-# Дополнительный вывод только NO3 и NH4
-ppm = calc.calculate_ppm()
-print("\nСодержание азота:")
-print(f"NO3-: {ppm.get('N (NO3-)', 0):.2f} ppm")
-print(f"NH4+: {ppm.get('N (NH4+)', 0):.2f} ppm")
-
+        print("\nЦЕЛЕВОЙ ПРОФИЛЬ (ppm):")
+        print(tabulate([(k, v) for k, v in profile.items()], 
+                      headers=["Элемент", "Концентрация"]))
+        
+        print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
+        table = []
+        for fert, data in self.results.items():
+            table.append([
+                fert,
+                f"{data['граммы']:.3f} г",
+                f"{data['миллиграммы']} мг",
+                f"+{list(data.keys())[2]} {data[list(data.keys())[2]]} ppm"
+            ])
+        print(tabulate(table, 
+                      headers=["Удобрение", "Граммы", "Миллиграммы", "Дополнительно"]))
+        
+        print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
+        print(tabulate([(k, v) for k, v in profile.items() if v > 0.1], 
+                      headers=["Элемент", "Недостача (ppm)"]))
+
+# Пример использования
+calc = NutrientCalculator(volume_liters=10)  # Для 10 литров раствора
+results = calc.calculate(TOMATO_PROFILE.copy())
+calc.print_report(TOMATO_PROFILE.copy())