from flask import Flask, request, render_template
import numpy as np
import tensorflow
from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing.image import img_to_array
import os
import tensorflow as tf
from PIL import Image
import cv2
from transformers import AutoModel
from huggingface_hub import hf_hub_download


# Loading trained model

os.environ["KERAS_BACKEND"] = "tensorflow"
import keras
model_path = hf_hub_download(repo_id="avimittal30/emotion_detector", filename="ed_model1.keras")
model = keras.models.load_model(model_path)

# model=load_model('my_model.keras')

app = Flask(__name__)


# Home route to render the upload form
@app.route('/')
def index():
    return render_template('index.html')
    

# Prediction route
@app.route('/predict', methods=['POST'])
def predict():
    
    if 'image' not in request.files:
        return render_template('index.html', error='No image uploaded!')

    file = request.files['image']
    
    
    filepath = os.path.join('static', file.filename)
    file.save(filepath)
    print(f'filepath:{filepath}')
    print(f'file:{file}')
    # Process the image to be fed to the model for prediction
    image = cv2.imread(filepath)
    test_image = cv2.resize(image, (256 ,256))
    im=tf.constant(test_image, dtype=tf.float32 )  # Resizing the image to make it compatible with model
    im=tf.expand_dims(im, axis=0)
    # Predict emotion
    predictions = model.predict(im)
    emotion_labels = ['Angry', 'Happy', 'Sad']  # Emotion labels
    predicted_emotion = emotion_labels[np.argmax(predictions)]

    return render_template('result.html', emotion=predicted_emotion, image_file=filepath)

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)