increased batch size and resiliance

App/Generate/database/Model.py

@@ -257,9 +257,12 @@ class Scene(orm.Model):
         async with aiohttp.ClientSession() as session:
             image_generator = AsyncImageGenerator(session)
             for payload in self.image_prompts:
-                result = await image_generator.generate_image(payload)
-                status = await image_generator.fetch_image_status(result["id"])
-                self.images.extend(status["output"])
+                while True:
+                    result = await image_generator.generate_image(payload)
+                    status = await image_generator.fetch_image_status(result["id"])
+                    if status["output"]:
+                        self.images.extend(status["output"])
+                        break
 
 
 class Transition(orm.Model):

App/Generate/generatorRoutes.py

@@ -30,7 +30,7 @@ async def from_dict_generate(data: Story):
     await generate_assets(generated_story=generated_strory)
 
 
-async def generate_assets(generated_story: Story, batch_size=4, threeD=True):
+async def generate_assets(generated_story: Story, batch_size=10, threeD=True):
     x = await Project.objects.create(name=str(uuid.uuid4()))
 
     # Assuming generated_story.scenes is a list of scenes