Update app.py

app.py

@@ -246,7 +246,14 @@ with gr.Blocks() as app:
                 out_trim=gr.Video(format="mp4")
                 hid_box = gr.Textbox(visible=True)
                 hid_fps = gr.Textbox(visible=True)
-
+    with gr.Tab("Quality"):
+        impro_btn = gr.Button("Run")
+        with gr.Row():
+            with gr.Column():
+                clip_in = gr.Video()
+            with gr.Column():
+                clip_out = gr.Video()
+                
     with gr.Tab("Frames"):
         with gr.Row():
             frame_btn = gr.Button("Get Frames")
@@ -304,6 +311,21 @@ with gr.Blocks() as app:
             out_trim,in_vid,trim_count=trim_vid(hid_box,start_f,end_f)
         return out_trim,in_vid,trim_count      
     #outp_vid.change(echo_fn,outp_vid,[out_trim])
+    esr = gr.Interface.load("spaces/Omnibus/Real-ESRGAN-mod")
+    def process_image_1(image):
+        out = esr(img=image, model_name="realesr-general-x4v3", denoise_strength=0.5, face_enhance=False, outscale=2)
+        print (out)
+        out = Image.open(out)
+        return out
+    def improve_quality():
+        clip1 = VideoFileClip(f"{uid}-clip.mp4")
+        clip = clip1.fl_image(process_image_1)
+        clip.write_videofile(f"{uid}-clip-high.mp4", audio=True)
+        return f"{uid}-clip-high.mp4"
+    
+    impro_btn.click(improve_quality,None,clip_out)
+
+        
     info_btn.click(dl_json,inp_url,info_json)
     speed_btn.click(update_speed,[hid_box,clip_speed,hid_fps],[outp_vid,hid_box])
     gif_btn.click(make_gif,fps,[gif_show,gif_file,gif_stat])