import video_handler as vh
import frame_handler_yolo as fh

model_path = "yolov8n.pt"  # YOLOv8 model path
original_fps = 30  # Original FPS of the input videos

video_path = "../input_data/football.mp4"  # Replace with your video path

output_folder = "../output_data/"  # Folder to save extracted frames
all_frames_folder = output_folder + "all_frames"

output_video_path_9_16 = output_folder + "output_video_9_16_test.mp4"  # Output video path
target_frame_rate = 30  # Frames per second of the output videos
target_resolution = (360, 640)  # Output resolution (9:16)

# FOR TESTING ONLY
test_key_frames_folder = output_folder + "test_key_frames"
test_key_frames_9_16_folder = output_folder + "test_key_frames_9_16"
#fh.crop_preserve_key_objects(test_key_frames_folder, test_key_frames_9_16_folder, model_path, target_resolution)
vh.create_video_from_frames(test_key_frames_9_16_folder, output_video_path_9_16, target_frame_rate, target_resolution)