Update app.py

app.py

@@ -39,6 +39,7 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
     mes_box=[]
     out_box=[]
     uid=uuid.uuid4()
+    out=None
     is_url=True
     if is_url:
         options = webdriver.ChromeOptions()
@@ -90,7 +91,7 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
                         im_roll.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'
                 else:
-                    out_box[0].save(f'comb-{uid}-tmp.png')
+                    tmp_im = Image.open(out_box[0]).save(f'comb-{uid}-tmp.png')
                     out = f'comb-{uid}-tmp.png'
                 if check_h:
                     head_paste = get_concat_v_cut(f'head-{uid}-tmp.png',f'comb-{uid}-tmp.png')