Update crew.py

crew.py

@@ -1,4 +1,4 @@
-import cv2, os
+import os
 from crewai import Agent, Crew, Process, Task
 from crewai.tools import tool
 from crewai_tools import (
@@ -10,7 +10,7 @@ from openai import OpenAI
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from pytubefix import YouTube
-from util import get_final_answer, get_img_b64
+from util import get_final_answer, get_img_b64, get_imgs_b64
 
 ## LLMs
 
@@ -171,23 +171,9 @@ def run_crew(question, file_path):
             #stream = yt.streams.get_highest_resolution()
             #stream.download(filename=file_path)
 
-            # Get video TODOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOo
-            
-            video = cv2.VideoCapture(file_path)
-            
-            base64Frames = []
-            
-            while video.isOpened():
-                success, frame = video.read()
-                
-                if not success:
-                    break
-                
-                _, buffer = cv2.imencode(".jpg", frame)
-                
-                base64Frames.append(base64.b64encode(buffer).decode("utf-8"))
-            
-            video.release()
+            # Get images
+
+            imgs_b64 = get_imgs_b64(file_path)
             
             # OpenAI
 
@@ -196,7 +182,7 @@ def run_crew(question, file_path):
             response = client.responses.create(
                 input=[{"role": "user",
                         "content": [{"type": "input_text", "text": (question)},
-                                    *[{"type": "input_image", "image_url": f"data:image/jpeg;base64,{frame}"} for frame in base64Frames]]}],
+                                    *[{"type": "input_image", "image_url": f"data:image/jpeg;base64,{img_b64}"} for img_b64 in imgs_b64]]}],
                 model=VIDEO_MODEL
             )