Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 from transformers import AutoProcessor, BarkModel
 import scipy
 from pytube import YouTube
+import ffmpeg
 
 
 # device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -40,6 +41,14 @@ def load_video_yt(vid):
     vid = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename="tmp.mp4")
     vid_aud = yt.streams.filter(only_audio=True)[0].download(filename="tmp_aud.mp3")
     return vid, vid_aud 
+
+def trim_clip(clip):
+    start_pos = 1
+    duration = 10
+    audio_input = ffmpeg.input(url)
+    audio_output = ffmpeg.output(audio_input, save_location, ss=start_pos, t=duration)
+    audio_output.run()
+    return audio_output
     
 with gr.Blocks() as app:
     with gr.Column():
@@ -56,6 +65,9 @@ with gr.Blocks() as app:
             with gr.Row():
                 in_aud_yt = gr.Textbox(label="YouTube URL")
                 load_yt_btn = gr.Button("Load URL")
+            with gr.Row():
+                trim_clip_btn = gr.Button("Trim Clip")\
+                trim_aud = gr.Audio(source='upload')
             yt_vid = gr.Video()
             alt_go_btn = gr.Button()
         #speaker_num = gr.Number(value=0)
@@ -65,6 +77,7 @@ with gr.Blocks() as app:
 
     go_btn.click(run_bark,[in_text, speaker_num, speaker_lang],out_audio)
     load_yt_btn.click(load_video_yt, in_aud_yt, [yt_vid,in_aud_file])
+    trim_clip_btn.click(trim_clip,None,trim_aud)
     #alt_go_btn.click()
 
 app.launch()