import os
from app import animate_images, AudioInput

def get_test_image_paths():
    gallery_dir = "temp/images"
    return [os.path.join(gallery_dir, img) for img in os.listdir(gallery_dir)][:2]
    
def get_test_audio_paths():
    audio_path = "temp/audio.mp3"
    audio_input = AudioInput(audio_path, 10, 10)
    return audio_input


image_paths = get_test_image_paths()
audio_input = get_test_audio_paths()

animate_images(image_paths, audio_input)