added choice of fonts

app.py

@@ -23,12 +23,21 @@ import yaml
 
 #########################################
 
+
+FONTS = {'amiko': "font/Amiko-Regular.ttf",
+        'nature': "font/LoveNature.otf", 
+        'painter':"font/PainterDecorator.otf",
+        'animals': "font/UncialAnimals.ttf", 
+        'zen': "font/ZEN.TTF"}
+
+
 def draw_keypoints_on_image(image,
                             keypoints,
                             map_label_id_to_str,
                             color='red',
                             radius=2,
                             use_normalized_coordinates=True,
+                            gr_pose_font_input,
                             ):
     """Draws keypoints on an image.
     Modified from:
@@ -43,7 +52,7 @@ def draw_keypoints_on_image(image,
     """
     # get a drawing context
     draw = ImageDraw.Draw(image)
-    # font = ImageFont.truetype("sans-serif.ttf", 16)
+    # font = ImageFont.truetype("sans-serif.ttf", 16)    
 
     im_width, im_height = image.size
     keypoints_x = [k[0] for k in keypoints]
@@ -55,7 +64,7 @@ def draw_keypoints_on_image(image,
         keypoints_y = tuple([im_height * y for y in keypoints_y])
 
     # draw ellipses around keypoints and add string labels
-    font = ImageFont.truetype("font/Amiko-Regular.ttf", 8) # font = ImageFont.truetype(<font-file>, <font-size>)
+    font = ImageFont.truetype(FONTS[gr_pose_font_input], 8) # font = ImageFont.truetype(<font-file>, <font-size>)
     for i, (keypoint_x, keypoint_y) in enumerate(zip(keypoints_x, keypoints_y)):
         draw.ellipse([(keypoint_x - radius, keypoint_y - radius),
                       (keypoint_x + radius, keypoint_y + radius)],
@@ -158,7 +167,9 @@ def predict_pipeline(img_input,
                      model_input_str,
                      flag_dlc_only,
                      bbox_likelihood_th,
-                     kpts_likelihood_th):
+                     kpts_likelihood_th,
+                     gr_pose_font_input,
+                     ):
 
     ############################################################
     ## Get DLC model and labels as strings   
@@ -210,7 +221,9 @@ def predict_pipeline(img_input,
                                 map_label_id_to_str,
                                 color='red',
                                 radius=2,
-                                use_normalized_coordinates=False)
+                                use_normalized_coordinates=False,
+                                gr_pose_font_input
+                                )
         return img_input, #list_kpts_per_crop
 
     else:
@@ -233,6 +246,7 @@ def predict_pipeline(img_input,
                                     color='red',
                                     radius=2,
                                     use_normalized_coordinates=False,  # if True, then I should use md_results.xyxyn
+                                    gr_pose_font_input,
                                     )
 
             ## Paste crop in original image
@@ -282,7 +296,7 @@ inputs = [gr_image_input,
                      gr_slider_conf_bboxes,
                      gr_slider_conf_keypoints,
                      #gr_keypt_color,
-                     #gr_pose_font_input,
+                     gr_pose_font_input,
                      #gr_slider_font_size, 
                      ]                                                                                
 
@@ -314,7 +328,8 @@ demo = gr.Interface(predict_pipeline,
              title=gr_title, 
              description=gr_description,
              theme="huggingface",
-             live=True)
+             #live=True
+             )
 
 demo.launch(enable_queue=True)