Update app.py

app.py

@@ -77,10 +77,31 @@ def add_text_to_image(image_path, text):
     
     return [img for _, img in sorted(images_with_text)]
 
+def process_story_lines(story, progress=gr.Progress()):
+    lines = [line.strip() for line in story.split('\n') if line.strip()]
+    images_with_text = []
+    
+    def process_line(idx, line):
+        prompt = line.replace(" ", "_")
+        img_file = download_image(prompt, idx)
+        if img_file:
+            return add_text_to_image(img_file, line)
+        return None
+
+    with ThreadPoolExecutor(max_workers=5) as executor:
+        future_to_idx = {executor.submit(process_line, idx, line): idx for idx, line in enumerate(lines)}
+        for future in progress.tqdm(as_completed(future_to_idx), total=len(lines), desc="Processing images"):
+            idx = future_to_idx[future]
+            result = future.result()
+            if result:
+                images_with_text.append((idx, result))
+    
+    return [img for _, img in sorted(images_with_text)]
+
 def visualize_story_images(story):
     if not story.strip():
         return []
-    return visualize_story_lines(story)
+    return process_story_lines(story)
 
 with gr.Blocks() as iface:
     gr.Markdown("# Story Visualizer")