Update app.py

app.py

@@ -1,5 +1,34 @@
 import gradio as gr
 from modeler import SfM
+from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
+from pytube import YouTube
+
+
+def load_video_yt(vid):
+    yt = YouTube(vid)
+    vid = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename=f"{uid}-tmp.mp4")
+    vid_aud = yt.streams.filter(only_audio=True)[0].download(filename=f"{uid}-tmp_aud.mp4")
+    print (f'Video Length: {yt.length}')
+    return vid, vid_aud, f"{uid}-tmp_aud.mp4"
+
+def trim_vid():
+    ffmpeg_extract_subclip("video1.mp4", start_time, end_time, targetname="test.mp4")
+
+
+def trim_clip(clip, start_t, end_t):
+    clip = Path(f"{uid}-tmp_aud.mp4")
+    song = AudioSegment.from_file(f"{uid}-tmp_aud.mp4", format="mp4")
+    start_min = int(start_t.split(":",1)[0])
+    start_sec = int(start_t.split(":",1)[1])
+    end_min = int(end_t.split(":",1)[0])
+    end_sec = int(end_t.split(":",1)[1])
+    start = ((start_min*60)+start_sec)*1000
+    end = ((end_min*60)+end_sec)*1000
+    song_clip = song[start: end]
+    song_clip.export(f"{uid}-trim.wav", format="wav")
+    print("New Audio file is created and saved")
+
+    return f"{uid}-trim.wav"
 
 
 def make_model(vid_path):