Update app.py

app.py

@@ -25,7 +25,11 @@ def add_text_to_image(image_path, text):
     try:
         img = Image.open(image_path)
         draw = ImageDraw.Draw(img)
-        font = ImageFont.truetype("arial.ttf", 20)  # Adjust font and size as needed
+        # Use a default font if arial.ttf is not available
+        try:
+            font = ImageFont.truetype("arial.ttf", 20)
+        except IOError:
+            font = ImageFont.load_default()
         lines = textwrap.fill(text, width=40)
         draw.text((10, 10), lines, font=font, fill="white", stroke_width=2, stroke_fill="black")
         img_with_text = f'text_added_{os.path.basename(image_path)}'
@@ -71,7 +75,7 @@ with gr.Blocks() as iface:
         story_input = gr.Textbox(lines=10, placeholder="Enter your story here...", label="Story Input")
         visualize_button = gr.Button("Visualize Story")
     
-    gallery_output = gr.Gallery(label="Visualized Story", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
+    gallery_output = gr.Gallery(label="Visualized Story", show_label=False, elem_id="gallery", columns=2, rows=2, height="auto")
     
     visualize_button.click(
         fn=visualize_story_images,