Update app.py

app.py

@@ -72,9 +72,12 @@ async def niji_api(prompt, progress=gr.Progress(), max_retries=5, backoff_factor
             return None
 
     def download_and_split_image(url):
-        img = fetch_image(url)
-        images = split_image(img)
-        return [save_image(i) for i in images]
+        try:
+            img = fetch_image(url)
+            images = split_image(img)
+            return [save_image(i) for i in images]
+        except Exception:
+            return [None]*4
 
     while prog < 100:
         try:
@@ -91,13 +94,7 @@ async def niji_api(prompt, progress=gr.Progress(), max_retries=5, backoff_factor
         prog = data.get('progress', 0)
         if progress_image_url := data.get('progressImageUrl'):
             iters = -100
-            result = None
-            try:
-                result = [(img, f"{prog}% done") for img in download_and_split_image(progress_image_url)]
-            except Exception:
-                pass
-            if result:
-                yield result
+            yield [(img, f"{prog}% done") for img in download_and_split_image(progress_image_url)]
 
         wait_time = random.uniform(1, 2) 
         await asyncio.sleep(wait_time)