views.py

import datetime
import json
import sys
import time
from collections import OrderedDict

from django.http import HttpResponse
from django.shortcuts import render
from django.views.decorators.csrf import csrf_exempt

sys.path.append('../')
from cv.controllers.food_controller import upload_and_rec_food
from cv.controllers.nswf_controller import upload_and_rec_porn
from cv.controllers.plant_controller import upload_and_rec_plant
from cv.controllers.plant_disease_controller import upload_and_rec_plant_disease
from cv.controllers.fbp_controller import upload_and_rec_beauty
from cv.controllers.skin_disease_controller import upload_and_rec_skin_disease
from cv.controllers.cbir_controller import upload_and_search
# from cv.controllers.deblur_controller import upload_and_deblur
from cv.cfg import cfg
from utils import db_utils


# Create your views here.


def welcome(request):
    """
    welcome page for computer vision welcome
    :param request:
    :return:
    """
    return render(request, 'welcome.html')


def index(request):
    return render(request, 'index.html')


def mcloud(request):
    return render(request, 'mcloud.html')


def fbp_view(request):
    return render(request, 'fbp.html')


@csrf_exempt
def fbp(request):
    return upload_and_rec_beauty(request)


def food_view(request):
    return render(request, 'food.html')


@csrf_exempt
def food(request):
    return upload_and_rec_food(request)


def plant_view(request):
    return render(request, 'plant.html')


@csrf_exempt
def plant(request):
    return upload_and_rec_plant(request)


def skin_view(request):
    return render(request, 'skin.html')


def nsfw_view(request):
    return render(request, 'nsfw.html')


@csrf_exempt
def nsfw(request):
    return upload_and_rec_porn(request)


def pdr_view(request):
    return render(request, 'pdr.html')


@csrf_exempt
def pdr(request):
    return upload_and_rec_plant_disease(request)


"""
def face_search_view(request):
    return render(request, 'facesearch.html')


@csrf_exempt
def face_search(request):
    return upload_and_search_face(request)
"""

'''
@csrf_exempt
def detect_face(request):
    """
    face detection
    @Note: currently supported by MTCNN
    :param request:
    :return:
    """
    face_img_path = request.GET.get('faceImagePath')
    result = OrderedDict()

    if not face_img_path or face_img_path.strip() == "":
        result['code'] = 1
        result['msg'] = 'Invalid Path for Face Image'
        result['data'] = None
    else:
        img = cv2.imread(face_img_path)
        detector = MTCNN()
        result['code'] = 0
        result['msg'] = 'success'
        result['data'] = detector.detect_faces(img)

    json_result = json.dumps(result, ensure_ascii=False)

    return HttpResponse(json_result)
'''


@csrf_exempt
def rec_skin(request):
    """
    recognize 198 skin disease
    :param request:
    :return:
    """
    skin_disease_result = upload_and_rec_skin_disease(request)
    skin_disease_result_json = json.loads(skin_disease_result.content.decode('utf-8'))

    print(skin_disease_result_json)

    if cfg['use_mysql'] and skin_disease_result_json['code'] == 0:
        conn = db_utils.connect_mysql_db()
        db_utils.insert_to_api(conn, 'LucasXU', 'cv/mcloud/skin', skin_disease_result_json['elapse'],
                               datetime.time(), 0, skin_disease_result_json['imgpath'],
                               skin_disease_result_json['results'][0]['disease'])

    return skin_disease_result


@csrf_exempt
def stat_skin(request):
    """
    skin API statistics
    :param request:
    :return:
    """
    username = request.GET.get('username')
    result = OrderedDict()
    tik = time.time()

    conn = db_utils.connect_mysql_db()

    result['code'] = 0
    result['msg'] = 'success'
    result['api'] = db_utils.query_api_hist(conn, username)
    tok = time.time()
    result['epalse'] = tok - tik

    json_result = json.dumps(result, ensure_ascii=False)

    return HttpResponse(json_result)


def cbir_view(request):
    return render(request, 'cbir.html')


@csrf_exempt
def cbir(request):
    return upload_and_search(request)


# def deblur_view(request):
#     return render(request, 'deblur.html')
#
#
# @csrf_exempt
# def deblur(request):
#     return upload_and_deblur(request)