app_basic.py

#import MySQLdb as mdb
import pymongo
import sys
import os
import pprint
import unicodedata
from pymongo import MongoClient
from flask import jsonify
from flask import Flask, render_template, request,redirect, url_for ,session,escape,json
import cv2, os
import numpy as np
from PIL import Image
import algorithm
from algorithm import predict_confidence


# Connection with MongoDB database :
client = MongoClient('mongodb://akhilesh_123:cmpe273@ds123361.mlab.com:23361/cmpe273')
db = client['cmpe273']
collection = db['photorecog']

filename = '';
originalImage = ''
tobeCompared = ''

APP_ROOT = os.path.dirname(os.path.abspath(__file__))
print "app path : " + APP_ROOT

app = Flask(__name__,static_url_path='/static', template_folder="templates")


#secret key
app.secret_key = 'dsjksdh88989djj'


@app.after_request
def add_header(r):
    """
    Add headers to both force latest IE rendering engine or Chrome Frame,
    and also to cache the rendered page for 10 minutes.
    """
    r.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
    r.headers["Pragma"] = "no-cache"
    r.headers["Expires"] = "0"
    r.headers['Cache-Control'] = 'public, max-age=0'
    return r

@app.route("/",methods=["GET"])
def defaultPage():
    print "Arrived in index - ROOT"
    return render_template("Index.html")

@app.route("/home",methods=["GET"])
def homePage():
    print "Arrived in index - home"
    return render_template("Home.html")

@app.route("/compare",methods=["GET"])
def comparePage():
    print "Arrived in index - Compare"
    if 'userId' in session:
        print session['userId']
        print "found"
        return render_template("compare.html")
    else:
        print "not found"
        return redirect(url_for('loginPage'))

@app.route("/login",methods=["GET"])
def loginPage():
    print "Arrived in index - Login"
    return render_template("login.html")

@app.route("/signUp",methods=["GET"])
def signUpPage():
    print "Arrived in index - signUp"
    return render_template("signUp.html")


@app.route("/complete",methods=["GET"])
def completePage():
    print "Arrived in index - complete"
    if 'userId' in session:
        print session['userId']
        return render_template("complete.html")
    else:
        print 'user is not logged in'
        return redirect(url_for('loginPage'))

@app.route("/upload",methods=["GET"])
def uploadPage():
    print "Arrived in index - upload"
    respons = {'status':'','msg':''}
    if 'userId' in session:
        print session['userId']
        return render_template("upload.html")
    else:
        print 'user is not logged in'
        return redirect(url_for('loginPage'))
        response['status'] = '400'

@app.route("/logout",methods=["GET"])
def logout():
    print "Arrived in index - Logout"
    response = {'status':'','msg':{}}
    if 'userId' in session:
        print session['userId']
         # remove the userId from the session if it is there
        print "clearing sesssion"
        session.clear()
        print "session cleared"
        response['status'] = "200"
        response['msg'] ='user is not logged in error'
        return jsonify(response)
    else:
        response['status'] = 404;
        response['msg'] ='user is not logged in error'
        return jsonify(response)

@app.route("/login",methods=["POST"])
def index():
    studentId = request.get_json()
    print studentId
    result = db.photorecog.find({"_id" : studentId});
    response = {'status' : '','msg' : {}}

    if result.count() == 0:
        print "NO record found"
        response['status'] = "404"
        response['msg'] = "student Id not found"
        print response
    else:
        print "found user...."
        response['status'] = "200"
        response['msg'] = result.next()
        session['userId']= studentId
        print studentId;
        print session['userId']
        print response
    return jsonify(response)

@app.route("/signUp",methods=["POST"])
def signUp():
    message = request.data
    dataDict = json.loads(message)
    print dataDict['studentId']
    print dataDict['firstName']
    print dataDict['lastName']
    studentId = dataDict['studentId']
    result = db.photorecog.find({"_id" : studentId});
    response = {'status' : '','msg' : {}}


    if result.count() != 0:
        print "User Already exists"
        response['status'] = "400"
        response['msg'] = "User Already exists"
        print response
    else:
        result = db.photorecog.insert_one({
                "_id" : dataDict['studentId'],
                "firstName" : dataDict['firstName'],
                "lastName" : dataDict['lastName'],
                "path" : {
                "originalImage" : "",
                "tobeComparedImage" : ""
                }
        })
        print result
        response['status'] = "200"
        response['msg'] = "successfully registered"
        session['userId']= dataDict['studentId']
        print session['userId']
    return jsonify(response)



@app.route("/uploadfile/<studentId>", methods=['POST'])
def upload(studentId):
    print "in Upload file"
    print request.files;
    target = os.path.join(APP_ROOT, 'static')

    if not os.path.isdir(target):
        os.mkdir(target)
    target = os.path.join(APP_ROOT, 'static/images')

    if not os.path.isdir(target):
        os.mkdir(target)
    target = os.path.join(APP_ROOT, 'static/images/' + studentId)
    print(target)

    if not os.path.isdir(target):
        os.mkdir(target)

    for file in request.files.getlist("file"):
        # print "filename : "
        print(file)
        filename =  "subject01"
        # print type(filename)
        destination = "/".join([target, filename])
        file.save(destination)
        # print destination
        destination = unicodedata.normalize('NFKD', destination).encode('ascii','ignore')
        result = db.photorecog.update_one({"_id" : studentId},{"$set" : {"path.originalImage" : destination}})
        # print result
    return render_template("complete.html")

@app.route("/index1/<studentId>")
def index1(studentId):
    return render_template("upload.html",studentId=studentId)


# The file location will stored across the Student Id, so while fetching the file from database, he needs to query wrt to StudentId
# and get the image and compare with the new image.
#
#

@app.route("/compare/<studentId>", methods=['POST'])
def compare(studentId):
    print "in Compare file"
    print request.files;
    target = os.path.join(APP_ROOT, 'static')

    if not os.path.isdir(target):
        os.mkdir(target)
    target = os.path.join(APP_ROOT, 'static/images')

    if not os.path.isdir(target):
        os.mkdir(target)
    target = os.path.join(APP_ROOT, 'static/images/' + studentId)
    print(target)

    if not os.path.isdir(target):
        os.mkdir(target)

    for file in request.files.getlist("file"):
        # print "filename : "
        print(file)
        filename =  "subject02"
        # print type(filename)
        destination = "/".join([target, filename])
        file.save(destination)
        # print destination
        destination = unicodedata.normalize('NFKD', destination).encode('ascii','ignore')
        result = db.photorecog.update_one({"_id" : studentId},{"$set" : {"path.tobeComparedImage" : destination}})
        # print result
    result = db.photorecog.find({"_id" : studentId},{"path" : "1"});

    print "result is : "
    for r in result:
        originalImagePath = r["path"]["originalImage"]
        tobeComparedImagePath = r["path"]["tobeComparedImage"]

    confidences = predict_confidence(originalImagePath,tobeComparedImagePath,studentId)

    originalImagePathToSend = 'http://ec2-34-208-241-189.us-west-2.compute.amazonaws.com:5000/static' + originalImagePath.split('static')[1]
    tobeComparedImagePathToSend = 'http://ec2-34-208-241-189.us-west-2.compute.amazonaws.com:5000/static' + tobeComparedImagePath.split('static')[1]

    if not confidences:
        print "len is zero"
        response = {'status' : '','msg' : {}}
        response['status'] = "400"
        return jsonify(response)
    else:
        confidenceToSend = round(min(confidences)/150,2)
        print "Done"
        print "found confidence " + str(confidenceToSend)
        msg = {'originalImagePathToSend' : '','tobeComparedImagePathToSend' : '','confidenceToSend' : ''}
        msg['originalImagePathToSend'] = originalImagePathToSend
        msg['tobeComparedImagePathToSend'] = tobeComparedImagePathToSend
        msg['confidenceToSend'] = confidenceToSend
        response = {'status' : '','msg' : {}}
        response['status'] = "200"
        response['msg']= msg
        return jsonify(response)

if __name__ == "__main__":
    # print "in main : creating connection"
    # curr = collection.find();
    # for document in curr:
    #     print(document);

    app.run(port=5000, debug=True,host='0.0.0.0')