awd

app.py

@@ -61,10 +61,22 @@ def get_fertilizer_recommendation(row, land_size_m2, fallow_years, thresholds, a
     for nutrient, threshold in thresholds.items():
         if row[nutrient] < threshold:
             nutrient_name = nutrient.split('_')[-1]
-            deficiencies.append(nutrient_name)
+            full_nutrient_name = {
+                'P': 'Phosphorus',
+                'K': 'Potassium',
+                'Ca': 'Calcium',
+                'Mg': 'Magnesium',
+                'S': 'Sulphur',
+                'Zn': 'Zinc',
+                'B': 'Boron',
+                'Fe': 'Iron',
+                'Cu': 'Copper',
+                'N': 'Nitrogen'
+            }[nutrient_name]
+            deficiencies.append(full_nutrient_name)
             base_amount_per_m2 = application_rates[nutrient_name] / 10000
             total_amount = base_amount_per_m2 * land_size_m2 * (1 + 0.1 * fallow_years)
-            fertilizer_amounts[nutrient_name] = round(total_amount, 2)
+            fertilizer_amounts[full_nutrient_name] = round(total_amount, 2)
     if deficiencies:
         return {'recommendation': f'fertilizer needed for {", ".join(deficiencies)}', 'fertilizer_amounts': fertilizer_amounts}
     else:
@@ -104,4 +116,4 @@ def gradio_application():
         demo.launch()
 
 if __name__ == "__main__":
-    gradio_application()
+    gradio_application()