add bb and text aprox

app.py

@@ -160,6 +160,23 @@ def crop_animal_detections(img_in,
 
     return list_np_animal_crops
 
+def draw_rectangle_text(img,results,font_style='amiko',font_size=8,):
+    bbxyxy = results.xyxy[0].tolist()[0]
+    w, h = bbxyxy[2], bbxyxy[3]
+    shape = [(bbxyxy[0], bbxyxy[1]), (w , h)]
+    imgR = ImageDraw.Draw(img)  
+    imgR.rectangle(shape,  outline ="red",width=5)
+
+    confidence = bbxyxy[4]
+    print(confidence)
+    font = ImageFont.truetype(FONTS[font_style], font_size) 
+    imgR.text((keypoint_x + marker_size, keypoint_y + marker_size),#(0.5*im_width, 0.5*im_height), #-------
+                      map_label_id_to_str[i],
+                      ImageColor.getcolor(keypt_color, "RGB"), # rgb
+                      font=font)
+
+    return imgR
+
 ##########################################
 def predict_dlc(list_np_crops,
                 kpts_likelihood_th,
@@ -286,7 +303,8 @@ def predict_pipeline(img_input,
             ## Paste crop in original image https://pillow.readthedocs.io/en/stable/reference/Image.html#PIL.Image.Image.paste
             img_background.paste(img_crop,
                                  box = tuple([int(t) for t in md_results.xyxy[0][ic,:2]]))
-                                                                
+
+            draw_rectangle_text(img_background, md_results)                                                    
         return img_background
 
 #############################################