Update app.py

app.py

@@ -1,22 +1,78 @@
 import gradio as gr
 import whisper
+import yt_dlp
+from pydub import AudioSegment
+import tempfile
+import os
 
 # Carregar o modelo Whisper
 model = whisper.load_model("base")
 
-# Função para transcrever o áudio
-def transcribe(audio):
+# Função para transcrever áudio local
+def transcrever_audio(audio):
+    if not audio:
+        return "Por favor, envie um arquivo de áudio."
     result = model.transcribe(audio.name)
     return result["text"]
 
-# Interface do Gradio
-interface = gr.Interface(
-    fn=transcribe,
-    inputs=gr.Audio(source="microphone", type="file", label="Envie ou grave um áudio"),
-    outputs="text",
-    title="Whisper Transcriber",
-    description="Transcreva áudio em texto usando o modelo Whisper da OpenAI.",
-)
+# Função para transcrever áudio de YouTube
+def transcrever_youtube(link):
+    if not link:
+        return "Por favor, insira um link do YouTube."
+    
+    # Baixar o áudio do YouTube
+    ydl_opts = {
+        "format": "bestaudio/best",
+        "outtmpl": "audio.%(ext)s",
+        "postprocessors": [{
+            "key": "FFmpegExtractAudio",
+            "preferredcodec": "mp3",
+            "preferredquality": "192",
+        }],
+        "quiet": True,
+    }
+    
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info = ydl.extract_info(link, download=True)
+        audio_path = info['requested_downloads'][0]['filepath']
+    
+    # Transcrever
+    result = model.transcribe(audio_path)
+    os.remove(audio_path)  # Limpar arquivo temporário
+    return result["text"]
+
+# Função para transcrever áudio gravado do microfone
+def transcrever_microfone(audio):
+    if not audio:
+        return "Por favor, grave um áudio."
+    result = model.transcribe(audio.name)
+    return result["text"]
+
+# Interface com abas
+with gr.Blocks() as demo:
+    gr.Markdown("## **Transcritor Whisper - Áudio para Texto**")
+    with gr.Tab("Microfone"):
+        gr.Markdown("Grave um áudio e transcreva automaticamente.")
+        microfone = gr.Audio(source="microphone", type="file", label="Gravar do microfone")
+        botao_microfone = gr.Button("Transcrever")
+        resultado_microfone = gr.Textbox(label="Transcrição")
+        botao_microfone.click(transcrever_microfone, inputs=microfone, outputs=resultado_microfone)
+    
+    with gr.Tab("Arquivo de Áudio"):
+        gr.Markdown("Envie um arquivo de áudio para transcrever.")
+        arquivo_audio = gr.Audio(type="file", label="Envie seu arquivo de áudio")
+        botao_audio = gr.Button("Transcrever")
+        resultado_audio = gr.Textbox(label="Transcrição")
+        botao_audio.click(transcrever_audio, inputs=arquivo_audio, outputs=resultado_audio)
+    
+    with gr.Tab("YouTube"):
+        gr.Markdown("Insira o link de um vídeo do YouTube para transcrição.")
+        link_youtube = gr.Textbox(label="URL do YouTube")
+        botao_youtube = gr.Button("Transcrever")
+        resultado_youtube = gr.Textbox(label="Transcrição")
+        botao_youtube.click(transcrever_youtube, inputs=link_youtube, outputs=resultado_youtube)
 
+# Executar o app
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)
+