Update app.py

app.py

@@ -54,8 +54,8 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
     tasks = [model_actor.generate_image.remote(prompt, f"Prompt {paragraph_number}") for paragraph_number, prompt in prompts]
     print("Tasks created for image generation.")
 
-    done_tasks, _ = await asyncio.wait([asyncio.create_task(ray.wait([task])) for task in tasks], return_when=asyncio.ALL_COMPLETED)
-    responses = [await ray.get(task) for task in done_tasks]
+    # Use asyncio to gather the results asynchronously
+    responses = await asyncio.gather(*[ray.get(task) for task in tasks])
     print("Responses received from image generation tasks.")
 
     images = {paragraph_number: response for (paragraph_number, _), response in zip(prompts, responses)}