File size: 3,781 Bytes
4662286 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import gradio as gr
import whisper
import subprocess
import os
import tempfile
from pytube import YouTube
def download_video_from_link(link):
temp_dir = tempfile.mkdtemp()
yt = YouTube(link)
video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
video_path = video.download(output_path=temp_dir)
return video_path, yt.title
def format_timestamp(seconds):
"""Converte segundos para o formato HH:MM:SS"""
hours = int(seconds // 3600)
minutes = int((seconds % 3600) // 60)
seconds = int(seconds % 60)
return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
def transcribe_video(video_file):
with tempfile.TemporaryDirectory() as temp_dir:
if isinstance(video_file, str):
video_path, title = download_video_from_link(video_file)
else:
video_path = os.path.join(temp_dir, "uploaded_video.mp4")
with open(video_path, 'wb') as file:
file.write(video_file)
audio_path = os.path.join(temp_dir, "temp_audio.wav")
command = f"ffmpeg -i \"{video_path}\" -acodec pcm_s16le -ar 16000 -ac 1 \"{audio_path}\""
subprocess.run(command, shell=True)
model = whisper.load_model("base")
result = model.transcribe(audio_path)
transcription_with_timestamps = ""
transcription_plain = ""
if "segments" in result:
for segment in result["segments"]:
start_time = format_timestamp(segment["start"])
end_time = format_timestamp(segment["end"])
text = segment["text"]
transcription_with_timestamps += f"[{start_time} - {end_time}] {text}\n"
transcription_plain += f"{text} "
else:
transcription_plain = result["text"]
return transcription_with_timestamps, transcription_plain.strip()
def clear_fields():
return "", ""
def start_transcription_interface():
css = """
body { font-family: Arial, sans-serif; }
.gr-button { background-color: #4CAF50; color: white; border: none; padding: 10px 24px; border-radius: 4px; cursor: pointer; }
.gr-button:hover { background-color: #45a049; }
.gr-textbox { border-radius: 4px; border: 1px solid #ccc; padding: 10px; }
.gr-row { margin-bottom: 20px; }
.gr-markdown a { color: #3182ce; }
.show-api {display: none !important;} .built-with {display: none !important;}
"""
with gr.Blocks(css=css) as app:
gr.Markdown("### Transcritor de Vídeo ou Áudio em Texto")
with gr.Row():
upload = gr.File(label="Fazer upload do vídeo", type="binary")
link_input = gr.Textbox(label="Ou insira o link do vídeo")
gr.Markdown("[Baixar vídeos do Youtube, Facebook e Instagram](https://ummy.net/pt33ZN/)")
submit_btn = gr.Button("Transcrever vídeo ou áudio p/ texto")
clear_btn = gr.Button("Limpar os Campos")
output_with_timestamps = gr.Textbox(label="Transcrição com tempo", lines=6)
output_plain = gr.Textbox(label="Transcrição sem tempo", lines=4)
def handle_input(upload_file, link):
if upload_file is not None:
return transcribe_video(upload_file)
elif link:
return transcribe_video(link)
else:
return "Por favor, faça upload de um vídeo ou insira um link.", ""
submit_btn.click(fn=handle_input, inputs=[upload, link_input], outputs=[output_with_timestamps, output_plain])
clear_btn.click(fn=clear_fields, inputs=[], outputs=[output_with_timestamps, output_plain])
app.launch(share=True)
start_transcription_interface() |