fertilizer recommendation complete

app.py

@@ -1,4 +1,5 @@
 from config import crop_model, crop_pipeline_encoder, crop_label_encoder
+from config import fertilizer_model, fertilizer_pipeline_encoder, fertilizer_label_encoder
 from utils import retrieve_image_by_name_from_mongodb, retrieve_data
 from flask import Flask, request, render_template, jsonify
 import requests
@@ -49,10 +50,44 @@ def crop_recommendation_output():
     return render_template('crop_recommendation_output.html', image_data_base64=image_data_base64, input_file_name=label[0], crop_details=crop_details)
 
 
-@app.route('/fertilizer_recommendation')
+@app.route('/fertilizer_recommendation', methods=['GET', 'POST'])
 def fertilizer_recommendation():
     return render_template('fertilizer_recommendation_input.html')
 
+@app.route('/fertilizer_recommendation_output', methods=['GET', 'POST'])
+def fertilizer_recommendation_output():
+    temperature = request.form.get("temperature")
+    humidity = request.form.get("humidity")
+    moisture = request.form.get("moisture")
+    nitrogen = request.form.get("nitrogen")
+    potassium = request.form.get("potassium")
+    phosphorous = request.form.get("phosphorous")
+    soil_type = request.form.get("soil_type")
+    crop_type = request.form.get("crop_type")
+
+    input_data = [int(temperature), int(humidity), int(moisture), soil_type, crop_type, int(nitrogen), int(potassium), int(phosphorous)]
+    input_array = np.array(input_data).reshape(-1, 8)
+    
+    transformed_data = fertilizer_pipeline_encoder.transform(input_array)
+    model_prediction = fertilizer_model.predict(transformed_data).astype(int)
+
+    label = fertilizer_label_encoder.inverse_transform(model_prediction)
+
+    # retrieving the image from mongodb dabase
+    image_data = retrieve_image_by_name_from_mongodb(database_name=os.getenv("FERTILIZER_DB_NAME"),
+                                                        collection_name=os.getenv("FERTILIZER_IMAGE_COLLECTION_NAME"),
+                                                        file_name=str(label[0]))
+    
+    # encoding the byte data recieved from the mongodb
+    image_data_base64 = base64.b64encode(image_data).decode('utf-8')
+
+    # retrieving text data from mongodb 
+    fertilizer_details = retrieve_data(database_name=os.getenv("FERTILIZER_DB_NAME"), collection_name= os.getenv("FERTILIZER_INFO_COLLECTION_NAME"), search_query=label[0])
+
+
+    return render_template('fertilizer_recommendation_ouput.html', image_data_base64=image_data_base64, label= label[0], fertilizer_details=fertilizer_details)
+
+
 @app.route('/image_classification')
 def image_classification():
     return render_template('image_classification_input.html')

config.py

@@ -1,6 +1,12 @@
 from artifacts import crop_model_path, crop_transformer_path, crop_target_encoder_path
+from artifacts import fertilizer_model_path, fertilizer_transformer_path, fertilizer_target_encoder_path
+
 from utils import load_model_and_encoders
 
 crop_model, crop_pipeline_encoder, crop_label_encoder = load_model_and_encoders(model_path=crop_model_path, 
                                                                                 transformer_path=crop_transformer_path,
-                                                                                target_encoder_path=crop_target_encoder_path)
+                                                                                target_encoder_path=crop_target_encoder_path)
+
+fertilizer_model, fertilizer_pipeline_encoder, fertilizer_label_encoder = load_model_and_encoders(model_path=fertilizer_model_path, 
+                                                                                transformer_path=fertilizer_transformer_path,
+                                                                                target_encoder_path=fertilizer_target_encoder_path)

templates/fertilizer_recommendation_input.html

@@ -8,7 +8,7 @@
 <div
   class="container-fluid recommendation-input-container d-flex justify-content-center"
 >
-  <form class="row g-3" method="GET" action="/fertilizer_recommendation_output">
+  <form class="row g-3" method="POST" action="/fertilizer_recommendation_output">
     <h3 class="text-center">AI-Powered Fertilizer Recommendations</h3>
     <div class="col-md-4" name="temperature">
       <label for="temperature" class="form-label">Temperature</label>

templates/fertilizer_recommendation_ouput.html

@@ -22,8 +22,8 @@
       <div
         class="col-md-6 text d-flex flex-column justify-content-center align-items-center"
       >
-        <h1 class="text-center">{{input_file_name.upper()}}</h1>
-        <p class="text-center">{{crop_details}}</p>
+        <h1 class="text-center">{{label.upper()}}</h1>
+        <p class="text-center">{{fertilizer_details}}</p>
       </div>
     </div>
     <a href="{{url_for('fertilizer_recommendation')}}" class="btn btn-primary" role="button"