Update app.py

app.py

@@ -53,16 +53,17 @@ regex = re.compile(".*?\((.*?)\)")
 def draw_polygons(polygons, colors, im_size=(512, 512), b_color="white", fpath=None):
 
     image = Image.new("RGBA", im_size, color="white")
-    draw = aggdraw.Draw(image)
+    #draw = aggdraw.Draw(image)
+    draw = ImageDraw.Draw(image)
 
     for poly, color, in zip(polygons, colors):
         #get initial polygon coordinates
         xy = poly.exterior.xy
         coords = np.dstack((xy[1], xy[0])).flatten()
         # draw it on canvas, with the appropriate colors
-        brush = aggdraw.Brush((0, 0, 0), opacity=255)
-        draw.polygon(coords, brush)
-        
+        #brush = aggdraw.Brush((0, 0, 0), opacity=255)
+        #draw.polygon(coords, brush)
+        draw.polygon(list(coords), fill=(0, 0, 0))
         
         #get inner polygon coordinates
         small_poly = poly.buffer(-1, resolution=32, cap_style=2, join_style=2, mitre_limit=5.0)
@@ -70,18 +71,21 @@ def draw_polygons(polygons, colors, im_size=(512, 512), b_color="white", fpath=N
             mycoordslist = [list(x.exterior.coords) for x in small_poly]
             for coord in mycoordslist:
                 coords = np.dstack((np.array(coord)[:,1], np.array(coord)[:, 0])).flatten()
-                brush2 = aggdraw.Brush((0, 0, 0), opacity=255)
-                draw.polygon(coords, brush2)
+                #brush2 = aggdraw.Brush((0, 0, 0), opacity=255)
+                #draw.polygon(coords, brush2)
+                draw.polygon(list(coords), fill=tuple(color)) 
         elif poly.geom_type == 'Polygon':
             #get inner polygon coordinates
             xy2 = small_poly.exterior.xy
             coords2 = np.dstack((xy2[1], xy2[0])).flatten()
             # draw it on canvas, with the appropriate colors
-            brush2 = aggdraw.Brush((color[0], color[1], color[2]), opacity=255)
-            draw.polygon(coords2, brush2)
-
-    image = Image.frombytes("RGBA", im_size, draw.tobytes()).transpose(Image.FLIP_TOP_BOTTOM)
+            #brush2 = aggdraw.Brush((color[0], color[1], color[2]), opacity=255)
+            #draw.polygon(coords2, brush2)
+            draw.polygon(list(coords2), fill=tuple(color))
 
+    #image = Image.frombytes("RGBA", im_size, draw.tobytes()).transpose(Image.FLIP_TOP_BOTTOM)
+    image = image.transpose(Image.FLIP_TOP_BOTTOM)
+    
     if(fpath):
         image.save(fpath, quality=100, subsampling=0)
 
@@ -255,11 +259,12 @@ textbox = gr.inputs.Textbox(placeholder='An apartment with two bedrooms and one
 generated = gr.outputs.Image(label='Generated Layout')
 
 iface = gr.Interface(fn=prompt_to_layout, inputs=[textbox, creative_slider], 
-                      outputs=[generated],
-                      css=custom_css,
-                      theme="default",
-                      allow_flagging=False,
-                      allow_screenshot=False,
-                      thumbnail="thumbnail_gradio.PNG")
+                     outputs=[generated],
+                     css=custom_css,
+                     theme="default",
+                     allow_flagging=False,
+                     allow_screenshot=False,
+                     thumbnail="thumbnail_gradio.PNG",
+                     enable_queue=True)
 
 iface.launch()