Update app.py

app.py

@@ -90,7 +90,24 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     spaces = [txt.split(':')[0] for txt in layout]
     coords = [txt.split(':')[1].rstrip() for txt in layout]
     coordinates = [re.findall(regex, coord) for coord in coords]
-    layout_dict = dict(zip(spaces, coords))
+    
+    num_coords = []
+    for coord in coordinates:
+        temp = []
+        for xy in coord:
+            numbers = xy.split(',')
+            temp.append(tuple([int(num)/14.2 for num in numbers]))
+        num_coords.append(temp)
+        
+    new_spaces = []
+    for i, v in enumerate(spaces):
+        totalcount = spaces.count(v)
+        count = spaces[:i].count(v)
+        new_spaces.append(v + str(count + 1) if totalcount > 1 else v)
+        
+    out_dict = dict(zip(new_spaces, num_coords))
+    out_dict = json.dumps(out_dict)
+       
     polygons = []
     for coord in coordinates:
         polygons.append([point.split(',') for point in coord])  
@@ -104,7 +121,7 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     legend = Image.open("labels.png")
     imgs_comb = np.vstack([im, legend])
     imgs_comb = Image.fromarray(imgs_comb)
-    return imgs_comb, layout_dict
+    return imgs_comb, out_dict 
      
     
 # Gradio App
@@ -137,6 +154,10 @@ custom_css="""
   max-width: 1500px;
 }
 
+.gradio_interface .panel:nth-child(2) .component:nth-child(3) { 
+    display:none
+}
+
 .gradio_wrapper .gradio_bg[theme=default] .panel_buttons {
   justify-content: flex-end;
 }