fix celltype heatmap gen

app.py

@@ -41,9 +41,9 @@ graph_file = hf_hub_download(repo_id="dylanplummer/islet-epigenome", filename=f"
 rna_file = unzip(rna_file)
 hic_file = unzip(hic_file)
 
-celltypes = ['Alpha', 'Beta', 'Delta', 'PP', 'Acinar', 'Duct', 'PSC', 'Endothelial']
+cooler_celltypes = ['Alpha', 'Beta', 'Delta', 'PP', 'Acinar', 'Duct', 'PSC', 'Endothelial']
 coolers = []
-for celltype in celltypes:
+for celltype in cooler_celltypes:
     cooler_file = hf_hub_download(repo_id="dylanplummer/islet-epigenome", filename=f"pseudobulk_coolers/{celltype}_50kb.cool", repo_type="dataset", token=os.environ['DATASET_SECRET'])
     coolers.append(cooler.Cooler(cooler_file))
 beta_cooler = coolers[1]
@@ -206,7 +206,7 @@ def get_heatmap(celltype, gene, locus1, locus2, heatmap_range):
 
 def get_heatmaps(gene, locus1, locus2, heatmap_range):
     res = []
-    for celltype in celltypes:
+    for celltype in cooler_celltypes:
         res.append(get_heatmap(celltype, gene, locus1, locus2, heatmap_range))
     alpha, beta, delta, pp, acinar, duct, psc, endothelial = res
     return alpha, beta, delta, pp, acinar, duct, psc, endothelial
@@ -341,7 +341,7 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
             heatmap_size = gr.Slider(label="Heatmap Range", info="kb range aroung input gene locus to expand", minimum=50, maximum=30000, value=2000)
     with gr.Row():
         out_heatmaps = []
-        for celltype, c in zip(celltypes, coolers):
+        for celltype, c in zip(cooler_celltypes, coolers):
             out_heatmaps.append(gr.Image(label=celltype, elem_id=f'out-heatmap-{celltype}', scale=1))
     with gr.Row():
         heatmap_button = gr.Button(value="Generate Heatmaps", elem_id='heatmap-button', scale=1)