Update app.py

app.py

@@ -1,38 +1,38 @@
 import re
-import streamlit as st
+import gradio as gr
 from youtube_transcript_api import YouTubeTranscriptApi
 
-# Função para extrair o ID do vídeo do YouTube
 def extract_video_id(youtube_url):
-    # Regex para extrair o ID do vídeo
+    # 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)
     return match.group(1) if match else None
 
-# Função para obter a transcrição do vídeo
 def get_transcript(video_id, language="en"):
     try:
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=[language])
-        # Formata a transcrição em texto legível
         return "\n".join([f"{t['start']}: {t['text']}" for t in transcript])
     except Exception as e:
         return f"Erro ao obter transcrição: {str(e)}"
 
-# Configuração do Streamlit
-st.title("Obter Transcrição de Vídeos do YouTube")
-st.write("Insira a URL de um vídeo do YouTube e selecione o idioma para obter a transcrição (se disponível).")
-
-# Entrada da URL do vídeo
-youtube_url = st.text_input("URL do Vídeo (YouTube)", placeholder="Ex: https://www.youtube.com/watch?v=tl1jHm0qC_4")
-
-# Seleção do idioma
-language = st.selectbox("Idioma da Transcrição", ["en", "pt", "es", "fr"], index=0)
-
-# Botão para processar
-if st.button("Obter Transcrição"):
-    # Validação da URL e obtenção da transcrição
+# Função que combina a extração do ID e a obtenção da transcrição
+def gradio_interface(youtube_url, language):
     video_id = extract_video_id(youtube_url)
     if not video_id:
-        st.error("Erro: URL inválida. Por favor, insira um link válido do YouTube.")
-    else:
-        transcript = get_transcript(video_id, language)
-        st.text_area("Transcrição", value=transcript, height=400)
+        return "Erro: URL inválida. Por favor, insira um link válido do YouTube."
+    
+    return get_transcript(video_id, language)
+
+# Criação da interface Gradio
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs=[
+        gr.Textbox(label="URL do Vídeo (YouTube)", placeholder="Ex: https://www.youtube.com/watch?v=tl1jHm0qC_4"),
+        gr.Dropdown(label="Idioma", choices=["en", "pt", "es", "fr"], value="en")
+    ],
+    outputs="text",
+    title="Obter Transcrição de Vídeos do YouTube",
+    description="Insira a URL de um vídeo do YouTube e selecione o idioma da transcrição."
+)
+
+# Execução da interface Gradio
+iface.launch()