Update app.py

app.py

@@ -1,8 +1,3 @@
-# With torch.nn as nn, we build the brain,
-# Gradio as gr, makes demos reign.
-# PIL's Image, Filter, Ops, and Chops,
-# transforms from torchvision, style never stops!
-### 🖥️ New and Improved Application Code
 import numpy as np
 import torch
 import torch.nn as nn
@@ -10,17 +5,12 @@ 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": "🗺️",
@@ -115,13 +105,11 @@ def apply_filter(line_img, filter_name, original_img):
         return Image.fromarray(noisy_array)
     return line_img
 
-# --- 🖼️ Main Processing Function (Updated) ---
-def process_image(input_img_path, line_style, filter_choice, gallery_state):
+# --- 🖼️ Main Processing Function ---
+def predict(input_img_path, line_style, filter_choice):
     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')
     
@@ -140,112 +128,50 @@ def process_image(input_img_path, line_style, filter_choice, gallery_state):
     
     final_image = apply_filter(line_drawing_full_res, filter_name, original_img)
     
-    # --- 💾 Save the output and update gallery state ---
+    # --- 💾 Save the output image ---
     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)
     
-    # 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
+    return final_image
 
 # --- 🚀 Gradio UI Setup ---
 title = "🖌️ Image to Line Art with Creative Filters"
-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")
-            # 
-            # ===== THIS IS THE CORRECTED LINE =====
-            # The 'root' argument has been removed to support older Gradio versions
-            #
-            input_image_path = gr.FileExplorer(
-                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]
-        )
+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'
+)
 
 if __name__ == "__main__":
-    demo.launch()
+    iface.launch()