artificialguybr's picture
Update app.py
7c4cd6e verified
raw
history blame
No virus
1.74 kB
import subprocess
import uuid
import ffmpeg
import gradio as gr
import os
import re
# Função para baixar o clipe da Twitch usando twitch-dl
def download_twitch_clip(url, auth_token):
# Gerar um UUID para o nome do arquivo
unique_id = uuid.uuid4()
output_filename = f"{unique_id}.mkv"
# Comando para baixar o vídeo
command = ["twitch-dl", "download", url, "-q", "source", "-o", output_filename]
# Adiciona o token de autenticação, se fornecido
if auth_token.strip():
command.extend(["-a", auth_token])
# Executa o comando de download
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = process.communicate()
if process.returncode != 0:
raise Exception(f"Erro no download do vídeo: {stderr.decode('utf-8')}")
return output_filename
# Função para converter MKV para MP4 usando ffmpeg-python
def convert_to_mp4(input_file):
output_file = input_file.replace('.mkv', '.mp4')
try:
(
ffmpeg
.input(input_file)
.output(output_file, vcodec='libx264', acodec='aac')
.run(overwrite_output=True)
)
return output_file
except ffmpeg.Error as e:
print(f"Erro ao converter o arquivo: {e}")
return None
# Interface Gradio
def gradio_interface(url, auth_token=""):
mkv_file = download_twitch_clip(url, auth_token)
mp4_file = convert_to_mp4(mkv_file)
return mp4_file
iface = gr.Interface(
fn=gradio_interface,
inputs=[
gr.Textbox(label="URL do Clipe da Twitch"),
gr.Textbox(label="Token de Autenticação (opcional)")
],
outputs=gr.Video()
)
# Executar a interface
iface.launch()