showing lines

app.py

@@ -19,8 +19,8 @@ linefit = LineFit(30, 0.3)
 
 def get_blob_centroids(mask):
     centers = []
-    print(mask.dtype)
-    print(mask.shape)
+    # print(mask.dtype)
+    # print(mask.shape)
     contours, hierarchies = cv2.findContours(
         mask, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
     for i in contours:
@@ -29,7 +29,7 @@ def get_blob_centroids(mask):
             cx = int(M['m10']/M['m00'])
             cy = int(M['m01']/M['m00'])
             centers.append([cx,cy])
-            print(cx,cy)
+            # print(cx,cy)
     return centers
     
 def predict_mask(input_img, threshold):
@@ -59,7 +59,7 @@ def predict_mask(input_img, threshold):
         d, lines = linefit.predict((p>threshold).astype(np.uint8)*255)
         ds.append(d)
         m = Image.fromarray(p>threshold)
-        imgd.bitmap([0,0], m, fill=random.choice(colors))
+        # imgd.bitmap([0,0], m, fill=random.choice(colors))
         for line in lines:
             imgd.line(line, fill ="blue", width = 1)
         
@@ -68,6 +68,12 @@ def predict_mask(input_img, threshold):
 demo = gr.Blocks()
 
 with demo:
+    with gr.Column():
+        gr.Markdown("""
+# Measure the diameter fibers
+Select fibers that you want to measure. (Only the center of each sketched blob is considered).
+The recommended pixel size for uploaded images is 256x256.
+        """)
     with gr.Row():
         with gr.Column():
             img = gr.Image(
@@ -85,7 +91,7 @@ with demo:
                 btn = gr.Button("Run")
         with gr.Column():
             img2 = gr.Image()
-            text = gr.Text()
+            text = gr.Text(label='Measurements')
 
     btn.click(fn=predict_mask, inputs=[img, threshold], outputs=[img2,text], )
     examples = gr.Examples(examples=['test0001.png',

line_fit.py

@@ -155,7 +155,7 @@ class LineFit:
         mask_meas_lines = self.mask_measured_lines(im, lines)
         #for line in lines:
         #    plt.plot(*self.line_to_arrays(line), 'c-')
-        return calculated_diameter, mask_meas_lines
+        return calculated_diameter, lines
  
 if __name__ == "__main__":
     import os