better way of getting youtube information

app.py

@@ -23,6 +23,13 @@ from takeaways import KeyTakeaways
 from mindmap import MindMap
 import models as md
 
+def get_initial_message():
+    messages=[
+            {"role": "system", "content": "You are a helpful AI Tutor. Who anwers brief questions about AI."},
+            {"role": "user", "content": "I want to learn AI"},
+            {"role": "assistant", "content": "Thats awesome, what do you want to know aboout AI"}
+        ]
+    return messages
 
 REGEXP_YOUTUBE_URL = "^(https?\:\/\/)?((www\.)?youtube\.com|youtu\.be)\/.+$"
 

transcription.py

@@ -4,6 +4,7 @@ from moviepy.editor import *
 from pydub import AudioSegment
 from pydub.utils import make_chunks
 import pydub
+from yt_dlp import YoutubeDL
 from pathlib import Path
 import subprocess
 
@@ -42,21 +43,15 @@ class DownloadAudio:
 
     def __init__(self, link) -> None:
         self.link = link
-        self.yt = YouTube(self.link)
+        with YoutubeDL() as ydl:
+            self.yt = ydl.extract_info(self.link, download=False)
+            
         self.YOUTUBE_VIDEO_ID = link.split("=")[1]
         self.WAV_FILE_NAME = f"{self.YOUTUBE_VIDEO_ID}.wav"
 
     def get_yt_title(self) -> str:
         """Returns the title of the youtube video"""
-        while True:
-            try:
-                title = self.yt.title
-                return title
-            except:
-                print("Failed to get name. Retrying...")
-                time.sleep(1)
-                self.yt = YouTube(self.link)
-                continue
+        return self.yt["title"]
 
     def download(self, pathname:str) -> str:
         """