Update app.py

app.py

@@ -824,9 +824,12 @@ class NutrientCalculator:
     def _apply(self, fert_name, main_element, required_ppm):
         if required_ppm <= 0:
             return
-            
+
         try:
-            content = self.fertilizers[fert_name][main_element]
+            content = self.fertilizers[fert_name].get(main_element, 0)
+            if content == 0:
+                raise KeyError(f"Элемент {main_element} отсутствует в удобрении {fert_name}")
+
             grams = (required_ppm * self.volume) / (content * 1000)
             
             if fert_name not in self.results:
@@ -852,7 +855,7 @@ class NutrientCalculator:
             self.results[fert_name]['вклад в EC'] += fert_ec
             self.total_ec += fert_ec
         except KeyError as e:
-            print(f"Ошибка: отсутствует элемент {str(e)} в удобрении {fert_name}")
+            print(f"Ошибка: {str(e)}")
             raise
 
     def _compensate_element(self, element):
@@ -865,8 +868,7 @@ class NutrientCalculator:
         candidates = []
         for fert_name, fert_data in self.fertilizers.items():
             if element in fert_data:
-                weight = next((w['weight'] for w in self.compensation_weights.values() 
-                             if w['fert'] == fert_name), 0.1)
+                weight = self.compensation_weights.get(fert_name, {}).get("weight", 0.1)
                 candidates.append({
                     'name': fert_name,
                     'weight': weight,
@@ -1000,8 +1002,6 @@ if __name__ == "__main__":
     except Exception as e:
         print(f"Критическая ошибка: {str(e)}")
 
-
-
 @app.route('/calculation', methods=['POST'])
 def handle_calculation():
     try: