added gallery

check_app.py

@@ -167,8 +167,10 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
     # Generate image
     image = pipe(**common_args).images[0]
     filepath = save_generated_image(image, model_name, prompt)
+    # Then, reload the gallery
+    images, load_message = load_images_from_directory(directory_path)
     print(f"Saved image to: {filepath}")
-    return seed, image
+    return seed, image, images
 
 @spaces.GPU(duration=170)
 def create_pipeline_logic(prompt_text, model_name, negative_prompt="",  seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=40,):
@@ -251,7 +253,7 @@ def main():
                     step=1,
                     value=40,
                 )
-
+        gallery : gr.Gallery = None
         for model_name, config in MODEL_CONFIGS.items():
             with gr.Tab(model_name):
                 button = gr.Button(f"Run {model_name}")
@@ -264,7 +266,16 @@ def main():
             width,
             height,
             guidance_scale,
-            num_inference_steps], outputs=[output, img])
+            num_inference_steps], outputs=[output, img, gallery])
+
+            gallery = gr.Gallery(
+            label="Image Gallery",
+            show_label=True,
+            columns=4,
+            rows=3,
+            height=600,
+            object_fit="contain"
+        )
 
     app.launch()
 
@@ -280,5 +291,26 @@ def save_generated_image(image, model_name, prompt):
     image.save(filepath)
     return filepath
 
+def load_images_from_directory(directory_path):
+    """
+    Load all images from the specified directory.
+    Returns a list of image file paths.
+    """
+    image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp'}
+    directory = Path(directory_path)
+    
+    if not directory.exists():
+        return [], f"Error: Directory '{directory_path}' does not exist"
+    
+    image_files = [
+        str(f) for f in directory.iterdir()
+        if f.suffix.lower() in image_extensions and f.is_file()
+    ]
+    
+    if not image_files:
+        return [], f"No images found in directory '{directory_path}'"
+    
+    return image_files, f"Found {len(image_files)} images"
+
 if __name__ == "__main__":
     main()