import os
from PIL import Image
from transformers import CLIPProcessor
import os

def get_image_list():
    # Get list of all files in images directory
    image_files = os.listdir('images/')
    
    # Filter for .jpg files only
    jpg_files = [f"images/{f}" for f in image_files if f.endswith('.jpg')]
    
    return jpg_files


def get_image_calibration_data():
    processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
    
    calibration_data = []
    for image_file in get_image_list():
        image_path = os.path.join(image_file)
        image = Image.open(image_path)
        inputs = processor(images=image, return_tensors="pt", padding="max_length", truncation=True)
        calibration_data.append(inputs.data["pixel_values"])
    
    return calibration_data