Update app.py

app.py

@@ -1,9 +1,7 @@
 import gradio as gr
-import torch
 from diffusers import AutoPipelineForText2Image
 from io import BytesIO
 from generate_propmts import generate_prompt
-from concurrent.futures import ThreadPoolExecutor
 import asyncio
 
 # Load the model once outside of the function
@@ -41,20 +39,11 @@ async def process_prompt(sentence_mapping, character_dict, selected_style):
         prompts.append((paragraph_number, prompt))
         print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
 
-    loop = asyncio.get_event_loop()
-    tasks = []
-    
-    with ThreadPoolExecutor() as executor:
-        for paragraph_number, prompt in prompts:
-            tasks.append(loop.run_in_executor(executor, generate_image, prompt))
+    tasks = [generate_image(prompt) for _, prompt in prompts]
 
-        for paragraph_number, task in zip(sentence_mapping.keys(), await asyncio.gather(*tasks)):
-            try:
-                image = task
-                if image:
-                    images[paragraph_number] = image
-            except Exception as e:
-                print(f"Error processing paragraph {paragraph_number}: {e}")
+    for (paragraph_number, _), image_bytes in zip(prompts, await asyncio.gather(*tasks)):
+        if image_bytes:
+            images[paragraph_number] = image_bytes
 
     return images