fixes

tasks.py

@@ -12,8 +12,6 @@ def process_video(video_path: str,vitpose: VitPose,user_id: str):
     
     logger.info(f"starting task {video_path}")
     
-    # Create the static directory if it doesn't exist
-    os.makedirs("static", exist_ok=True)
     
     new_file_name = video_path.split(".")[0] + "edited." + video_path.split(".")[1]
     new_file_name = os.path.join("static", new_file_name)
@@ -25,16 +23,6 @@ def process_video(video_path: str,vitpose: VitPose,user_id: str):
     
     logger.info(f"Video processed {annotated_video_path}")
     
-    # Add debug logging to check if the file exists
-    if not os.path.exists(annotated_video_path):
-        logger.error(f"File not found at {annotated_video_path}")
-        # Try to find where the file might be
-        logger.info(f"Current working directory: {os.getcwd()}")
-        # Use the output_video_path from vitpose as backup
-        if os.path.exists(vitpose.output_video_path):
-            logger.info(f"Found file at {vitpose.output_video_path}")
-            annotated_video_path = vitpose.output_video_path
-    
     with open(annotated_video_path, "rb") as f:
         contents = f.read()
         

vitpose.py

@@ -38,7 +38,7 @@ class VitPose:
             if not ret:
                 break
             frames.append(frame)
-        return frames[:10]
+        return frames
     
     def run(self,video):
         frames = self.video_to_frames(video)