Fixed empty image error

app.py

@@ -20,7 +20,7 @@ service_account_json = {
     "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/veytel-pulmoviz-pentool%40veytel-cloud-store.iam.gserviceaccount.com",
     "universe_domain": "googleapis.com"
 }
-
+count = 0
 
 def set_min_dense_1(max_dense_0):
     global scaled_thresh1
@@ -78,7 +78,8 @@ def upload_csv_to_gcs(csv_content, filename):
 
 def new__cxr(max_dense_0, max_dense_1, max_dense_2):
     global textured_cxr, lung_noised, max_val, cxr, mask, scaled_thresh1, scaled_thresh2, scaled_thresh3, image_id, index, count
-    index += 1
+    if count >= 0:
+        index += 1
     if index > 3:
         index = 1
         image_id += 1
@@ -87,20 +88,19 @@ def new__cxr(max_dense_0, max_dense_1, max_dense_2):
 
     csv_data = read_csv_from_gcs(user)
     # write count, thresh1, thresh2, thresh3 to csv file
-    if count <= 30:
+    if count < 30:
         csv_content = ""
         for row in csv_data:
             csv_content += ",".join(row) + "\n"
-        csv_content += f"{count},{max_dense_0},{max_dense_1},{max_dense_2}\n"
+        if count >0:
+            csv_content += f"{count},{max_dense_0},{max_dense_1},{max_dense_2}\n"
         filename = f"density_{user}.csv"
         upload_csv_to_gcs(csv_content, filename)
 
-
-
-    if (count > 30):
+    if (count >= 30):
         empty_image = np.zeros((256, 256), dtype=np.uint8)
         label1.update(visible=False)
-        count = 1
+        count = -1
         image_id = 1
         index = 1
         csv_content = ""
@@ -108,10 +108,10 @@ def new__cxr(max_dense_0, max_dense_1, max_dense_2):
             csv_content += ",".join(row) + "\n"
         fieldnames = ['count', 'thresh_1', 'thresh_2', 'thresh_3']
         csv_content += ",".join(fieldnames) + "\n"
-        csv_content += f"{count},{max_dense_0},{max_dense_1},{max_dense_2}\n"
+        #csv_content += f"{count},{max_dense_0},{max_dense_1},{max_dense_2}\n"
         filename = f"density_{user}.csv"
         upload_csv_to_gcs(csv_content, filename)
-        return empty_image, empty_image, empty_image, empty_image, empty_image, empty_image, empty_image, 30
+        return empty_image, empty_image, empty_image, empty_image, empty_image, empty_image, empty_image, empty_image, 0
     cxr_file = "cxr" + str(image_id) + "_cxr.png"
     mask_file = "cxr" + str(image_id) + "_mask.png"
     textured_cxr_file = "cxr" + str(image_id) + "_textured_" + str(index) + ".png"
@@ -135,7 +135,7 @@ def new__cxr(max_dense_0, max_dense_1, max_dense_2):
     dense_1 = np.where(((textured_cxr < scaled_thresh2) & (textured_cxr >= scaled_thresh1)), textured_cxr, 0)
     dense_2 = np.where(((textured_cxr < scaled_thresh3) & (textured_cxr >= scaled_thresh2)), textured_cxr, 0)
     dense_3 = np.where(((textured_cxr < 255) & (textured_cxr >= scaled_thresh3)), textured_cxr, 0)
-    return cxr, textured_cxr, lung_noised, lung_noised, dense_0, dense_1, dense_2, dense_3, count
+    return cxr, textured_cxr, lung_noised, lung_noised, dense_0, dense_1, dense_2, dense_3, count+1
 
 
 def create_csv():
@@ -148,8 +148,7 @@ def create_csv():
     else:
         for row in csv_data:
             csv_content += ",".join(row) + "\n"
-    csv_content += f"{count},{thresh1},{thresh2},{thresh3}\n"
-
+    #csv_content += f"{count+1},{thresh1},{thresh2},{thresh3}\n"
     filename = f"density_{user}.csv"
     upload_csv_to_gcs(csv_content, filename)
 
@@ -174,7 +173,6 @@ def change_vis():
 
 
 with gr.Blocks() as demo:
-    count = 1
     image_id = 1
     index = 1
     executable_path = os.path.dirname(os.path.realpath(__file__))