Update app.py

app.py

@@ -193,21 +193,12 @@ def gif_clip(program,fps,qual,model_name,denoise_strength,face_enhance,outscale,
         clip = ImageSequenceClip(out_box, fps = fps) 
     else:
         clip = ImageSequenceClip(fbox, fps = fps) 
-    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"    
-
+    yield _,_,"Writing GIF" 
+    clip.speedx(2).to_gif(f"{uid}/clip_gif.gif",program=program)    
+    out = f"{uid}/clip_gif.gif"
+    yield out,out,"GIF Complete"    
+    
+    
 
 def update_speed(inp,clip_speed,fps):
     if "-split.mp4" in inp:
@@ -255,13 +246,13 @@ with gr.Blocks(css=css) as app:
     with gr.Row():
         gr.Column()
         with gr.Column():
-            #with gr.Box():
-            with gr.Tab("Image"):
+            choose_media = gr.Radio(label="Media Type",choices=["Image","Video","GIF"],value="Image")
+            with gr.Box(visible=True) as im_box:
                 with gr.Group():
                     inp_img_url=gr.Textbox(label="Image URL")
                     inp_img=gr.Image(label = "Input Image")
                     outp_img=gr.Image(label="Enhanced Image")
-            with gr.Tab("Video/GIF"):
+            with gr.Box(visible=False) as vid_box:
                 with gr.Group():
                     idx_vid=gr.Radio(choices=["Video","GIF"])
                     inp_url = gr.Textbox(label="Video URL")
@@ -303,9 +294,12 @@ with gr.Blocks(css=css) as app:
                                 samp_btn=gr.Button("Get Sample")
                             samp_im = gr.Image(label="Sample Frame")    
                         
-                    with gr.Row():
+                    with gr.Row(visible=False) as gif_row:
                         gif_btn = gr.Button("Make GIF")
                         gif_stat=gr.Textbox(label="Status")
+                    with gr.Row(visible=False) as vid_row:
+                        vid_btn = gr.Button("Make Video")
+                        vid_stat=gr.Textbox(label="Status")
                     with gr.Row():
                         with gr.Column():
                             gif_show = gr.Video()
@@ -344,9 +338,33 @@ with gr.Blocks(css=css) as app:
         yield _,"Creating Sample Frame"
         out = esr(out_url, model_name, float(denoise_strength), face_enhance, int(outscale))
         yield out,"Sample Frame Complete"        
+
+    def new_media(inp):
+        if inp == "Image":
+            im_box = gr.update(visible=True)
+            vid_box = gr.update(visible=False)
+            gif_row = gr.update(visible=False)
+            vid_row = gr.update(visible=False)
+            return (im_box,vid_box,gif_row,vid_row)
+        if inp == "Video":
+            im_box = gr.update(visible=False)
+            vid_box = gr.update(visible=True)
+            gif_row = gr.update(visible=False)
+            vid_row = gr.update(visible=True)
+            return (im_box,vid_box,gif_row,vid_row)
+        if inp == "GIF":
+            im_box = gr.update(visible=False)
+            vid_box = gr.update(visible=True)
+            gif_row = gr.update(visible=True)
+            vid_row = gr.update(visible=False)            
+            return (im_box,vid_box,gif_row,vid_row)
+            
+    choose_media.change(new_media,choose_media,[im_box,vid_box,gif_row,vid_row])
     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,idx_vid],[gif_show,gif_file,gif_stat])
+    gif_btn.click(gif_clip,[choose_prog,fps,qual,model_name,denoise_strength,face_enhance,outscale],[gif_show,gif_file,gif_stat])
+    vid_btn.click(vid_clip,[choose_prog,fps,qual,model_name,denoise_strength,face_enhance,outscale],[gif_show,gif_file,vid_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])