Update app.py

app.py

@@ -13,6 +13,7 @@ uid=uuid.uuid4()
 if not os.path.exists(f'{uid}'): os.makedirs(f'{uid}')
 if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}-frames')
 if not os.path.exists(f'{uid}-rembg'): os.makedirs(f'{uid}-rembg')
+esr = gr.Interface.load("spaces/Omnibus/Real-ESRGAN-mod")
 
 load_js = """
 function(text_input, url_params) {
@@ -223,7 +224,36 @@ def update_speed(inp,clip_speed,fps):
     final.write_videofile(f'{out_inp}', fps=mod_fps) 
     out = f'{out_inp}'
     return out,out 
-    
+
+def echo_fn(inp):
+    return inp 
+def check_load(inp_url,outp_vid,hid_box,start_f,end_f):
+    if outp_vid == None and inp_url !="":
+        out_trim,in_vid,trim_count=trim_vid(hid_box,start_f,end_f)
+    elif outp_vid !=None and inp_url == "" and hid_box =="":
+        out_trim = None
+        in_vid=outp_vid
+        trim_count = ""
+    elif outp_vid !=None and inp_url == "" and hid_box !="":
+        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])
+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"
+
+
+
+
+
+
 with gr.Blocks() as app:
     with gr.Tab("Load"):
         with gr.Row():
@@ -298,30 +328,7 @@ with gr.Blocks() as app:
         text_output=gr.Textbox()
         url_params=gr.JSON()
 
-    def echo_fn(inp):
-        return inp 
-    def check_load(inp_url,outp_vid,hid_box,start_f,end_f):
-        if outp_vid == None and inp_url !="":
-            out_trim,in_vid,trim_count=trim_vid(hid_box,start_f,end_f)
-        elif outp_vid !=None and inp_url == "" and hid_box =="":
-            out_trim = None
-            in_vid=outp_vid
-            trim_count = ""
-        elif outp_vid !=None and inp_url == "" and hid_box !="":
-            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)