Update app.py

app.py

@@ -45,21 +45,6 @@ def save_generated_image(image, prompt):
     
     return filepath
 
-def load_generated_images():
-    if not os.path.exists(SAVE_DIR):
-        return []
-    
-    # Load all images from the directory
-    image_files = [os.path.join(SAVE_DIR, f) for f in os.listdir(SAVE_DIR) 
-                  if f.endswith(('.png', '.jpg', '.jpeg', '.webp'))]
-    # Sort by creation time (newest first)
-    image_files.sort(key=lambda x: os.path.getctime(x), reverse=True)
-    return image_files
-
-def load_predefined_images():
-    # Return empty list since we're not using predefined images
-    return []
-
 @spaces.GPU(duration=120)
 def inference(
     prompt: str,
@@ -89,8 +74,8 @@ def inference(
     # Save the generated image
     filepath = save_generated_image(image, prompt)
     
-    # Return the image, seed, and updated gallery
-    return image, seed, load_generated_images()
+    # Return the image and seed (gallery removed)
+    return image, seed
 
 
 examples = [
@@ -118,106 +103,80 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, analytics_enabled=False) as demo
     gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fginigen-cartoon.hf.space">
                <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fginigen-cartoon.hf.space&countColor=%23263759" />
                </a>""")
-
     
-    with gr.Tabs() as tabs:
-        with gr.Tab("Generation"):
-            with gr.Column(elem_id="col-container"):
-                with gr.Row():
-                    prompt = gr.Text(
-                        label="Prompt",
-                        show_label=False,
-                        max_lines=1,
-                        placeholder="Enter your prompt",
-                        container=False,
-                    )
-                    run_button = gr.Button("Run", scale=0)
-
-                # Modified to include the default image
-                result = gr.Image(
-                    label="Result",
+    with gr.Tab("Generation"):
+        with gr.Column(elem_id="col-container"):
+            with gr.Row():
+                prompt = gr.Text(
+                    label="Prompt",
                     show_label=False,
-                    value=DEFAULT_IMAGE_PATH  # Set the default image
-                )
-
-                with gr.Accordion("Advanced Settings", open=False):
-                    seed = gr.Slider(
-                        label="Seed",
-                        minimum=0,
-                        maximum=MAX_SEED,
-                        step=1,
-                        value=42,
-                    )
-                    randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-
-                    with gr.Row():
-                        width = gr.Slider(
-                            label="Width",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=1024,
-                        )
-                        height = gr.Slider(
-                            label="Height",
-                            minimum=256,
-                            maximum=MAX_IMAGE_SIZE,
-                            step=32,
-                            value=768,
-                        )
-
-                    with gr.Row():
-                        guidance_scale = gr.Slider(
-                            label="Guidance scale",
-                            minimum=0.0,
-                            maximum=10.0,
-                            step=0.1,
-                            value=3.5,
-                        )
-                        num_inference_steps = gr.Slider(
-                            label="Number of inference steps",
-                            minimum=1,
-                            maximum=50,
-                            step=1,
-                            value=30,
-                        )
-                        lora_scale = gr.Slider(
-                            label="LoRA scale",
-                            minimum=0.0,
-                            maximum=1.0,
-                            step=0.1,
-                            value=1.0,
-                        )
-
-                gr.Examples(
-                    examples=examples,
-                    inputs=[prompt],
-                    outputs=[result, seed],
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
                 )
+                run_button = gr.Button("Run", scale=0)
 
-        with gr.Tab("Gallery"):
-            gallery_header = gr.Markdown("### Generated Images Gallery")
-            generated_gallery = gr.Gallery(
-                label="Generated Images",
-                columns=6,
+            result = gr.Image(
+                label="Result",
                 show_label=False,
-                value=load_generated_images(),
-                elem_id="generated_gallery",
-                height="auto"
+                value=DEFAULT_IMAGE_PATH
             )
-            refresh_btn = gr.Button("🔄 Refresh Gallery")
 
+            with gr.Accordion("Advanced Settings", open=False):
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=42,
+                )
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
-    # Event handlers
-    def refresh_gallery():
-        return load_generated_images()
+                with gr.Row():
+                    width = gr.Slider(
+                        label="Width",
+                        minimum=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=1024,
+                    )
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=768,
+                    )
 
-    refresh_btn.click(
-        fn=refresh_gallery,
-        inputs=None,
-        outputs=generated_gallery,
-    )
+                with gr.Row():
+                    guidance_scale = gr.Slider(
+                        label="Guidance scale",
+                        minimum=0.0,
+                        maximum=10.0,
+                        step=0.1,
+                        value=3.5,
+                    )
+                    num_inference_steps = gr.Slider(
+                        label="Number of inference steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=30,
+                    )
+                    lora_scale = gr.Slider(
+                        label="LoRA scale",
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.1,
+                        value=1.0,
+                    )
 
+            gr.Examples(
+                examples=examples,
+                inputs=[prompt],
+                outputs=[result, seed],
+            )
+    
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=inference,
@@ -231,8 +190,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, analytics_enabled=False) as demo
             num_inference_steps,
             lora_scale,
         ],
-        outputs=[result, seed, generated_gallery],
+        outputs=[result, seed],
     )
 
 demo.queue()
-demo.launch()
+demo.launch()