Update app.py

app.py

@@ -683,6 +683,7 @@ def view_image():
 
 
 
+from tabulate import tabulate
 
 # Профиль питательного раствора для томатов (ppm)
 TOMATO_PROFILE = {
@@ -695,7 +696,7 @@ TOMATO_PROFILE = {
     'S': 100
 }
 
-# База данных удобрений
+# Обновленная база данных удобрений с добавлением сульфата калия
 fertilizers_db = {
     "Кальциевая селитра": {
         "N (NO3-)": 0.118,
@@ -705,6 +706,10 @@ fertilizers_db = {
         "N (NO3-)": 0.138,
         "K": 0.387,
     },
+    "Калий сернокислый": {  # Добавлен новый компонент
+        "K": 0.448,  # 44.8% K
+        "S": 0.184,   # 18.4% S
+    },
     "Аммоний азотнокислый": {
         "N (NO3-)": 0.175,
         "N (NH4+)": 0.175,
@@ -731,7 +736,7 @@ class NutrientCalculator:
         self._apply_magnesium_sulfate()
         self._apply_calcium_nitrate()
         self._apply_mkp()
-        self._apply_potassium_nitrate()
+        self._apply_potassium_fertilizers()  # Измененный метод для работы с двумя калийными удобрениями
         self._apply_ammonium_nitrate()
         return self.results
 
@@ -772,17 +777,54 @@ class NutrientCalculator:
         self._apply_fertilizer("Монофосфат калия", grams, {'внесет K': round(added_k, 1)})
         self.final_profile['P'] = 0
 
-    def _apply_potassium_nitrate(self):
+    def _apply_potassium_fertilizers(self):
+        """Новый метод для распределения калия между двумя удобрениями"""
         k_need = self.final_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
-
-        self.final_profile['N (NO3-)'] -= added_n
-        self._apply_fertilizer("Калий азотнокислый", grams, {'внесет NO3': round(added_n, 1)})
-        self.final_profile['K'] = 0
+            
+        # Сначала вносим часть калия через сернокислый калий для компенсации серы
+        s_deficit = max(0, self.final_profile['S'])
+        if s_deficit > 0:
+            # Рассчитываем сколько сернокислого калия нужно для компенсации дефицита серы
+            s_content = fertilizers_db["Калий сернокислый"]["S"]
+            k2so4_grams = (s_deficit * self.volume) / (s_content * 1000)
+            
+            # Сколько калия это добавит
+            added_k = k2so4_grams * fertilizers_db["Калий сернокислый"]["K"] * 1000 / self.volume
+            
+            # Если этим удобрением мы покрываем больше калия, чем нужно
+            if added_k > k_need:
+                k2so4_grams = (k_need * self.volume) / (fertilizers_db["Калий сернокислый"]["K"] * 1000)
+                added_k = k_need
+                added_s = k2so4_grams * fertilizers_db["Калий сернокислый"]["S"] * 1000 / self.volume
+            else:
+                added_s = s_deficit
+                
+            self._apply_fertilizer(
+                "Калий сернокислый", 
+                k2so4_grams, 
+                {
+                    'внесет K': round(added_k, 1),
+                    'внесет S': round(added_s, 1)
+                }
+            )
+            self.final_profile['K'] -= added_k
+            self.final_profile['S'] -= added_s
+            k_need = self.final_profile['K']
+        
+        # Остаток калия вносим азотнокислым калием
+        if k_need > 0:
+            kno3_grams = (k_need * self.volume) / (fertilizers_db["Калий азотнокислый"]["K"] * 1000)
+            added_n = kno3_grams * fertilizers_db["Калий азотнокислый"]["N (NO3-)"] * 1000 / self.volume
+            
+            self._apply_fertilizer(
+                "Калий азотнокислый", 
+                kno3_grams, 
+                {'внесет NO3': round(added_n, 1)}
+            )
+            self.final_profile['K'] = 0
+            self.final_profile['N (NO3-)'] -= added_n
 
     def _apply_ammonium_nitrate(self):
         nh4_need = self.final_profile['N (NH4+)']
@@ -823,11 +865,12 @@ class NutrientCalculator:
         print("\nРЕКОМЕНДУЕМЫЕ УДОБРЕНИЯ:")
         table = []
         for fert, data in self.results.items():
+            additions = [f"+{k}: {v} ppm" for k, v in data.items() if k.startswith('внесет')]
             table.append([
                 fert,
                 f"{data['граммы']:.3f} г",
                 f"{data['миллиграммы']} мг",
-                f"+{list(data.keys())[2]} {data[list(data.keys())[2]]} ppm"
+                "\n".join(additions)
             ])
         print(tabulate(table, headers=["Удобрение", "Граммы", "Миллиграммы", "Добавит"]))
         
@@ -856,8 +899,5 @@ if __name__ == "__main__":
 
 
 
-
-
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))