Update app.py

app.py

@@ -1,7 +1,7 @@
 import subprocess
 import uuid
-import ffmpeg
 import gradio as gr
+from youtube_dl import YoutubeDL
 
 title_and_description = """
 # Download de clipes da Twitch
@@ -20,48 +20,23 @@ Sinta-se à vontade para usar e gerar seus próprios videoclipes!
 Para baixar o vídeo é só clicar no botão de download em cima do vídeo. Seta apontada para baixo.
 """
 
-# Function to download Twitch clip using twitch-dl
-def download_twitch_clip(url, auth_token):
-    # Generate a UUID for the file name
+# Function to download Twitch clip using youtube-dl
+def download_twitch_clip(url):
     unique_id = uuid.uuid4()
-    output_filename = f"{unique_id}.mkv"
+    output_filename = f"{unique_id}.mp4"
+    ydl_opts = {
+        'format': 'best[ext=mp4]',
+        'outtmpl': output_filename
+    }
 
-    # Command to download the video
-    command = ["twitch-dl", "download", url, "-q", "source", "-o", output_filename]
-
-    # Add authentication token if provided
-    if auth_token.strip():
-        command.extend(["-a", auth_token])
-
-    # Execute the download command
-    process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    stdout, stderr = process.communicate()
-
-    if process.returncode != 0:
-        raise Exception(f"Error in video download: {stderr.decode('utf-8')}")
+    with YoutubeDL(ydl_opts) as ydl:
+        ydl.download([url])
 
     return output_filename
 
-# Function to convert MKV to MP4 using 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"Error in file conversion: {e}")
-        return None
-
 # Gradio interface
-def gradio_interface(url, auth_token=""):
-    mkv_file = download_twitch_clip(url, auth_token)
-    mp4_file = convert_to_mp4(mkv_file)
+def gradio_interface(url):
+    mp4_file = download_twitch_clip(url)
     return mp4_file, mp4_file  # Return the file path twice, for both the video and file components
 
 with gr.Blocks() as app:
@@ -69,19 +44,18 @@ with gr.Blocks() as app:
     
     with gr.Row():
         with gr.Column():
-            result_video = gr.Video(label="Vídeo Output")
-            download_link = gr.File(label="Download MP4")  # File component for downloading
+            result_video = gr.Video(label="Video Output")
+            download_link = gr.File(label="Download MP4")
     
     with gr.Row():
         with gr.Column():
             url_input = gr.Textbox(label="Twitch Clip URL")
-            auth_token_input = gr.Textbox(label="Authentication Token (optional)", type="password")
             run_btn = gr.Button("Download Clip")
             
     run_btn.click(
         gradio_interface,
-        inputs=[url_input, auth_token_input],
-        outputs=[result_video, download_link]  # Include the file component in the outputs
+        inputs=[url_input],
+        outputs=[result_video, download_link]
     )
 
 app.queue()