Update app.py

app.py

@@ -82,18 +82,21 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
                 except Exception:
                     mes="<center>Some blocks returned an error"
             if out_box:
-                im_roll = get_concat_v_cut(f'{out_box[0]}',f'{out_box[1]}')
-                im_roll.save(f'comb-{uid}-tmp.png')                
-                for i in range(1,len(out_box)-2):
-                    im_roll = get_concat_v_cut(f'comb-{uid}-tmp.png',f'{out_box[i+1]}')
-                    im_roll.save(f'comb-{uid}-tmp.png')
-
-                
+                if len(out_box)>1:
+                    im_roll = get_concat_v_cut(f'{out_box[0]}',f'{out_box[1]}')
+                    im_roll.save(f'comb-{uid}-tmp.png')                
+                    for i in range(1,len(out_box)-2):
+                        im_roll = get_concat_v_cut(f'comb-{uid}-tmp.png',f'{out_box[i+1]}')
+                        im_roll.save(f'comb-{uid}-tmp.png')
+                        out = f'comb-{uid}-tmp.png'
+                else:
+                    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')
                     head_paste.save(f'comb-{uid}-tmp.png')
                     out = f'comb-{uid}-tmp.png'
-            
+