Update app.py

app.py

@@ -5,12 +5,17 @@ import gradio as gr
 from PIL import Image, ImageFilter, ImageOps, ImageChops
 import torchvision.transforms as transforms
 import os
+import random
 import pathlib
 
 # --- ⚙️ Configuration ---
+# Create a directory to save output images
 output_dir = "outputs"
 os.makedirs(output_dir, exist_ok=True)
 
+# Define allowed image extensions for the file explorer
+IMAGE_EXTENSIONS = [".png", ".jpg", ".jpeg", ".bmp", ".gif", ".tiff"]
+
 # --- 🎨 Filters ---
 FILTERS = {
     "Standard": "📄", "Invert": "⚫⚪", "Blur": "🌫️", "Sharpen": "🔪", "Contour": "🗺️",
@@ -105,11 +110,13 @@ def apply_filter(line_img, filter_name, original_img):
         return Image.fromarray(noisy_array)
     return line_img
 
-# --- 🖼️ Main Processing Function ---
-def predict(input_img_path, line_style, filter_choice):
+# --- 🖼️ Main Processing Function (Updated) ---
+def process_image(input_img_path, line_style, filter_choice, gallery_state):
     if not model1 or not model2:
         raise gr.Error("Models are not loaded. Please check for 'model.pth' and 'model2.pth'.")
-    
+    if not input_img_path:
+        raise gr.Error("Please select an image from the file explorer first.")
+
     filter_name = filter_choice.split(" ", 1)[1]
     original_img = Image.open(input_img_path).convert('RGB')
     
@@ -128,50 +135,110 @@ def predict(input_img_path, line_style, filter_choice):
     
     final_image = apply_filter(line_drawing_full_res, filter_name, original_img)
     
-    # --- 💾 Save the output image ---
+    # --- 💾 Save the output and update gallery state ---
     base_name = pathlib.Path(input_img_path).stem
     output_filename = f"{base_name}_{filter_name}.png"
     output_filepath = os.path.join(output_dir, output_filename)
     final_image.save(output_filepath)
     
-    return final_image
+    # Add new image path to the beginning of the list
+    gallery_state.insert(0, output_filepath)
+
+    # Return the single latest image for the main output and the updated list for the gallery
+    return final_image, gallery_state
 
 # --- 🚀 Gradio UI Setup ---
 title = "🖌️ Image to Line Art with Creative Filters"
-description = "Upload an image, choose a line style (Complex or Simple), and select a filter from the dropdown to transform your picture. Results are saved in the 'outputs' folder."
-
-filter_choices = [f"{emoji} {name}" for name, emoji in FILTERS.items()]
-
-# --- ✅ New Curated Examples Section ---
-examples = []
-example_images = [f"{i:02d}.jpeg" for i in range(1, 11)]
-# A selection of 6 interesting filters to demonstrate
-demo_filters = ["🗺️ Contour", "🔵⚫ BlueOnBlack", "✖️ Multiply", "🏞️ Emboss", "🔪 Sharpen", "❄️ Noise"]
-
-# Create one example for each of the 10 image files, cycling through the demo filters
-for i, img_file in enumerate(example_images):
-    if os.path.exists(img_file):
-        # Use modulo to cycle through the 6 demo filters for the 10 images
-        chosen_filter = demo_filters[i % len(demo_filters)]
-        examples.append([img_file, 'Simple Lines', chosen_filter])
-
-if not examples:
-    print("⚠️ Warning: No example images ('01.jpeg' to '10.jpeg') found. Examples will be empty.")
-
-# Reverted to the simpler and more stable gr.Interface
-iface = gr.Interface(
-    fn=predict,
-    inputs=[
-        gr.Image(type='filepath', label="Upload Image"),
-        gr.Radio(['Complex Lines', 'Simple Lines'], label='Line Style', value='Simple Lines'),
-        gr.Dropdown(filter_choices, label="Filter", value=filter_choices[0])
-    ],
-    outputs=gr.Image(type="pil", label="Filtered Line Art"),
-    title=title,
-    description=description,
-    examples=examples,
-    allow_flagging='never'
-)
+description = "1. Browse and select an image using the file explorer. 2. Choose a line style. 3. Pick a filter. Your results will be saved to the 'outputs' folder and appear in the gallery below."
+
+# --- Dynamic Examples Generation ---
+def generate_examples():
+    example_images = [f"{i:02d}.jpeg" for i in range(1, 11)]
+    # Filter for only existing example images
+    valid_example_images = [img for img in example_images if os.path.exists(img)]
+    
+    if not valid_example_images:
+        print("⚠️ Warning: No example images ('01.jpeg' through '10.jpeg') found. Examples will be empty.")
+        return []
+
+    examples = []
+    for name, emoji in FILTERS.items():
+        filter_choice = f"{emoji} {name}"
+        random_image = random.choice(valid_example_images)
+        line_style = random.choice(['Simple Lines', 'Complex Lines'])
+        examples.append([random_image, line_style, filter_choice])
+    # Shuffle to make the order random on each load
+    random.shuffle(examples)
+    return examples
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown(f"<h1 style='text-align: center;'>{title}</h1>")
+    gr.Markdown(description)
+
+    # Stores the list of gallery image paths
+    gallery_state = gr.State(value=[])
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### 1. Select an Image")
+            # File explorer for a better user experience
+            input_image_path = gr.FileExplorer(
+                root=".",
+                glob=f"**/*[{''.join(ext[1:] for ext in IMAGE_EXTENSIONS)}]",
+                label="Browse Your Images",
+                height=400
+            )
+            gr.Markdown("### 2. Choose a Line Style")
+            line_style_radio = gr.Radio(
+                ['Complex Lines', 'Simple Lines'],
+                label="Line Style",
+                value='Simple Lines'
+            )
+        
+        with gr.Column(scale=3):
+            gr.Markdown("### 3. Pick a Filter")
+            filter_buttons = [gr.Button(value=f"{emoji} {name}") for name, emoji in FILTERS.items()]
+            
+            # Hidden radio to store the selected button's value
+            selected_filter = gr.Radio(
+                [b.value for b in filter_buttons],
+                label="Selected Filter",
+                visible=False,
+                value=filter_buttons[0].value
+            )
+
+            gr.Markdown("### 4. Result")
+            main_output_image = gr.Image(type="pil", label="Latest Result")
+    
+    with gr.Row():
+        gr.Markdown("---")
+        
+    with gr.Row():
+         # --- Dynamic Examples ---
+        gr.Examples(
+            examples=generate_examples(),
+            inputs=[input_image_path, line_style_radio, selected_filter],
+            label="✨ Click an Example to Start",
+            examples_per_page=10
+        )
+        
+    with gr.Row():
+        gr.Markdown("## 🖼️ Result Gallery (Saved in 'outputs' folder)")
+        gallery_output = gr.Gallery(label="Your Generated Images", height=600, columns=5)
+    
+    # --- Event Handling ---
+    def handle_filter_click(btn_value, current_path, style, state):
+        # When a filter button is clicked, it triggers the main processing function
+        new_main_img, new_state = process_image(current_path, style, btn_value, state)
+        # Update the hidden radio, the main image, and the gallery
+        return btn_value, new_main_img, new_state
+
+    for btn in filter_buttons:
+        btn.click(
+            fn=handle_filter_click,
+            inputs=[btn, input_image_path, line_style_radio, gallery_state],
+            outputs=[selected_filter, main_output_image, gallery_state]
+        )
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()