Update app.py

app.py

@@ -16,7 +16,7 @@ def download_image(prompt, index):
 def add_text_to_image(image_path, text):
     img = Image.open(image_path)
     draw = ImageDraw.Draw(img)
-    font = ImageFont.load_default()  # You can load a custom font if you want
+    font = ImageFont.load_default()  # You can load a custom font if needed
     # Wrap text for multiple lines
     lines = textwrap.fill(text, width=40)
     draw.text((10, 10), lines, font=font, fill="white")
@@ -39,7 +39,8 @@ def visualize_story_lines(story):
 
 # Gradio interface
 def visualize_story_images(story):
-    return visualize_story_lines(story)
+    images = visualize_story_lines(story)
+    return images[0]  # Return first image for now
 
 iface = gr.Interface(fn=visualize_story_images, inputs="text", outputs="image")
-iface.launch()
+iface.launch(share=True)