Update app.py

app.py

@@ -38,7 +38,21 @@ def detect_ellipses(img_path):
     
     result = aamed.run_AAMED(imgG)
 
-    return [Image.fromarray(imgG), json.dumps(result.tolist())]
+    if len(result) > 0:
+        imgN = imgG.copy()
+        imgN = cv2.cvtColor(imgN, cv2.COLOR_BRAY2RGB)
+        for e in result:
+            x, y, w, h, a, _ = e
+            imgN = cv2.ellipse(imgN, (x, y), (w, h), -a, 0, 360, (0, 255, 0), 3) # image, center_coordinates, axesLength, angle, startAngle, endAngle, color, thickness
+    		"""
+            temp.center.x = detEllipses[i].center.y;
+    		temp.center.y = detEllipses[i].center.x;
+    		temp.size.height = detEllipses[i].size.width;
+    		temp.size.width = detEllipses[i].size.height;
+    		temp.angle = -detEllipses[i].angle;
+            """
+
+    return [Image.fromarray(imgN), json.dumps(result.tolist())]
 
 examples = []