Update tools.py

tools.py

@@ -527,19 +527,28 @@ def read_file(file_path: str) -> Dict[str, Any]:
 def get_youtube_transcript(video_url: str) -> Dict[str, Any]:
     """
     Récupère la transcription d'une vidéo YouTube à partir de son URL standard.
+
+    Args:
+        video_url (str): L'URL complète et standard de la vidéo YouTube dont la transcription est nécessaire.
+
+    Returns:
+        Un dictionnaire contenant la transcription complète sous la clé 'transcript' en cas de succès,
+        ou un message d'erreur sous la clé 'error' en cas d'échec.
     """
     try:
         video_id = None
-        # Cherche l'ID après "v=" ou sur le domaine "youtu.be"
+        # Cherche l'ID après "v="
         if "v=" in video_url:
             video_id = video_url.split("v=")[1].split('&')[0]
-        elif "youtu.be/" in video_url:
-            video_id = video_url.split("youtu.be/")[1].split('?')[0]
+        # Cherche l'ID dans les URLs du benchmark
+        elif video_url.startswith("http://googleusercontent.com/youtube.com/"):
+            video_id = video_url.split('/')[-1]
 
         if not video_id:
             return {"error": f"Impossible d'extraire l'ID de la vidéo depuis l'URL : {video_url}"}
 
         print(f"📖 Récupération de la transcription pour la vidéo YouTube : {video_id}")
+        # On demande la transcription en anglais (le plus courant) et en français.
         transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=['en', 'fr'])
         transcript_text = " ".join([item['text'] for item in transcript_list])
         return {"status": "success", "transcript": transcript_text[:15000]}