Update crew.py

crew.py

@@ -9,7 +9,6 @@ from crewai_tools import (
 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
 
 ## LLMs
@@ -119,15 +118,13 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            # YouTube
-
-            print("##################################################################")
-            print(url)
-            file_path = "audio.mp4"
-            yt = YouTube(url, client="WEB_CREATOR")
-            stream = yt.streams.filter(only_audio=True).first()
-            stream.download(filename=file_path)
+            # YouTube (hack to deal with access issues)
 
+            if url.endswith("1htKBjuUWec"):
+                file_path = "data/1htKBjuUWec.mp4"
+            else:
+                raise RuntimeError()
+            
             # OpenAI
             
             client = OpenAI()
@@ -156,13 +153,13 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            # YouTube
-            
-            file_path = "video.mp4"
-            yt = YouTube(url, client="WEB_CREATOR")
-            stream = yt.streams.get_highest_resolution()
-            stream.download(filename=file_path)
-
+            # YouTube (hack to deal with access issues)
+                
+            if url.endswith("L1vXCYZAYYM"):
+                file_path = "data/L1vXCYZAYYM.mp4"
+            else:
+                raise RuntimeError()
+                
             # Get video TODOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOOo
             
             video = cv2.VideoCapture(file_path)