--- a
+++ b/index.py
@@ -0,0 +1,66 @@
+import json
+import os
+
+from app import app
+from flask import render_template, request
+from werkzeug.utils import secure_filename
+
+from classifier import getPrediction
+
+
+
+@app.route('/')
+def index():
+    return render_template('about.html')
+
+
+@app.route('/upload-image/', methods=['POST'])
+def submit_file():
+    if request.method == 'POST':
+        if 'file' not in request.files:
+            data = "File not found in request body"
+            response = app.response_class(
+                response=json.dumps(data),
+                mimetype='application/json'
+            )
+            # flash('No file part')
+            # return redirect(request.url)
+            return response
+        file = request.files['file']
+        if file.filename == '':
+            data = 'No file selected for uploading'
+            response = app.response_class(
+                response=json.dumps(data),
+                mimetype='application/json'
+            )
+            # flash('No file selected for uploading')
+            # return redirect(request.url)
+            return response
+        if file:
+            filename = secure_filename(file.filename)
+            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            # upload images to DB
+            # upload_image_data_mongodb(file)
+            label = getPrediction.processImg(filename)
+            # flash({"Given class of image is ": label})
+            data = {"Given class of image is ": label}
+            response = app.response_class(
+                response=json.dumps(data),
+                mimetype='application/json'
+            )
+            if remove_img(path=os.path.join(app.config['UPLOAD_FOLDER']), img_name=file.filename):
+                print('Image removed')
+            # return redirect('/')
+            return response
+
+
+def remove_img(path, img_name):
+    os.remove(path + '/' + img_name)
+    # check if file exists or not
+    if os.path.exists(path + '/' + img_name) is False:
+        # file did not exists
+        return True
+
+
+if __name__ == "__main__":
+    app.run()