Retry 5 times.

app.py

@@ -287,7 +287,22 @@ examples = [
     ["three bedrooms with a kitchen adjacent to the dining room", "Medium"]
 ]
 
-iface = gr.Interface(fn=prompt_to_layout, inputs=[textbox, creative_slider], 
+
+def retry_prompt_to_layout(user_prompt, intensity, fpath=None):
+    max_attempts = 5
+    attempts = 0
+
+    while attempts < max_attempts:
+        try:
+            # Call the original function
+            result = prompt_to_layout(user_prompt, intensity, fpath)
+            return result
+        except Exception as e:
+            print(f"Attempt {attempts+1} failed with error: {e}")
+            attempts += 1
+
+
+iface = gr.Interface(fn=retry_prompt_to_layout, inputs=[textbox, creative_slider], 
                      outputs=[generated, layout],
                      css=custom_css,
                      theme="default",