Update app.py

app.py

@@ -681,8 +681,6 @@ def view_image():
 
 
 
-
-
 TOMATO_PROFILE = {
     'N (NO3-)': 200,
     'N (NH4+)': 20,
@@ -697,129 +695,112 @@ fertilizers_db = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118,
         "Ca": 0.169,
-        "EC_factor": 1.2  # мСм/см на 1 г/л
     },
     "Калий азотнокислый": {
         "N (NO3-)": 0.138,
         "K": 0.387,
-        "EC_factor": 1.4
     },
     "Аммоний азотнокислый": {
         "N (NO3-)": 0.175,
         "N (NH4+)": 0.175,
-        "EC_factor": 1.5
     },
     "Сульфат магния": {
         "Mg": 0.098,
         "S": 0.13,
-        "EC_factor": 1.0
     },
     "Монофосфат калия": {
         "P": 0.227,
         "K": 0.287,
-        "EC_factor": 0.7
     }
 }
 
+
 class NutrientCalculator:
     def __init__(self, volume_liters=1.0):
         self.volume = volume_liters
         self.results = {}
-        self.ec_total = 0.0
-        
+
     def calculate(self, profile):
-        self._calc_single_component(profile)
-        self._calc_complex(profile)
+        self._apply_magnesium_sulfate(profile)
+        self._apply_calcium_nitrate(profile)
+        self._apply_mkp(profile)
+        self._apply_potassium_nitrate(profile)
+        self._apply_ammonium_nitrate(profile)
         return self.results
-    
-    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']
-            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_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_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_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_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_ec("Аммоний азотнокислый", nh4_grams)
+
+    def _apply_fertilizer(self, fert_name, grams, additions):
+        self.results[fert_name] = {
+            'граммы': round(grams, 3),
+            'миллиграммы': int(grams * 1000),
+        }
+        self.results[fert_name].update(additions)
+
+    def _apply_magnesium_sulfate(self, profile):
+        mg_need = profile['Mg']
+        mg_content = fertilizers_db["Сульфат магния"]["Mg"]
+        grams = (mg_need * self.volume) / (mg_content * 1000)
+        added_s = grams * fertilizers_db["Сульфат магния"]["S"] * 1000 / self.volume
+
+        profile['S'] -= added_s
+        self._apply_fertilizer("Сульфат магния", grams, {'внесет S': round(added_s, 1)})
+
+    def _apply_calcium_nitrate(self, profile):
+        ca_need = profile['Ca']
+        ca_content = fertilizers_db["Кальциевая селитра"]["Ca"]
+        grams = (ca_need * self.volume) / (ca_content * 1000)
+        added_n = grams * fertilizers_db["Кальциевая селитра"]["N (NO3-)"] * 1000 / self.volume
+
+        profile['N (NO3-)'] -= added_n
+        self._apply_fertilizer("Кальциевая селитра", grams, {'внесет NO3': round(added_n, 1)})
+
+    def _apply_mkp(self, profile):
+        p_need = profile['P']
+        p_content = fertilizers_db["Монофосфат калия"]["P"]
+        grams = (p_need * self.volume) / (p_content * 1000)
+        added_k = grams * fertilizers_db["Монофосфат калия"]["K"] * 1000 / self.volume
+
+        profile['K'] -= added_k
+        self._apply_fertilizer("Монофосфат калия", grams, {'внесет K': round(added_k, 1)})
+
+    def _apply_potassium_nitrate(self, profile):
+        k_need = profile['K']
+        if k_need <= 0:
+            return
+        k_content = fertilizers_db["Калий азотнокислый"]["K"]
+        grams = (k_need * self.volume) / (k_content * 1000)
+        added_n = grams * fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+
+        profile['N (NO3-)'] -= added_n
+        self._apply_fertilizer("Калий азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
+
+    def _apply_ammonium_nitrate(self, profile):
+        nh4_need = profile['N (NH4+)']
+        if nh4_need <= 0:
+            return
+        nh4_content = fertilizers_db["Аммоний азотнокислый"]["N (NH4+)"]
+        grams = (nh4_need * self.volume) / (nh4_content * 1000)
+        added_n = grams * fertilizers_db["Аммоний азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+
+        profile['N (NO3-)'] -= added_n
+        self._apply_fertilizer("Аммоний азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
 
     def print_report(self, profile):
-        print("\n" + "="*50)
+        print("=" * 50)
         print(f"РАСЧЕТ ДЛЯ {self.volume} ЛИТРОВ РАСТВОРА")
-        print("="*50)
-        
-        print("\nЦЕЛЕВОЙ ПРОФИЛЬ (ppm):")
-        print(tabulate([(k, v) for k, v in profile.items()], 
-                      headers=["Элемент", "Концентрация"]))
+        print("=" * 50)
 
         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(f"\n🌊 Примерная электропроводимость (EC): {round(self.ec_total, 2)} мСм/см")
+            print(f"- {fert}: {data['граммы']} г ({data['миллиграммы']} мг)")
+            for k, v in data.items():
+                if k.startswith("внесет"):
+                    print(f"  ➕ {k}: {v} ppm")
 
         print("\nОСТАТОЧНЫЙ ДЕФИЦИТ:")
-        print(tabulate([(k, v) for k, v in profile.items() if v > 0.1], 
-                      headers=["Элемент", "Недостача (ppm)"]))
+        for el, val in profile.items():
+            if val > 0.1:
+                print(f"  🔸 {el}: {round(val, 1)} ppm")
+
 
 # Пример использования
 calc = NutrientCalculator(volume_liters=10)
@@ -844,7 +825,6 @@ calc.print_report(TOMATO_PROFILE.copy())
 
 
 
-
 
 
 if __name__ == '__main__':