logging

tasks.py

@@ -4,8 +4,13 @@ import os
 from fastapi import UploadFile
 from config import API_URL
 import time
+import logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 def process_video(video_path: str,vitpose: VitPose,user_id: str):
+    
+    logger.info(f"Processing video {video_path}")
         
     new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
     new_file_name = os.path.join("static", new_file_name)
@@ -14,13 +19,16 @@ def process_video(video_path: str,vitpose: VitPose,user_id: str):
     annotated_frames = vitpose.run(video_path)
     annotated_video_path = vitpose.frames_to_video(annotated_frames,rotate=True)
     
+    logger.info(f"Video processed {annotated_video_path}")
+    
     with open(annotated_video_path, "rb") as f:
         contents = f.read()
         
     url = API_URL+ "/excercises/webhooks/video-processed"
-    
+    logger.info(f"Sending video to {url}")
     files = {"file": (annotated_video_path, contents, "video/mp4")}
     response = requests.post(url, files=files, data={"user_id":user_id,"typeMessage":"video_processed","file_name":annotated_video_path}, stream=True)
+    logger.info(f"Video sent to {url}")
     print(response.json())
     os.remove(video_path)
     os.remove(annotated_video_path)

vitpose.py

@@ -5,6 +5,11 @@ import supervision as sv
 import numpy as np
 from rt_pose import PoseEstimationPipeline, PoseEstimationOutput
 
+import logging
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 
 class VitPose:
     def __init__(self):
@@ -38,10 +43,12 @@ class VitPose:
     def run(self,video):
         frames = self.video_to_frames(video)
         annotated_frames = []
-        for frame in frames:
+        for i, frame in enumerate(frames):
+            logger.info(f"Processing frame {i} of {len(frames)}")
             output = self.pipeline(frame)
             annotated_frame = self.visualize_output(frame,output)
             annotated_frames.append(annotated_frame)
+        logger.info(f"Processed {len(annotated_frames)} frames")
         return annotated_frames