test recon mean as original violin

app.py

@@ -356,6 +356,7 @@ def perturb(gene, locus1, locus2, use_max_gene):
     else:
         gene_idx = gene_idx_map[gene]
     rna.obs['log2FC'] = ism[:, gene_idx]
+    rna.obs['old_mean'] = rna_recon[:, gene_idx]
     rna.obs['new_mean'] = perterbed_rna_recon[:, gene_idx] 
     
     fig = sc.pl.umap(rna, 
@@ -378,7 +379,8 @@ def perturb(gene, locus1, locus2, use_max_gene):
     image_from_plot = image_from_plot[:, :, [1, 2, 3, 0]]
 
     fig, ax = plt.subplots(figsize=(6, 4))
-    sc.pl.violin(rna, max_gene if use_max_gene else gene, groupby='celltype', layer='counts', show=False, ax=ax, rotation=90)
+    #sc.pl.violin(rna, max_gene if use_max_gene else gene, groupby='celltype', layer='counts', show=False, ax=ax, rotation=90)
+    sc.pl.violin(rna, 'old_mean', groupby='celltype', show=False, ax=ax, rotation=90)
     ax.set_title(f'Normal {max_gene if use_max_gene else gene} expression')
     fig.tight_layout()
     fig.canvas.draw()
@@ -438,7 +440,8 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
     outputs = [out_img, out_violin, out_violin_new, out_violin_fc]
 
     gr.Examples(examples=[['INS', 'chr11:2,289,895', 'chr11:2,298,840', False],
-                          ['INS', 'INS', 'IGF1', False],
+                          ['OTUD3', 'chr1:20457786', 'chr1:20230413', False],
+                          ['IRX2', 'chr5:2704405', 'chr5:2164879', False],
                           ['TSPAN1', 'TSPAN1', 'PIK3R3', False],
                           ['GCG', 'GCG', 'FAP', False],
                           ['KRT19', 'chr17:22,220,637', 'chr17:39,591,813', False],