Avoid coord error with lacking ":".

app.py

@@ -90,7 +90,11 @@ def prompt_to_layout(user_prompt, intensity, fpath=None):
     output = tokenizer.batch_decode(output, skip_special_tokens=True)
     layout = output[0].split('[User prompt]')[1].split('[Layout] ')[1].split(', ')
     spaces = [txt.split(':')[0] for txt in layout]
-    coords = [txt.split(':')[1].rstrip() for txt in layout]
+    coords = []
+    for txt in layout:
+        if ':' in txt:
+            split_txt = txt.split(':')
+            coords.append(split_txt[1].rstrip())
     coordinates = [re.findall(regex, coord) for coord in coords]
     
     num_coords = []