Update app.py

app.py

@@ -1,24 +1,35 @@
 import re
 import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api._errors import TranscriptsDisabled
 import traceback
 
 def extract_video_id(youtube_url):
-    match = re.search(r'(?:v=|\/)([0-9A-Za-z_-]{11}).*', youtube_url)
-    return match.group(1) if match else None
+    # Regex para extrair o ID do vídeo da URL do YouTube
+    match = re.search(r'(?:youtu\.be\/|(?:www\.)?youtube\.com\/(?:watch\?v=|embed\/|v\/|.+\?v=))([^&]{11})', youtube_url)
+    video_id = match.group(1) if match else None
+    print(f"URL recebida: {youtube_url}")
+    print(f"ID do vídeo extraído: {video_id}")
+    return video_id
 
 def get_available_transcripts(video_id):
     try:
         transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
         available_languages = [t.language_code for t in transcript_list]
         return f"Transcrições disponíveis: {available_languages}"
+    except TranscriptsDisabled:
+        return "Não há transcrições disponíveis para este vídeo."
     except Exception as e:
         return f"Erro ao listar transcrições: {str(e)}"
 
 def get_transcript(video_id, language="en"):
     try:
+        print(f"Tentando obter transcrição para o vídeo ID: {video_id} no idioma: {language}")
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=[language])
+        print("Transcrição obtida com sucesso.")
         return "\n".join([f"{t['start']}: {t['text']}" for t in transcript])
+    except TranscriptsDisabled:
+        return "Erro: As legendas estão desabilitadas para este vídeo."
     except Exception as e:
         available_transcripts = get_available_transcripts(video_id)
         error_message = f"Erro ao obter transcrição: {str(e)}\n{available_transcripts}\n"
@@ -29,9 +40,8 @@ def gradio_interface(youtube_url, language):
     video_id = extract_video_id(youtube_url)
     if not video_id:
         return "Erro: URL inválida. Por favor, insira um link válido do YouTube."
-    else:
-        print(f"ID do vídeo extraído: {video_id}")
-    return get_transcript(video_id, language)
+    transcript = get_transcript(video_id, language)
+    return transcript
 
 iface = gr.Interface(
     fn=gradio_interface,
@@ -44,4 +54,4 @@ iface = gr.Interface(
     description="Insira a URL de um vídeo do YouTube e selecione o idioma da transcrição."
 )
 
-iface.launch()
+iface.launch(share=True)