fix violin fr

app.py

@@ -193,13 +193,15 @@ def perturb(gene, locus1, locus2):
     # convert from argb to rgba
     image_from_plot = image_from_plot[:, :, [1, 2, 3, 0]]
 
-    fig = sc.pl.violin(rna, gene, groupby='celltype', return_fig=True, show=False)
+    fig, ax = plt.subplots(figsize=(6, 6))
+    sc.pl.violin(rna, gene, groupby='celltype', show=False, ax=ax)
     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, show=False)
+    fig, ax = plt.subplots(figsize=(6, 6))
+    sc.pl.violin(rna, 'log2FC', groupby='celltype', show=False, ax=ax)
     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,))