Update app.py

app.py

@@ -1,90 +1,90 @@
-# app_colab.py
 import cv2
 import numpy as np
-import matplotlib.pyplot as plt
-from IPython.display import clear_output
-import imageio
+import gradio as gr
+import time
 
-def load_image(image_path, scale_percent=50):
-    """Load and resize image"""
-    img = cv2.imread(image_path)
-    if img is None:
-        raise FileNotFoundError(f"Image not found: {image_path}")
-    
-    width = int(img.shape[1] * scale_percent / 100)
-    height = int(img.shape[0] * scale_percent / 100)
-    img = cv2.resize(img, (width, height))
-    return img
-
-def pencil_sketch(img, blur_ksize=21):
-    """Convert image to pencil sketch"""
-    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    inv_gray = 255 - gray
-    blur = cv2.GaussianBlur(inv_gray, (blur_ksize, blur_ksize), 0)
-    sketch = cv2.divide(gray, 255 - blur, scale=256)
-    return sketch
+# ----------------------------
+# Sketch Conversion Functions
+# ----------------------------
 
-def add_texture(sketch, intensity=15):
-    """Add pencil texture to sketch"""
-    noise = np.random.normal(0, intensity, sketch.shape).astype(np.uint8)
-    sketch_textured = cv2.add(sketch, noise)
-    return sketch_textured
+def dodgeV2(image, mask):
+    """Helper function for pencil sketch effect"""
+    return cv2.divide(image, 255 - mask, scale=256)
 
-def animate_sketch(sketch, save_gif=False, gif_name="pencil_sketch_animation.gif", show_animation=True):
-    """Animate sketch line by line"""
-    canvas = np.ones_like(sketch) * 255
-    y_indices, x_indices = np.where(sketch < 255)
-    coords = list(zip(x_indices, y_indices))
-    coords.sort(key=lambda pt: sketch[pt[1], pt[0]])  # draw dark pixels first
+def pencil_sketch(img, scale=256, blur_value=21, texture=False):
+    """Convert image to realistic pencil sketch"""
+    # Convert to grayscale
+    gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     
-    frames = []
-    for i, (x, y) in enumerate(coords):
-        canvas[y, x] = sketch[y, x]
-        if i % 200 == 0:
-            if show_animation:
-                clear_output(wait=True)
-                plt.imshow(canvas, cmap='gray')
-                plt.axis('off')
-                plt.show()
-            if save_gif:
-                frames.append(canvas.copy())
+    # Invert image
+    inv_gray = 255 - gray_img
+    
+    # Apply Gaussian blur
+    blur_img = cv2.GaussianBlur(inv_gray, (blur_value, blur_value), 0)
+    
+    # Blend using dodge technique
+    sketch = dodgeV2(gray_img, blur_img)
     
-    # Final frame
-    if show_animation:
-        clear_output(wait=True)
-        plt.imshow(canvas, cmap='gray')
-        plt.axis('off')
-        plt.show()
+    # Optional: Add paper texture
+    if texture:
+        h, w = sketch.shape
+        paper = np.random.normal(loc=200, scale=30, size=(h, w)).astype(np.uint8)
+        sketch = cv2.multiply(sketch, paper, scale=1/255)
     
-    if save_gif:
-        imageio.mimsave(gif_name, frames, duration=0.05)
-        print(f"[INFO] GIF saved as {gif_name}")
+    return sketch
 
-    return canvas
+# ----------------------------
+# Animation Function
+# ----------------------------
 
-# -----------------------------
-# Colab Usage Example
-# -----------------------------
+def animate_sketch(image, scale=256, blur=21, texture=False, delay=0.01):
+    """Simulate hand-drawn line-by-line animation"""
+    img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
+    sketch = pencil_sketch(img, scale=scale, blur_value=blur, texture=texture)
+    
+    # Create blank canvas
+    h, w = sketch.shape
+    canvas = np.ones((h, w), dtype=np.uint8) * 255
+    
+    frames = []
+    for i in range(h):
+        canvas[i] = sketch[i]  # Draw line by line
+        frame_rgb = cv2.cvtColor(canvas, cv2.COLOR_GRAY2RGB)
+        frames.append(frame_rgb)
+    
+    return frames
 
-# 1️⃣ Upload your image in Colab:
-from google.colab import files
-uploaded = files.upload()
-image_path = list(uploaded.keys())[0]  # take the first uploaded file
+# ----------------------------
+# Gradio Interface
+# ----------------------------
 
-# 2️⃣ Load the image
-img = load_image(image_path, scale_percent=50)
-plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
-plt.axis('off')
-plt.title("Original Image")
-plt.show()
+def gradio_sketch(image, scale, blur, texture):
+    frames = animate_sketch(image, scale, blur, texture)
+    return frames  # Gradio will display as GIF automatically
 
-# 3️⃣ Convert to pencil sketch
-sketch = pencil_sketch(img, blur_ksize=21)
+# Gradio UI
+title = "🎨 Realistic Pencil Sketch Animation"
+description = """
+Upload any image and watch it transform into a realistic pencil sketch with hand-drawn animation!  
+You can adjust:
+- Scale: pencil intensity
+- Blur: smoothness of strokes
+- Texture: add paper texture
+"""
 
-# 4️⃣ Add pencil texture
-sketch_textured = add_texture(sketch, intensity=15)
+iface = gr.Interface(
+    fn=gradio_sketch,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Slider(64, 512, value=256, step=1, label="Scale"),
+        gr.Slider(1, 51, value=21, step=2, label="Blur"),
+        gr.Checkbox(label="Add Paper Texture?")
+    ],
+    outputs=gr.Gallery(label="Animated Sketch"),
+    title=title,
+    description=description,
+    allow_flagging="never",
+)
 
-# 5️⃣ Animate and save as GIF
-final_canvas = animate_sketch(sketch_textured, save_gif=True)
-cv2.imwrite("final_sketch.png", final_canvas)
-print("[INFO] Final sketch saved as final_sketch.png")
+if __name__ == "__main__":
+    iface.launch()