from typing import Dict
from typing import List


def make_data_lists(img_pth_to_cap: Dict[str, str], image_paths: List[str]):
    """Make lists of data paths and respective captions
    Args:
        img_pth_to_cap: Dictionary of image paths to captions
        image_paths: List of image paths
    Returns:
        img_list: List of image paths
        cap: List of captions
    """
    cap, img_list = [], []
    for im_pth in image_paths:
        caption_list = img_pth_to_cap[im_pth]
        cap.extend(caption_list)
        img_list.extend([im_pth] * len(caption_list))
    return img_list, cap