Update app.py

app.py

@@ -67,56 +67,65 @@ def load_model(model_drop):
 
 
 def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
+    uid=uuid.uuid4()
     p_seed=""
     out_box=[]
     out_html=""
-    for i,ea in enumerate(im_path.root):
-        print(f'root::{im_path.root[i]}')
-        #print(f'ea:: {ea}')        
-        #print(f'impath:: {im_path.path}')
-        url = base_url+im_path.root[i].image.path
-        myimg = cv2.imread(im_path.root[i].image.path)
-        avg_color_per_row = numpy.average(myimg, axis=0)
-        avg_color = numpy.average(avg_color_per_row, axis=0)
-        #print(avg_color)
-        
-        #h=color.lstrip('#')
-        #h = input('Enter hex: ').lstrip('#')
-        #print('RGB =', tuple(int(h[i:i+2], 16) for i in (0, 2, 4)))
-        #color=tuple(int(h[i:i+2], 16) for i in (0, 2, 4))
-        r,g,b= avg_color
-        
-        color = (int(r),int(g),int(b))
-    
+    #for i,ea in enumerate(im_path.root):
+    cnt = 0
+    for hh in range(int(im_height/grid_wide)):
+        for b in range(int(im_width/grid_wide)):
         
-        print (color)
+            print(f'root::{im_path.root[cnt]}')
+            #print(f'ea:: {ea}')        
+            #print(f'impath:: {im_path.path}')
+            url = base_url+im_path.root[cnt].image.path
+            print(url)
 
-        #for i,ea in enumerate(loaded_model):
+            myimg = cv2.imread(im_path.root[cnt].image.path)
+            avg_color_per_row = numpy.average(myimg, axis=0)
+            avg_color = numpy.average(avg_color_per_row, axis=0)
+            r,g,b= avg_color
+            color = (int(r),int(g),int(b))
+            print (color)
+            rand=random.randint(1,500)
+            for i in range(rand):
+                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)
     
+                raw=Image.open(out_img)
+                raw=raw.convert('RGB')
     
-    
-        #for i in range(int(cnt)):
-        rand=random.randint(1,500)
-        for i in range(rand):
-            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)
-
-            raw=Image.open(out_img)
-            raw=raw.convert('RGB')
-
-            colorize = RGBTransform().mix_with(color,factor=float(tint)).applied_to(raw)    
+                colorize = RGBTransform().mix_with(color,factor=float(tint)).applied_to(raw)    
+                
+                out_box.append(colorize)           
+                if out_box:
+                    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(2,len(out_box)):
+                            im_roll = get_concat_v_cut(f'comb-{uid}-tmp.png',f'{out_box[i]}')
+                            im_roll.save(f'comb-{uid}-tmp.png')
+                        out = f'comb-{uid}-tmp.png'
+                    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'                   
             
-            out_box.append(colorize)           
-        except Exception as e:
-            print(e)
-            out_html=str(e)
-            pass
             
-        yield out_box,out_html
+            except Exception as e:
+                print(e)
+                out_html=str(e)
+                pass
+ 
+            cnt+=1
+       
+        yield out,out_html