Purpose of the notebook is to generate reports via saved images

import os
import sys
sys.path.append("/oak/stanford/groups/akundaje/manyu/C2H2_ZNF_project/utils/modisco_utils")
sys.path.append(os.path.abspath("/oak/stanford/groups/akundaje/manyu/softwares/tfmodisco_tf_models/src/"))
sys.path.append(os.path.abspath("/oak/stanford/groups/akundaje/manyu/softwares/tfmodisco_tf_models/notebooks/reports/"))
from tfmodisco.run_tfmodisco import import_shap_scores, import_tfmodisco_results
from motif.read_motifs import pfm_info_content, trim_motif_by_ic
import motif.moods as moods
import plot.viz_sequence as viz_sequence
from util import figure_to_vdom_image, import_peak_table
import h5py
import pandas as pd
import numpy as np
import modisco
import sklearn.decomposition
import umap
import matplotlib.pyplot as plt
import matplotlib.font_manager as font_manager
import vdom.helpers as vdomh
from IPython.display import display
import tqdm
import io
import base64
import urllib
import deepdish as dd
import shutil
from vdom.helpers import h1, p, img, div, b

Main class to generate reports in HTML form using the saved images and h5 files

from shutil import copyfile
def get_read_handle(h5_file):
    '''Copy to temp with a hashed name and give the read handle'''
    name = h5_file.split('/')[-1]
    #copy_file = '/users/manyu/temp/{}_{}'.format(str(hash(h5_file)),name)
    copy_file = '/srv/scratch/manyu/tmp/{}_{}'.format(str(hash(h5_file)),name)
    
    if os.path.exists(copy_file):
        os.remove(copy_file)
    copyfile(h5_file,copy_file)
    f = h5py.File(copy_file,'r')
    return(f)


def close_h5file_read_handle(h5_file,read_handle):
    '''Removes the copied h5 file in temp and the read handle to avoid bloating'''
    read_handle.close()
    name = h5_file.split('/')[-1]
    #copy_file = '/users/manyu/temp/{}_{}'.format(str(hash(h5_file)),name)
    copy_file = '/srv/scratch/manyu/tmp/{}_{}'.format(str(hash(h5_file)),name)
    if os.path.exists(copy_file):
        os.remove(copy_file)
    
    
    
  

    

class ReportGenerator():
    def __init__(self,tf,dataset):
        self.dataset = dataset
        self.tf = tf
        # Path to saved modisco results on /oak. Images and h5 files
        self.reports_base = '/oak/stanford/groups/akundaje/manyu/C2H2_ZNF_project/train_profile_models_2020/modisco_reports'
        self.results_dir = '{}/{}/{}'.format(self.reports_base,dataset,tf)
        
        ## Paths to the saved modisco report results to copy/copied on Mitra: Images and h5 files
        self.modisco_reports_results_mitra = '/srv/www/kundaje/manyu/C2H2_ZNF_project/modisco_report_saved_information'
        self.mitra_results_dataset_dir = '{}/{}'.format(self.modisco_reports_results_mitra,self.dataset)
        if not os.path.exists(self.mitra_results_dataset_dir):
            os.makedirs(self.mitra_results_dataset_dir)
        self.save_results_mitra_dir  = '{}/{}'.format(self.mitra_results_dataset_dir,self.tf)
        
        self.main_patterns_dir = '{}/main_patterns'.format(self.save_results_mitra_dir)
        self.main_patterns = '{}/all_motifs.h5'.format(self.main_patterns_dir)
        
        self.subpatterns_dir  = '{}/sub_patterns'.format(self.save_results_mitra_dir)
        self.subpatterns = '{}/all_motif_subclusters.h5'.format(self.subpatterns_dir)
        
        
        ## Paths to the modisco reports html on lab cluster
        self.modisco_reports_base = '/srv/www/kundaje/manyu/C2H2_ZNF_project/modisco_reports'
        self.modisco_reports_dataset_base = '{}/{}'.format(self.modisco_reports_base,self.dataset)
        if not os.path.exists(self.modisco_reports_dataset_base):
            os.makedirs(self.modisco_reports_dataset_base)
        self.modisco_report_filename = '{}/{}.html'.format(self.modisco_reports_dataset_base,self.tf)
        
        ## Path to webpages and weblinks to images(needs weblinks to load; not local links)
        self.mitra_web_base = 'http://mitra.stanford.edu/kundaje/manyu/C2H2_ZNF_project/'
        self.mitra_web_results_base = '{}/modisco_report_saved_information/{}/{}'.format(self.mitra_web_base,self.dataset,self.tf)
        self.mitra_web_results_main_patterns = '{}/main_patterns'.format(self.mitra_web_results_base)
        self.mitra_web_results_sub_patterns = '{}/sub_patterns'.format(self.mitra_web_results_base)
        self.mitra_web_results_profiles = '{}/profiles_dir'.format(self.mitra_web_results_base)
        self.mitra_web_results_b1h_aln = '{}/b1h_aligments'.format(self.mitra_web_results_base)
        
        ##Do we need this here?
        #self.mitra_web_reports_base = '{}/modisco_reports'.format(self.mitra_web_base)
        
        
    
    def copy_to_mitra(self):
        assert(os.path.exists(self.results_dir))
        if os.path.exists(self.save_results_mitra_dir):
            shutil.rmtree(self.save_results_mitra_dir)
        shutil.copytree(self.results_dir,self.save_results_mitra_dir)
        print('Finished copying files to Mitra')
        permissions_cmd = 'chmod -R 777 {}'.format(self.save_results_mitra_dir)
        os.system(permissions_cmd)
        
        
        
    def show_subcluster_table(self):
      
        assert(os.path.exists(self.main_patterns))
        
        ## Opening a read handle for h5. Remmeber to close it also
        try:
            patterns_read_handle = get_read_handle(self.main_patterns)
        except Exception as e:
            print(e)
        
        try:
            sub_patterns_read_handle = get_read_handle(self.subpatterns)
        except Exception as e:
            print(e)

        num_patterns  = len(patterns_read_handle.keys())

     
        
        for pattern_i in range(num_patterns):
            
            display(vdomh.h3("Pattern {}/{}".format(str(pattern_i + 1), num_patterns)))
        
            

            # Sort this out:
            colgroup = vdomh.colgroup(
                vdomh.col(style={"width": "5%"}),
                vdomh.col(style={"width": "5%"}),
                vdomh.col(style={"width": "50%"}),
                vdomh.col(style={"width": "40%"})
            )
            header = vdomh.thead(
                vdomh.tr(
                    vdomh.th("Subpattern", style={"text-align": "center"}),
                    vdomh.th("Seqlets", style={"text-align": "center"}),
                    vdomh.th("Embeddings", style={"text-align": "center"}),
                    vdomh.th("CWM", style={"text-align": "center"}),
                    vdomh.th("PFM", style={"text-align": "center"})
                )
            )
            
            
            ## Fill in the paths and seqlets of the main pattern
            total_seqlets = patterns_read_handle[str(pattern_i)]['n_seqlets'].value
            emb_agg_path = '{}/{}_subcluster_agg.png'.format(self.mitra_web_results_sub_patterns,str(pattern_i))
            pwm_path = '{}/{}_pfm_full.png'.format(self.mitra_web_results_main_patterns,str(pattern_i))
            cwm_path = '{}/{}_cwm_full.png'.format(self.mitra_web_results_main_patterns,str(pattern_i))
            
            
            table_rows = [vdomh.tr(
                vdomh.td("Agg."),
                vdomh.td(str(total_seqlets)),
                vdomh.td(img(src=emb_agg_path)),
                vdomh.td(img(src=cwm_path)),
                vdomh.td(img(src=pwm_path))
            )]

            
            total_sub_pats = len(sub_patterns_read_handle[str(pattern_i)].keys())
            for subpat_i in range(total_sub_pats):
                emb_path = '{}/{}_subcluster_{}.png'.format(self.mitra_web_results_sub_patterns,str(pattern_i),str(subpat_i))
            
                total_seqlets = sub_patterns_read_handle[str(pattern_i)]['subcluster_{}'.format(str(subpat_i))]['n_seqlets'].value
                pwm_path = '{}/{}_subcluster_{}_pfm_trimmed.png'.format(self.mitra_web_results_sub_patterns,str(pattern_i),str(subpat_i))
                cwm_path = '{}/{}_subcluster_{}_cwm_trimmed.png'.format(self.mitra_web_results_sub_patterns,str(pattern_i),str(subpat_i))
                table_rows.append(vdomh.tr(
                vdomh.td('{}'.format(str(subpat_i))),
                vdomh.td(str(total_seqlets)),
                vdomh.td(img(src=emb_path)),
                vdomh.td(img(src=cwm_path)),
                vdomh.td(img(src=pwm_path))
            ))
            table = vdomh.table(header,vdomh.tbody(*table_rows))
            
            display(table)
            
        #Remove unnecessary temp files
        close_h5file_read_handle(self.main_patterns,patterns_read_handle)
        close_h5file_read_handle(self.subpatterns,sub_patterns_read_handle)
    
    
    def show_footprints_and_peak_dist(self):
        try:
            patterns_read_handle = get_read_handle(self.main_patterns)
        except Exception as e:
            print(e)
        
        num_patterns  = len(patterns_read_handle.keys())
        for pattern_i in range(num_patterns):
            
            display(vdomh.h3("Pattern {}/{} Footprint".format(str(pattern_i + 1), num_patterns)))
            footprint_plot = '{}/profiles_pattern_{}.png'.format(self.mitra_web_results_profiles,str(pattern_i))
            motif_distplot  = '{}/motif_distances_pattern_{}.png'.format(self.mitra_web_results_profiles,str(pattern_i))
            display(img(src=footprint_plot))
            display(vdomh.h3("Pattern {}/{} Motif Distance Distribution".format(str(pattern_i + 1), num_patterns)))
            display(img(src=motif_distplot))
    
    
    def show_b1h_alignments(self):
        try:
            patterns_read_handle = get_read_handle(self.main_patterns)
        except Exception as e:
            print(e)
        


        num_patterns  = len(patterns_read_handle.keys())
        header = vdomh.thead(
            vdomh.tr(
                vdomh.th("Pattern", style={"text-align": "center"}),
                vdomh.th("N-Seqlets", style={"text-align": "center"}),
                vdomh.th("Alignment", style={"text-align": "center"})

            )
        )

         
        display(vdomh.h2("B1H alignments of all patterns"))
        table_rows = []
        for pattern_i in range(num_patterns):
            
            
        
            total_seqlets = patterns_read_handle[str(pattern_i)]['n_seqlets'].value
            alignment_img = '{}/pattern_{}_aligned.png'.format(self.mitra_web_results_b1h_aln,str(pattern_i))
            

            table_rows.append(vdomh.tr(
            vdomh.td('{}'.format(str(pattern_i+1))),
            vdomh.td('{}'.format(str(total_seqlets))),
            vdomh.td(img(src=alignment_img)),

            ))
        close_h5file_read_handle(self.main_patterns,patterns_read_handle)
        table = vdomh.table(header,vdomh.tbody(*table_rows))
        display(table)

# Define parameters/fetch arguments
dataset = os.environ["DATASET"]
tf = os.environ["TF"]

# tf = 'ENCSR904JEY'
# dataset = 'ENCODE_new'

vdomh.h1('Running {} in dataset: {}'.format(tf,dataset))

Running ENCSR550HCT in dataset: ENCODE_new

a = ReportGenerator(tf=tf,dataset=dataset)

Show motifs and submotifs and relevant embeddings

a.show_subcluster_table()

Pattern 1/31

/users/manyu/.local/lib/python3.6/site-packages/ipykernel_launcher.py:127: H5pyDeprecationWarning: dataset.value has been deprecated. Use dataset[()] instead.
/users/manyu/.local/lib/python3.6/site-packages/ipykernel_launcher.py:146: H5pyDeprecationWarning: dataset.value has been deprecated. Use dataset[()] instead.

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	1214
0	274
1	178
2	163
3	120
4	107
5	93
6	77
7	61
8	51
9	35
10	27
11	20
12	8

Pattern 2/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	1068
0	182
1	163
2	154
3	147
4	135
5	125
6	91
7	42
8	11
9	10
10	6
11	2

Pattern 3/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	1081
0	187
1	136
2	124
3	112
4	110
5	80
6	75
7	70
8	62
9	52
10	39
11	27
12	7

Pattern 4/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	1045
0	200
1	191
2	121
3	108
4	100
5	92
6	68
7	43
8	42
9	33
10	30
11	17

Pattern 5/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	955
0	146
1	115
2	112
3	106
4	92
5	92
6	73
7	56
8	47
9	39
10	39
11	38

Pattern 6/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	529
0	80
1	71
2	50
3	45
4	45
5	41
6	40
7	40
8	35
9	32
10	25
11	25

Pattern 7/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	446
0	57
1	54
2	47
3	42
4	40
5	35
6	35
7	35
8	34
9	33
10	20
11	14

Pattern 8/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	449
0	52
1	52
2	50
3	45
4	44
5	38
6	32
7	29
8	28
9	25
10	24
11	19
12	11

Pattern 9/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	378
0	45
1	43
2	43
3	37
4	35
5	33
6	33
7	32
8	26
9	25
10	20
11	6

Pattern 10/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	334
0	88
1	61
2	53
3	42
4	31
5	30
6	19
7	5
8	5

Pattern 11/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	207
0	31
1	27
2	25
3	21
4	21
5	19
6	18
7	17
8	15
9	11
10	2

Pattern 12/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	173
0	36
1	32
2	30
3	27
4	19
5	9
6	9
7	8
8	3

Pattern 13/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	180
0	36
1	32
2	26
3	25
4	23
5	19
6	17
7	2

Pattern 14/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	160
0	29
1	25
2	22
3	21
4	20
5	17
6	14
7	12

Pattern 15/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	165
0	28
1	27
2	24
3	24
4	22
5	21
6	19

Pattern 16/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	147
0	31
1	31
2	30
3	26
4	15
5	14

Pattern 17/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	123
0	33
1	31
2	25
3	14
4	10
5	5
6	3
7	2

Pattern 18/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	127
0	37
1	23
2	20
3	17
4	11
5	8
6	7
7	4

Pattern 19/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	106
0	25
1	23
2	19
3	18
4	16
5	5

Pattern 20/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	114
0	17
1	17
2	16
3	15
4	15
5	14
6	12
7	6
8	2

Pattern 21/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	100
0	23
1	22
2	22
3	15
4	10
5	5
6	3

Pattern 22/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	84
0	21
1	18
2	10
3	10
4	9
5	8
6	8

Pattern 23/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	57
0	14
1	13
2	11
3	10
4	9

Pattern 24/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	56
0	17
1	12
2	12
3	10
4	5

Pattern 25/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	78
0	20
1	19
2	14
3	13
4	9
5	3

Pattern 26/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	64
0	27
1	24
2	7
3	6

Pattern 27/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	48
0	16
1	12
2	11
3	9

Pattern 28/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	78
0	19
1	18
2	18
3	16
4	5
5	2

Pattern 29/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	57
0	31
1	19
2	7

Pattern 30/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	66
0	15
1	13
2	12
3	11
4	7
5	6
6	2

Pattern 31/31

Subpattern	Seqlets	Embeddings	CWM	PFM
Agg.	46
0	20
1	19
2	7

Show footprints and motif distributions within peaks

a.show_footprints_and_peak_dist()

Pattern 1/31 Footprint

Pattern 1/31 Motif Distance Distribution

Pattern 2/31 Footprint

Pattern 2/31 Motif Distance Distribution

Pattern 3/31 Footprint

Pattern 3/31 Motif Distance Distribution

Pattern 4/31 Footprint

Pattern 4/31 Motif Distance Distribution

Pattern 5/31 Footprint

Pattern 5/31 Motif Distance Distribution

Pattern 6/31 Footprint

Pattern 6/31 Motif Distance Distribution

Pattern 7/31 Footprint

Pattern 7/31 Motif Distance Distribution

Pattern 8/31 Footprint

Pattern 8/31 Motif Distance Distribution

Pattern 9/31 Footprint

Pattern 9/31 Motif Distance Distribution

Pattern 10/31 Footprint

Pattern 10/31 Motif Distance Distribution

Pattern 11/31 Footprint

Pattern 11/31 Motif Distance Distribution

Pattern 12/31 Footprint

Pattern 12/31 Motif Distance Distribution

Pattern 13/31 Footprint

Pattern 13/31 Motif Distance Distribution

Pattern 14/31 Footprint

Pattern 14/31 Motif Distance Distribution

Pattern 15/31 Footprint

Pattern 15/31 Motif Distance Distribution

Pattern 16/31 Footprint

Pattern 16/31 Motif Distance Distribution

Pattern 17/31 Footprint

Pattern 17/31 Motif Distance Distribution

Pattern 18/31 Footprint

Pattern 18/31 Motif Distance Distribution

Pattern 19/31 Footprint

Pattern 19/31 Motif Distance Distribution

Pattern 20/31 Footprint

Pattern 20/31 Motif Distance Distribution

Pattern 21/31 Footprint

Pattern 21/31 Motif Distance Distribution

Pattern 22/31 Footprint

Pattern 22/31 Motif Distance Distribution

Pattern 23/31 Footprint

Pattern 23/31 Motif Distance Distribution

Pattern 24/31 Footprint

Pattern 24/31 Motif Distance Distribution

Pattern 25/31 Footprint

Pattern 25/31 Motif Distance Distribution

Pattern 26/31 Footprint

Pattern 26/31 Motif Distance Distribution

Pattern 27/31 Footprint

Pattern 27/31 Motif Distance Distribution

Pattern 28/31 Footprint

Pattern 28/31 Motif Distance Distribution

Pattern 29/31 Footprint

Pattern 29/31 Motif Distance Distribution

Pattern 30/31 Footprint

Pattern 30/31 Motif Distance Distribution

Pattern 31/31 Footprint

Pattern 31/31 Motif Distance Distribution

Show B1H alignments of main motifs

a.show_b1h_alignments()

B1H alignments of all patterns

/users/manyu/.local/lib/python3.6/site-packages/ipykernel_launcher.py:207: H5pyDeprecationWarning: dataset.value has been deprecated. Use dataset[()] instead.

Pattern	N-Seqlets	Alignment
1	1214
2	1068
3	1081
4	1045
5	955
6	529
7	446
8	449
9	378
10	334
11	207
12	173
13	180
14	160
15	165
16	147
17	123
18	127
19	106
20	114
21	100
22	84
23	57
24	56
25	78
26	64
27	48
28	78
29	57
30	66
31	46