Update app.py

app.py

@@ -76,6 +76,12 @@ def encode_image(image):
     buffered = BytesIO()
     #image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode()
+
+def fake_gan():
+    base_dir = os.getcwd()  # Get the current base directory
+    img_files = [file for file in os.listdir(base_dir) if file.lower().endswith((".png", ".jpg", ".jpeg"))]  # List all files ending with ".jpg" or ".jpeg"
+    images = [(random.choice(img_files), os.path.splitext(file)[0]) for file in img_files]
+    return images
     
 def predict(prompt, guidance, steps, seed=1231231):
     generator = torch.manual_seed(seed)
@@ -151,7 +157,25 @@ with gr.Blocks(css=css) as demo:
                 )
                 generate_bt = gr.Button("Generate", scale=1)
 
+        # Image Result from last prompt
         image = gr.Image(type="filepath")
+
+        # Gallery
+        with gr.Row(variant="compact"):
+            text = gr.Textbox(
+                label="Image Sets",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+            ).style(
+                container=False,
+            )
+            btn = gr.Button("Generate image").style(full_width=False)
+
+        gallery = gr.Gallery(
+            label="Generated images", show_label=False, elem_id="gallery"
+        ).style(grid=[2], height="auto")
+        
         with gr.Accordion("Advanced options", open=False):
             guidance = gr.Slider(
                 label="Guidance", minimum=0.0, maximum=5, value=0.3, step=0.001
@@ -186,6 +210,7 @@ with gr.Blocks(css=css) as demo:
 
         inputs = [prompt, guidance, steps, seed]
         generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
+        btn.click(fake_gan, None, gallery)
         prompt.input(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         guidance.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)
         steps.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)