src/app/utils/preprocess.py

import os
import cv2
import numpy as np

def load_images_from_folder(folder_path, img_size):
    emotions = sorted(os.listdir(folder_path))
    x_data = []
    y_data = []
    for idx, emotion in enumerate(emotions):
        emotion_folder = os.path.join(folder_path, emotion)
        for filename in os.listdir(emotion_folder):
            img_path = os.path.join(emotion_folder, filename)
            img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
            if img is not None:
                img = cv2.resize(img, img_size)
                x_data.append(img)
                y_data.append(idx)
    x_data = np.array(x_data)
    y_data = np.array(y_data)
    x_data = x_data.reshape(-1, img_size[0], img_size[1], 1)
    return x_data, y_data, emotions