filter_content.py

import os
from PIL import Image
import cv2

def filter_images(images_directory):
    if not os.path.exists(images_directory):
        os.makedirs(images_directory)
        
    all_files_images = os.listdir(images_directory)
    # Filter image files
    IMAGE_FILES = []
    for file in all_files_images:
        file_path = os.path.join(images_directory, file)
        if file.lower().endswith(('.jpg', '.jpeg', '.png', '.gif', '.webp')):
            # Open image and get dimensions
            try:
                with Image.open(file_path) as img:
                    width, height = img.size
                    if width == 1024 and height == 576:
                        IMAGE_FILES.append(file)
            except Exception as e:
                print(f"Error processing image '{file}': {e}")
    return IMAGE_FILES

def filter_images_short(images_short_directory):
    if not os.path.exists(images_short_directory):
        os.makedirs(images_short_directory)
        
    all_files_images = os.listdir(images_short_directory)
    # Filter image files
    IMAGE_FILES = []
    for file in all_files_images:
        file_path = os.path.join(images_short_directory, file)
        if file.lower().endswith(('.jpg', '.jpeg', '.png', '.gif', '.webp')):
            # Open image and get dimensions
            try:
                with Image.open(file_path) as img:
                    width, height = img.size
                    if width == 536 and height == 960:
                        IMAGE_FILES.append(file)
            except Exception as e:
                print(f"Error processing image '{file}': {e}")
    return IMAGE_FILES

def filter_videos(videos_directory):
    if not os.path.exists(videos_directory):
        os.makedirs(videos_directory)
        
    all_files_video = os.listdir(videos_directory)    
    # Filter video files
    VIDEO_FILES = []
    for file in all_files_video:
        file_path = os.path.join(videos_directory, file)
        if file.lower().endswith(('.mp4', '.avi', '.mov', '.mkv')):
            # Read video properties
            cap = cv2.VideoCapture(file_path)
            width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
            height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
            cap.release()
            
            # Check if video resolution matches 1280x720
            if width == 1280 and height == 720:
                VIDEO_FILES.append(file)
    return VIDEO_FILES