Update app.py

app.py

@@ -109,9 +109,9 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
     
                 colorize = RGBTransform().mix_with(color,factor=float(tint)).applied_to(raw)    
                 print (colorize)
-                colorize.save(f'tmp-{uid}.png')
+                colorize.save(f'tmp-{cnt}-{uid}.png')
                 #out_box.append(f'tmp-{uid}.png')           
-                out_box.append(f'tmp-{uid}.png') 
+                out_box.append(f'tmp-{cnt}-{uid}.png') 
 
                 print(f'out_box:: {out_box}')
                 if out_box:
@@ -119,16 +119,18 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                         #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(2,len(out_box)):
-                            #im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'{out_box[i]}')
-                            im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{uid}.png')
+                            im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'{out_box[i]}')
+                            #im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{uid}.png')
                             im_roll.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'
+                        yield out,out_html
+                        
                     else:
                         tmp_im = Image.open(out_box[0])
                         #tmp_im = out_box[0]
                         tmp_im.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'                   
-                    yield out,out_html
+                        yield out,out_html
             
             
             except Exception as e: