add violin

app.py

@@ -7,6 +7,7 @@ import os
 import gzip
 import shutil
 import numpy as np
+import plotly.plotly as py
 import matplotlib.pyplot as plt
 
 from huggingface_hub import hf_hub_download
@@ -193,7 +194,19 @@ def perturb(gene, locus1, locus2):
     # convert from argb to rgba
     image_from_plot = image_from_plot[:, :, [1, 2, 3, 0]]
 
-    return image_from_plot
+    fig = sc.pl.violin(rna, gene, groupby='celltype', return_fig=True)
+    fig.canvas.draw()
+    violin_img = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
+    violin_img = violin_img.reshape(fig.canvas.get_width_height()[::-1] + (4,))
+    violin_img = violin_img[:, :, [1, 2, 3, 0]]
+
+    fig = sc.pl.violin(rna, 'log2FC', groupby='celltype', return_fig=True)
+    fig.canvas.draw()
+    violin_img_fc = np.frombuffer(fig.canvas.tostring_argb(), dtype=np.uint8)
+    violin_img_fc = violin_img_fc.reshape(fig.canvas.get_width_height()[::-1] + (4,))
+    violin_img_fc = violin_img_fc[:, :, [1, 2, 3, 0]]
+
+    return image_from_plot, violin_img, violin_img_fc
 
 
 
@@ -207,8 +220,14 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
         run_button = gr.Button(label="Run", elem_id='run-button', scale=1)
     with gr.Row():
         out_img = gr.Image(elem_id='out-img', scale=1)
+    with gr.Row():
+        out_violin = gr.Image(elem_id='out-violin', scale=1)
+        out_violin_fc = gr.Image(elem_id='out-violin-fc', scale=1)
         #out_plot = gr.Plot(elem_id='out-plot', scale=1)
 
+    inputs = [in_locus, anchor1, anchor2]
+    outputs = [out_img, out_violin, out_violin_fc]
+
     gr.Examples(examples=[['INS', 'chr11:2,289,895', 'chr11:2,298,840'],
                           ['INS', 'INS', 'IGF1'],
                           ['TSPAN1', 'TSPAN1', 'PIK3R3'],
@@ -216,10 +235,10 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
                           ['KRT19', 'chr17:22,220,637', 'chr17:39,591,813'],
                           ['LPP', 'chr3:188,097,749', 'chr3:197,916,262'],
                           ['CEL', 'chr9:135,937,365', 'chr9:135,973,107']],
-                inputs=[in_locus, anchor1, anchor2],
-                outputs=[out_img],
+                inputs=inputs,
+                outputs=outputs,
                 fn=perturb, cache_examples=os.getenv('SYSTEM') == 'spaces')
-    run_button.click(perturb, [in_locus, anchor1, anchor2], outputs=[out_img])
+    run_button.click(perturb, [in_locus, anchor1, anchor2], outputs=outputs)
 
 
 if __name__ == "__main__":