ok

requirements.txt

@@ -1,8 +1,8 @@
 fastapi==0.111.1
+imageio==2.34.1
 moviepy==1.0.3
-transformers==4.42.4
 openai==1.34.0
 python-dotenv==1.0.1
 torch==2.3.1
 transformers==4.42.4
-uvicorn==0.30.3
+uvicorn==0.30.3

video_analyzer.py

@@ -5,6 +5,7 @@ import requests
 from tqdm import tqdm
 from dotenv import load_dotenv
 from moviepy.editor import VideoFileClip
+import imageio
 import re
 import urllib.request
 
@@ -28,7 +29,7 @@ class VideoAnalyzer:
             self.download_video()
 
     def download_video(self):
-        self.video_path = "./tmp/video.mp4"
+        self.video_path = "./tmp/video.mkv"
         password_mgr = urllib.request.HTTPPasswordMgrWithDefaultRealm()
         password_mgr.add_password(None, self.video_url, TWILIO_ACCOUNT_SID, TWILIO_AUTH_TOKEN)
         handler = urllib.request.HTTPBasicAuthHandler(password_mgr)
@@ -37,13 +38,16 @@ class VideoAnalyzer:
         urllib.request.urlretrieve(self.video_url, self.video_path)
 
     def extract_images(self):
-        video = VideoFileClip(self.video_path)
-        duration = video.duration
-        time_intervals = [i * (duration / (self.num_images + 1)) for i in range(1, self.num_images + 1)]
+        reader = imageio.get_reader(self.video_path, 'ffmpeg')
+        duration = reader.get_meta_data()['duration']
+        fps = reader.get_meta_data()['fps']
+        num_frames = int(duration * fps)
+        frame_indices = [int(i * (num_frames / (self.num_images + 1))) for i in range(1, self.num_images + 1)]
         self.image_filenames = [f"./tmp/image_{i}.jpg" for i in range(self.num_images)]
 
-        for i, t in enumerate(time_intervals):
-            video.save_frame(self.image_filenames[i], t=t)
+        for i, frame_idx in enumerate(frame_indices):
+            frame = reader.get_data(frame_idx)
+            imageio.imwrite(self.image_filenames[i], frame)
 
     def encode_image(self, image_path):
         with open(image_path, "rb") as image_file:
@@ -107,8 +111,7 @@ class VideoAnalyzer:
 
 if __name__ == "__main__":
     # Use either video_path or video_url
-    # video_path = "./raw/zach.mov"
-    video_url = "https://video.twilio.com/v1/Recordings/RT2c1baf50b6343802964c98e5a6f979e3/Media"
+    video_url = "https://video.twilio.com/v1/Recordings/RTd8f67290c5665868602625bec14518e5/Media"
     analyzer = VideoAnalyzer(video_url=video_url)
     traits = analyzer.retrieve_traits()
     print(traits)