Add logging

app.py

@@ -4,6 +4,9 @@ import mediapipe as mp
 import numpy as np
 import tempfile
 import mediapy as media
+import log_utils
+
+logger = log_utils.get_logger()
 
 mp_hands = mp.solutions.hands
 mp_hands_connections = mp.solutions.hands_connections
@@ -20,24 +23,27 @@ connections = {
     'HAND_PINKY_FINGER_CONNECTIONS': mp_hands_connections.HAND_PINKY_FINGER_CONNECTIONS,
 }
 
-def process_image(img, selected_connection, draw_background):
-  print(img)
+def draw_landmarks(img, selected_connections, draw_background):
   results = hands.process(img)
   output_img = img if draw_background else np.zeros_like(img)  
   if results.multi_hand_landmarks:
     for hand_landmarks in results.multi_hand_landmarks:
-      mp_draw.draw_landmarks(output_img, hand_landmarks, connections[selected_connection])    
+      mp_draw.draw_landmarks(output_img, hand_landmarks, connections[selected_connections])    
   return output_img
 
+def process_image(img, selected_connections, draw_background):
+  logger.info(f"Processing image with connections: {selected_connections}, draw background: {draw_background}")
+  return draw_landmarks(img, selected_connections, draw_background)
 
-def process_video(video_path, selected_connection, draw_background):
+def process_video(video_path, selected_connections, draw_background):
+  logger.info(f"Processing video with connections: {selected_connections}, draw background: {draw_background}")
   with tempfile.NamedTemporaryFile() as f:
     out_path = f"{f.name}.{video_path.split('.')[-1]}"
     with media.VideoReader(video_path) as r:
       with media.VideoWriter(
           out_path, shape=r.shape, fps=r.fps, bps=r.bps) as w:
         for image in r:
-          w.add_image(process_image(image, selected_connection, draw_background))
+          w.add_image(draw_landmarks(image, selected_connections, draw_background))
     return out_path
 
 
@@ -53,7 +59,7 @@ with demo:
   with gr.Column():  
     draw_background = gr.Checkbox(value=True, label="Draw background?")
     connection_keys = list(connections.keys())
-    selected_connection = gr.Dropdown(
+    selected_connections = gr.Dropdown(
       label="Select connections to draw", 
       choices=connection_keys,
       value=connection_keys[0],
@@ -77,9 +83,10 @@ with demo:
     processed_image = gr.Image()
 
   gr.Markdown('<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=kristyc.mediapipe-hands" />')
-  submit_uploaded_image.click(fn=process_image, inputs=[uploaded_image, selected_connection, draw_background], outputs=[processed_image])
-  submit_camera_picture.click(fn=process_image, inputs=[camera_picture, selected_connection, draw_background], outputs=[processed_image])
-  submit_recorded_video.click(fn=process_video, inputs=[recorded_video, selected_connection, draw_background], outputs=[processed_video])
-  submit_uploaded_video.click(fn=process_video, inputs=[recorded_video, selected_connection, draw_background], outputs=[processed_video])
+  setting_inputs = [selected_connections, draw_background]
+  submit_uploaded_image.click(fn=process_image, inputs=[uploaded_image, *setting_inputs], outputs=[processed_image])
+  submit_camera_picture.click(fn=process_image, inputs=[camera_picture, *setting_inputs], outputs=[processed_image])
+  submit_recorded_video.click(fn=process_video, inputs=[recorded_video, *setting_inputs], outputs=[processed_video])
+  submit_uploaded_video.click(fn=process_video, inputs=[recorded_video, *setting_inputs], outputs=[processed_video])
 
 demo.launch()

log_utils.py

@@ -0,0 +1,9 @@
+import logging
+
+logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+
+def get_logger(name=None, level=logging.INFO):
+  """Return a logger"""
+  logger = logging.getLogger(name)
+  logger.setLevel(level)
+  return logger