Added database

app.py

@@ -2,15 +2,20 @@ import numpy as np
 from flask import Flask, request, jsonify, render_template
 import pickle
 import json
+from flask_pymongo import PyMongo
 
 app = Flask(__name__) #Initialize the flask App
+
+mongodb_client = PyMongo(app, uri="mongodb://localhost:27017/pred_records")
+db = mongodb_client.db
+
 model = pickle.load(open('model.pickle', 'rb'))
 mlb = pickle.load(open('mlb.pickle','rb'))
 vectorizer = pickle.load(open('vectorizer.pickle','rb'))
 
 @app.route('/')
 def home():
-    return render_template('index.html')
+    return render_template('index.ejs')
 
 @app.route('/predict',methods=['POST'])
 def predict():
@@ -18,10 +23,12 @@ def predict():
     vector = vectorizer.transform([reqInput])
     prediction = model.predict(vector)
     output = json.dumps(mlb.inverse_transform(prediction)[0])
+    db.preds.insert_one({'plot': reqInput, 'prediction': output})
     return {
         "predictionText":output,
         "predictionRaw": json.dumps(prediction[0].tolist())
     }
 
+
 if __name__ == "__main__":
     app.run(threaded=True)
\ No newline at end of file