from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import os

def overlay_file(im_path, heat_map_path):
    im = np.array(Image.open(im_path))
    heat_map = np.array(Image.open(heat_map_path))

    overlay_numpy(im, heat_map, im_path)

def overlay_numpy(im, heat_map, path):
    fig = plt.figure()
    plt.imshow(im, 'gray', interpolation='none')
    plt.imshow(heat_map, 'jet', interpolation='none', alpha=0.35),
    plt.axis('off')
    #plt.show()
    file_name = os.path.splitext(path)[0] + '_overlay.png'
    fig.savefig(file_name, bbox_inches='tight', pad_inches=0, transparent=True, dpi=1200)
    plt.close()