Update app.py

app.py

@@ -7,7 +7,6 @@ from PIL import ImageFont
 import gradio as gr
 from datasets import load_dataset
 
-dataset = load_dataset("chansung/testds")
 print(dataset)
 
 images = []
@@ -15,21 +14,23 @@ titles = []
 stories = []
 
 def add_title(image, title):
-    dr = ImageDraw.Draw(image)
+  dr = ImageDraw.Draw(image)
     
-    myFont = ImageFont.truetype('arial_bold.ttf', 30)
-    _, _, w, h = dr.textbbox((0, 0), title, font=myFont)
-    dr.rectangle([((image.width-w)/2, image.height-80), (((image.width-w)/2)+w, (image.height-80)+h)], fill="white", outline="white")
-    dr.text(((image.width-w)/2, image.height-80), title, font=myFont, fill=(0, 0, 0))
+  myFont = ImageFont.truetype('arial_bold.ttf', 30)
+  _, _, w, h = dr.textbbox((0, 0), title, font=myFont)
+  dr.rectangle([(0, image.height-80), (image.width, (image.height-80)+h)], fill="white", outline="white")
+  dr.text(((image.width-w)/2, image.height-80), title, font=myFont, fill=(0, 0, 0))
 
-    return image
+  return image
 
 def gallery_select(gallery, evt: gr.SelectData):
-    print(evt.value)
-    print(evt.index)
-    print(evt.target)
+  print(evt.value)
+  print(evt.index)
+  print(evt.target)
 
-with gr.Blocks() as demo:
+def get_gallery():
+  dataset = load_dataset("chansung/testds")
+    
   for row in dataset['train']:
     base64_image = row['image']
     base64_decoded = base64.b64decode(base64_image)
@@ -38,8 +39,11 @@ with gr.Blocks() as demo:
     titles.append(row['title'])
     stories.append(row['story'])
     images.append(add_title(image, row['title']))
-      
-  gallery = gr.Gallery(images, columns=10, height=500, container=False)
+
+  return images
+
+with gr.Blocks() as demo:
+  gallery = gr.Gallery(get_gallery, columns=10, height=500, container=False)
 
   gallery.select(
       fn=gallery_select,