Update app.py

app.py

@@ -32,17 +32,16 @@ grid_wide=10
 
 
 
-def get_concat_h_cut(in1, in2):
+def get_concat_h_cut(in1, in2, adj_h):
     print(in1)
     print(in2)
     im1=Image.open(in1)
     im2=Image.open(in2)
     #im1=in1
     #im2=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))
+    dst = Image.new('RGB', (im1.width + im2.width, min(im1.height, im2.height)))
+    dst.paste(im1, (0, int(adj_h)))
+    dst.paste(im2, (im1.width, int(adj_h)))
     return dst
 
 
@@ -72,7 +71,7 @@ def load_model(model_drop):
 
 def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
     uid=uuid.uuid4()
-
+    model=loaded_model[int(model_drop)]
     p_seed=""
     out_box=[]
     out_html=""
@@ -80,6 +79,7 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
     im_width=int(im_width)
     #for i,ea in enumerate(im_path.root):
     cnt = 0
+    adj_h=0
     for hh in range(int(im_height/grid_wide)):
         for b in range(int(im_width/grid_wide)):
         
@@ -100,7 +100,7 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                 p_seed+=" "
             try:
                 #model=gr.load(f'models/{model[int(model_drop)]}')        
-                model=loaded_model[int(model_drop)]
+
                 out_img=model(out_prompt+p_seed)
                 #print(out_img)
     
@@ -119,8 +119,8 @@ 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[cnt]}')
-                        #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[cnt]}')
+                        im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{cnt}-{uid}.png',adj_h=adj_h)
                         im_roll.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'
                         yield gr.Image(out),out_html
@@ -130,7 +130,7 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                         #tmp_im = out_box[0]
                         tmp_im.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'                   
-                        yield out,out_html
+                        yield gr.Image(out),out_html
             
             
             except Exception as e:
@@ -139,8 +139,8 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                 pass
  
             cnt+=1
-       
-        yield out,out_html
+        adj_h+=grid_wide   
+        yield gr.Image(out),out_html