simplify template gallery

app_dialogue.py

@@ -75,7 +75,12 @@ PROCESSOR = AutoProcessor.from_pretrained(
 )
 
 BOT_AVATAR = "IDEFICS_logo.png"
-
+IMAGE_GALLERY_PATHS = [
+            f"example_images/{image_dir}/{ex_image}"
+            for image_dir in os.listdir("example_images")
+            for ex_image in os.listdir(f"example_images/{image_dir}")
+        ]
+random.shuffle(IMAGE_GALLERY_PATHS)
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger()
 
@@ -729,7 +734,7 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
                 info="Choose the type of gallery you want to see.",
             )
             template_gallery = gr.Gallery(
-                # value= value given by gallery_type_choice,
+                value=IMAGE_GALLERY_PATHS,
                 label="Templates Gallery",
                 allow_preview=False,
                 columns=6,
@@ -917,12 +922,12 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         queue=False,
     )
 
-    gallery_type_choice.change(
-        fn=choose_gallery,
-        inputs=[gallery_type_choice],
-        outputs=[template_gallery],
-        queue=False,
-    )
+    # gallery_type_choice.change(
+    #     fn=choose_gallery,
+    #     inputs=[gallery_type_choice],
+    #     outputs=[template_gallery],
+    #     queue=False,
+    # )
     template_gallery.select(
         fn=add_file_gallery,
         inputs=[template_gallery],
@@ -930,9 +935,9 @@ with gr.Blocks(title="AI Meme Generator", theme=gr.themes.Base(), css=css) as de
         queue=False,
     )
     demo.load(
-        fn=choose_gallery,
-        inputs=[gallery_type_choice],
-        outputs=[template_gallery],
+        # fn=choose_gallery,
+        # inputs=[gallery_type_choice],
+        # outputs=[template_gallery],
         queue=False,
     )
 demo.queue(max_size=40, api_open=False)