Update app.py

app.py

@@ -182,14 +182,14 @@ def save_dxf_spline(inflated_contours, scaling_factor, height):
 
     doc = ezdxf.new(units=0)
     doc.units = ezdxf.units.IN
-    doc.header['$INSUNITS'] = ezdxf.units.IN
+    doc.header["$INSUNITS"] = ezdxf.units.IN
 
     msp = doc.modelspace()
 
     for contour in inflated_contours:
         resampled_contour = resample_contour(contour)
         points = [
-            (x * scaling_factor, (height - y)* scaling_factor)
+            (x * scaling_factor, (height - y) * scaling_factor)
             for x, y in resampled_contour
         ]
         if len(points) >= 3:
@@ -378,7 +378,7 @@ def predict(image, offset_inches):
     objects_mask = resize_img(
         objects_mask, (shrunked_img.shape[1], shrunked_img.shape[0])
     )
-    offset_pixels = (offset_inches / scaling_factor)*2+1
+    offset_pixels = (offset_inches / scaling_factor) * 2 + 1
     dilated_mask = cv2.dilate(
         objects_mask, np.ones((int(offset_pixels), int(offset_pixels)), np.uint8)
     )
@@ -387,9 +387,18 @@ def predict(image, offset_inches):
     # objects_mask_scaled = scale_image(objects_mask, scaling_factor)
     Image.fromarray(dilated_mask).save("./outputs/scaled_mask_new.jpg")
     outlines, contours = extract_outlines(dilated_mask)
+    shrunked_img_contours = cv2.drawContours(
+        shrunked_img, contours, -1, (0, 0, 255), thickness=2
+    )
     dxf = save_dxf_spline(contours, scaling_factor, processed_size[0])
 
-    return outlines, dxf, dilated_mask, scaling_factor, reference_obj_img
+    return (
+        cv2.cvtColor(shrunked_img_contours, cv2.COLOR_BGR2RGB),
+        outlines,
+        dxf,
+        dilated_mask,
+        scaling_factor,
+    )
 
 
 if __name__ == "__main__":
@@ -397,16 +406,19 @@ if __name__ == "__main__":
 
     ifer = gr.Interface(
         fn=predict,
-        inputs=[gr.Image(label="Input Image"), gr.Number(label="Offset value for Mask(inches)", value=0.075)],
+        inputs=[
+            gr.Image(label="Input Image"),
+            gr.Number(label="Offset value for Mask(inches)", value=0.075),
+        ],
         outputs=[
             gr.Image(label="Ouput Image"),
+            gr.Image(label="Outlines of Objects"),
             gr.File(label="DXF file"),
             gr.Image(label="Mask"),
             gr.Textbox(
                 label="Scaling Factor(mm)",
-                placeholder="Every pixel is equal to mentioned number in mm(milimeter)",
+                placeholder="Every pixel is equal to mentioned number in inches",
             ),
-            gr.Image(label="Image used for calculating scaling factor"),
         ],
         examples=[
             ["./examples/Test20.jpg", 0.075],