artificialguybr commited on
Commit
8eefd04
1 Parent(s): 4765650

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -1,22 +1,27 @@
1
  import gradio as gr
2
  import subprocess
3
  import os
 
4
 
5
  # Função para baixar o clipe da Twitch
6
  def download_twitch_clip(url, auth_token):
7
  # Comando básico para download
8
- command = ["twitch-dl", "download", url]
9
 
10
  # Adiciona o token de autenticação, se fornecido
11
  if auth_token.strip():
12
  command.extend(["-a", auth_token])
13
 
14
- # Executa o comando de download
15
- subprocess.run(command)
 
16
 
17
- # Extrai o ID do vídeo da URL para criar o nome do arquivo
18
- video_id = url.split('/')[-1]
19
- file_name = f"{video_id}.mkv" # Assumindo que o formato padrão é mkv
 
 
 
20
 
21
  return file_name
22
 
 
1
  import gradio as gr
2
  import subprocess
3
  import os
4
+ import re
5
 
6
  # Função para baixar o clipe da Twitch
7
  def download_twitch_clip(url, auth_token):
8
  # Comando básico para download
9
+ command = ["twitch-dl", "download", "-q", "source", url]
10
 
11
  # Adiciona o token de autenticação, se fornecido
12
  if auth_token.strip():
13
  command.extend(["-a", auth_token])
14
 
15
+ # Executa o comando de download e captura a saída
16
+ process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
17
+ stdout, stderr = process.communicate()
18
 
19
+ # Usa expressão regular para encontrar o nome do arquivo na saída
20
+ match = re.search(r"Target: (.+\.mp4)", stderr.decode("utf-8"))
21
+ if match:
22
+ file_name = match.group(1)
23
+ else:
24
+ raise Exception("Não foi possível encontrar o nome do arquivo de vídeo.")
25
 
26
  return file_name
27