import gradio as gr
import os
import keras
from keras.preprocessing.image import img_to_array
import imutils
import cv2
from keras.models import load_model
import numpy as np

# parameters for loading data and images
detection_model_path = 'haarcascade_files/haarcascade_frontalface_default.xml'
# emotion_model_path = 'model2/model2_entire_model.h5'
emotion_model_path = 'model_2_aug_nocall_BEST/model_2_aug_nocall_entire_model.h5'



# hyper-parameters for bounding boxes shape
# loading models
face_detection = cv2.CascadeClassifier(detection_model_path)
emotion_classifier = load_model(emotion_model_path, compile=False)
EMOTIONS = ['neutral','happiness','surprise','sadness','anger','disgust','fear','contempt','unknown']


def predict(frame):

    frame = imutils.resize(frame, width=300)
    gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
    faces = face_detection.detectMultiScale(gray, scaleFactor=1.1,
                                            minNeighbors=5, minSize=(30, 30),
                                            flags=cv2.CASCADE_SCALE_IMAGE)

    frameClone = frame.copy()
    if len(faces) > 0:
        faces = sorted(faces, reverse=True,
                       key=lambda x: (x[2] - x[0]) * (x[3] - x[1]))[0]
        (fX, fY, fW, fH) = faces
        # Extract the ROI of the face from the grayscale image, resize it to a fixed 28x28 pixels, and then prepare
        # the ROI for classification via the CNN
        roi = gray[fY:fY + fH, fX:fX + fW]
        roi = cv2.resize(roi, (48, 48))
        roi = roi.astype("float") / 255.0
        roi = img_to_array(roi)
        roi = np.expand_dims(roi, axis=0)

        preds = emotion_classifier.predict(roi)[0]
        label = EMOTIONS[preds.argmax()]
    else:
        return frameClone, "Can't find your face"

    probs = {}
    cv2.putText(frameClone, label, (fX, fY - 10),
                cv2.FONT_HERSHEY_DUPLEX, 1, (238, 164, 64), 1)
    cv2.rectangle(frameClone, (fX, fY), (fX + fW, fY + fH),
                  (238, 164, 64), 2)

    for (i, (emotion, prob)) in enumerate(zip(EMOTIONS, preds)):
        probs[emotion] = float(prob)

    return frameClone, probs

# Define Gradio input and output components
image_input = gr.components.Image(type='numpy', label="Upload Image or Video")
image_output = gr.components.Image(label="Predicted Emotion")
label_output = gr.components.Label(num_top_classes=2, label="Top 2 Probabilities")

inp = [
        gr.components.Image(sources="webcam", label="Your face"),
        gr.components.File(label="Upload Image or Video")
      ]
out = [
        gr.components.Image(label="Predicted Emotion"),
        gr.components.Label(num_top_classes=2, label="Top 2 Probabilities")
        ]

example_images = [
    [
        os.path.join(os.path.dirname(__file__), "images/chandler.jpeg"),
        os.path.join(os.path.dirname(__file__), "images/janice.jpg"),
        os.path.join(os.path.dirname(__file__), "images/joey.jpg"),
        os.path.join(os.path.dirname(__file__), "images/phoebe.jpg"),
        os.path.join(os.path.dirname(__file__), "images/rachel_monica.jpg"),
        os.path.join(os.path.dirname(__file__), "images/ross.jpg"),
        os.path.join(os.path.dirname(__file__), "images/gunther.jpg")
    ]
]
# example_images = [
#     ["images/chandler.jpeg"],
#     ["images/janice.jpg"],
#     ["images/joey.jpg"],
#     ["images/phoebe.jpg"],
#     ["images/rachel_monica.jpg"],
#     ["images/ross.jpg"],
#     ["images/gunther.jpg"]
# ]

title = "Facial Emotion Recognition"
description = "How well can this model predict your emotions? Take a picture with your webcam, and it will guess if" \
              " you are: happy, sad, angry, disgusted, scared, surprised, or neutral."
thumbnail = "https://raw.githubusercontent.com/gradio-app/hub-emotion-recognition/master/thumbnail.png"

# gr.Interface(predict, inp, out, capture_session=True, title=title, thumbnail=thumbnail,
#              description=description).launch(inbrowser=True)
gr.Interface(fn=predict, inputs=inp, outputs=out,
            examples=example_images,title=title, thumbnail=thumbnail).launch()



# # Launch Gradio interface
# gr.Interface(fn=predict, inputs=image_input, outputs=[image_output, label_output],
#              title="Facial Emotion Recognition", description="How well can this model predict your emotions?").launch()