Update app.py

app.py

@@ -153,7 +153,7 @@ def video_clip(program,fps,qual,model_name,denoise_strength,face_enhance,outscal
     out = f"{uid}/enhanced_vid.mp4"
     yield out,out,"Video Complete"   
 
-def gif_clip(program,fps,qual,model_name,denoise_strength,face_enhance,outscale):
+def gif_clip(program,fps,qual,model_name,denoise_strength,face_enhance,outscale,idx_vid):
     _=None    
     fps=float(fps)
     new_video_in = str(f"{uid}-clip.mp4")
@@ -192,13 +192,21 @@ def gif_clip(program,fps,qual,model_name,denoise_strength,face_enhance,outscale)
         print (f'out_box::{out_box}')
         clip = ImageSequenceClip(out_box, fps = fps) 
     else:
-        mes="Writing GIF"
         clip = ImageSequenceClip(fbox, fps = fps) 
-    yield _,_,"Writing GIF"    
-    clip.speedx(2).to_gif(f"{uid}/clip_gif.gif",program=program)    
-    yield _,_,"Saving GIF"    
-    out = f"{uid}/clip_gif.gif"
-    yield out,out,"GIF Complete"    
+    yield _,_,"Writing" 
+    if idx_vid == "GIF":
+        clip.speedx(2).to_gif(f"{uid}/clip_gif.gif",program=program)    
+        out = f"{uid}/clip_gif.gif"
+        yield out,out,"GIF Complete"    
+        
+    if idx_vid == "Video"
+        videoclip = VideoFileClip(new_video_in)
+        audioclip=videoclip.audio
+        #clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(images, fps)
+        clip.audio=audioclip
+        clip.write_videofile(f"{uid}/enhanced_vid.mp4")        
+        out = f"{uid}/enhanced_vid.mp4"
+        yield out,out,"Video Complete"    
 
 
 def update_speed(inp,clip_speed,fps):
@@ -251,7 +259,7 @@ with gr.Blocks(css=css) as app:
             with gr.Tab("Image"):
                 with gr.Group():
                     inp_img_url=gr.Textbox(label="Image URL")
-                    inp_img=gr.Image(label = "Input Image"):
+                    inp_img=gr.Image(label = "Input Image")
                     outp_img=gr.Image(label="Enhanced Image")
             with gr.Tab("Video/GIF"):
                 with gr.Group():
@@ -338,7 +346,7 @@ with gr.Blocks(css=css) as app:
         yield out,"Sample Frame Complete"        
     samp_btn.click(run_samp,[choose_prog,fps,qual,model_name,denoise_strength,face_enhance,outscale,frame_targ],[samp_im,gif_stat])    
     speed_btn.click(update_speed,[hid_box,clip_speed,hid_fps],[outp_vid,hid_box,outp_file])
-    gif_btn.click(gif_clip,[choose_prog,fps,qual,model_name,denoise_strength,face_enhance,outscale],[gif_show,gif_file,gif_stat])
+    gif_btn.click(gif_clip,[choose_prog,fps,qual,model_name,denoise_strength,face_enhance,outscale,idx_vid],[gif_show,gif_file,gif_stat])
     trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,trim_count])
     outp_vid.change(load_video,outp_vid,[frame_count,fps,end_f,hid_box,idx_vid]).then(trim_vid,[hid_box,start_f,end_f],[out_trim,trim_count])
     go_btn.click(dl,[inp_url,outp_vid],[outp_vid,outp_file,out_trim,hid_box,hid_fps])