add function to process image data

1 files changed, 9 insertions, 3 deletions

diff --git a/wqflask/wqflask/wgcna/gn3_wgcna.py b/wqflask/wqflask/wgcna/gn3_wgcna.py
index 2657a099..225bef22 100644
--- a/wqflask/wqflask/wgcna/gn3_wgcna.py
+++ b/wqflask/wqflask/wgcna/gn3_wgcna.py
@@ -3,7 +3,6 @@ and process data to be rendered by datatables
 """
 
 
-
 def process_wgcna_data(response):
     """function for processing modeigene genes
     for create row data for datataba"""
@@ -23,5 +22,12 @@ def process_wgcna_data(response):
     }
 
 
-def process_image():
-    pass
\ No newline at end of file
+def process_image(response):
+    """function to process image check if byte string is empty"""
+    image_data = response["output"]["image_data"]
+    return ({
+        "image_generated": True,
+        "image_data": image_data
+    } if image_data else {
+        "image_generated": False
+    })