Update app.py

app.py

@@ -32,7 +32,7 @@ grid_wide=10
 
 
 
-def get_concat_h_cut(in1, in2, pos_x,pos_y):
+def get_concat_h_cut_old(in1, in2, pos_x,pos_y):
     print(in1)
     print(in2)
     im1=Image.open(in1)
@@ -59,6 +59,15 @@ def get_concat_h_cut(in1, in2, pos_x,pos_y):
     dst.paste(im2, (im1.width, im2.height *pos_y))
     return dst
 
+def get_concat_h_cut(in1, in2):
+    print(in1)
+    print(in2)
+    im1=Image.open(in1)
+    im2=Image.open(in2)
+    dst = Image.new('RGB', (im1.width + im2.width, min(im1.height,im2.height)))
+    dst.paste(im1, (0, 0))
+    dst.paste(im2, (im1.width,0))
+    return dst
 
 def get_concat_v_cut(in1, in2):
     print(in1)
@@ -89,6 +98,7 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
     model=loaded_model[int(model_drop)]
     p_seed=""
     out_box=[]
+    big_box=[]
     out_html=""
     im_height=int(im_height)
     im_width=int(im_width)
@@ -135,25 +145,33 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                         #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[cnt]}')
-                        im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{cnt}-{uid}.png',b,hh)
-                        im_roll.save(f'comb-{uid}-tmp.png')
-                        out = f'comb-{uid}-tmp.png'
+                        im_roll = get_concat_h_cut(f'comb-{uid}-{hh}-tmp.png',f'tmp-{cnt}-{uid}.png')
+                        im_roll.save(f'comb-{uid}-{hh}-tmp.png')
+                        out = f'comb-{uid}-{hh}-tmp.png'
                         yield gr.Image(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'                   
+                        tmp_im.save(f'comb-{uid}-{hh}-tmp.png')
+                        out = f'comb-{uid}-{hh}-tmp.png'                  
                         yield gr.Image(out),out_html
             
-            
             except Exception as e:
                 print(e)
                 out_html=str(e)
                 pass
  
             cnt+=1
+        big_box.append(f'comb-{uid}-{hh}-tmp.png')
+
+        if big_box:
+            print(f'big_box:: {big_box}')
+            if len(big_box)>1:
+                im_roll2 = get_concat_v_cut(big_box[hh],big_box[hh-1])
+                im_roll2.save(f'comb-{uid}-{hh}-out.png')
+                out=[]
+                yield gr.Image(f'comb-{uid}-{hh}-out.png'),out_html
         adj_h+=1   
     yield gr.Image(out),out_html