Add Example UI

app.py

@@ -2,12 +2,14 @@ import os
 import random
 import sys
 import subprocess
-from typing import Sequence, Mapping, Any, Union
+import spaces
 import torch
 import gradio as gr
+
+from typing import Sequence, Mapping, Any, Union
+from examples_db import ZEN_EXAMPLES
 from PIL import Image, ImageChops
 from huggingface_hub import hf_hub_download
-import spaces
 
 # Setup ComfyUI if not already set up
 #if not os.path.exists("ComfyUI"):
@@ -418,17 +420,24 @@ with gr.Blocks(css="footer {visibility: hidden}") as app:
                 
         with gr.Column(scale=1):
             output_image = gr.Image(label="Generated Image")
-            
-    # gr.Examples(
-    #     examples=[
-    #         ["A beautiful landscape with mountains and a lake", "examples/structure1.jpg", "examples/style1.jpg", 20, 0.4, 0.6, 30],
-    #         ["A cyberpunk cityscape at night", "examples/structure2.jpg", "examples/style2.jpg", 15, 0.35, 0.7, 28],
-    #     ],
-    #     inputs=[prompt_input, structure_image, style_image, depth_strength, canny_strength, style_strength, steps],
-    #     outputs=output_image,
-    #     fn=generate_image,
-    #     cache_examples=True
-    # )
+
+    gr.Examples(
+        examples=ZEN_EXAMPLES,
+        inputs=[
+            prompt_input,
+            structure_image,
+            style_image,
+            output_image,
+            depth_strength,
+            canny_strength,
+            style_strength,
+            steps,
+        ],
+        fn=generate_image,
+        cache_examples=True,
+        label="Presets",
+        examples_per_page=6
+    )
 
     generate_btn.click(
         fn=generate_image,