Update app.py

app.py

@@ -1,13 +1,29 @@
 import gradio as gr
 import requests
 import numpy as np
+from time import sleep
 
 
 def run(prompt, token):
-    result = requests.post(
+    job = requests.post(
         "https://api.workranked.com/ai-web-design",
         data={"prompt": prompt, "token": token},
-    ).json()
+    ).json()["id"]
+
+    result = {}
+    for attempt in range(0, 12):
+        gr.Info(f"Waiting for results (attempt {attempt + 1})")
+        sleep(10)
+
+        result = requests.post(
+            f"https://api.workranked.com/ai-web-design-result",
+            data={"job": job},
+        ).json()
+        if result.get("output"):
+            break
+
+    if not result.get("output"):
+        gr.Error("Sorry, your request timed out. Please try again later...")
 
     images = result.get("data", {}).get("images")
     image = np.array(images[0]) if images else None
@@ -54,6 +70,7 @@ Have an idea or suggestion? Provide feedback directly to me (the developer) at [
         ],
         [
             "1920px, award-winning, Best Mobile User Experience\npurple background, bold sans-serif text, minimalist, two-column layout, app store buttons, centered elements, playful, modern, clean design, colors: pastel purple, white, black\ntemplates, marketplace, pricing, download, connect socials, monetize, audience building, analytics, creators, influencers, small businesses, e-commerce, mobile app, community",
+            "homepage, 1920px, award-winning, tech\nminimalist, playful, modern, clean design, bold sans-serif text, colors: pastel purple, white, black\nanalytics, small businesses, e-commerce, marketplace, audience building, pricing, download, connect socials, monetize, creators, influencers, mobile app, community",
             None,
         ],
         [