Merge branch 'IT18058338' into 'master'

It18058338

See merge request !23

FlaskAPI/review_prediction/app/main.py

@@ -13,6 +13,14 @@ import pickle
 
 print('Loading Model ...')
 model = keras.models.load_model('app/model.h5')
+
+# marks prediction model
+model1 = pickle.load(open('app/g3_model.pkl', 'rb'))
+# pass/fail prediction model
+model2 = pickle.load(open('app/passfail_model.pkl','rb'))
+# grade prediction model
+model3 = pickle.load(open('app/grade_model.pkl','rb'))
+
 print('Model Loaded')
 
 # Enabling Cors
@@ -166,3 +174,43 @@ def review_prediction_for_tutor():
 
     
     return jsonify({"result":finalyPrecentage})
+
+#------------------------- IT18058338 Flask API Routes ------------------------------
+
+@app.route('/predict_quiz_marks', methods=['POST'])
+def predict_quiz_marks():
+    data = request.get_json(force=True)
+    print("json values - marks")
+    print(data.values())
+
+    prediction = model1.predict([np.array(list(data.values()))])
+
+    # output = prediction[0]
+    output = round(prediction[0],2)
+    print(output)
+    return jsonify(output)
+
+@app.route('/predict_pass_fail', methods=['POST'])
+def predict_pass_fail():
+    data = request.get_json(force=True)
+    print("json values - status")
+    print(data.values())
+
+    prediction = model2.predict([np.array(list(data.values()))])
+
+    output = prediction[0]
+    # output = round(prediction[0],2)
+    print(output)
+    return jsonify(int(output))
+
+@app.route('/predict_grade', methods=['POST'])
+def predict_grade():
+    data = request.get_json(force=True)
+    print("json values - grade")
+    print(data.values())
+
+    prediction = model3.predict([np.array(list(data.values()))])
+
+    output = prediction[0]
+    print(output)
+    return jsonify(str(output))