Update app.py

app.py

@@ -3,8 +3,10 @@ from modeler import SfM
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 from pytube import YouTube
 import uuid
+import os 
 uid=uuid.uuid4()
-
+if not os.path.exists(f'{uid}'):
+   os.makedirs(f'{uid}')
 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")
@@ -25,26 +27,12 @@ def trim_vid(vid,start_time,end_time):
     ffmpeg_extract_subclip(vid, start, end, targetname=f"{uid}-clip.mp4")
     return f"{uid}-clip.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):
-    sfm = SfM('results/', False, 'videos/vid1.mp4', 27)
+    sfm = SfM(f'{uid}/', False, f'{uid}-clip.mp4', 27)
     sfm.structure_from_motion()
-
+    return f'{uid}/'
+    
 with gr.Blocks() as app:
     with gr.Row():
         inp_url=gr.Textbox(label="URL")
@@ -56,7 +44,9 @@ with gr.Blocks() as app:
         start_time = gr.Textbox(label = "Start", value = "0:00:00", placeholder = "0:00:23")
         end_time = gr.Textbox(label = "End", value = "0:00:05", placeholder = "0:00:54")
         trim_clip_btn = gr.Button("Trim Clip")        
+    make_btn=gr.Button()
     out=gr.Files()
+    make_btn.click(make_model,clip_vid,out)
     load_yt_btn.click(load_video_yt, inp_url, [pre_vid])
     trim_clip_btn.click(trim_vid,[pre_vid,start_time,end_time],clip_vid)
 app.launch()