Spaces:

gaur3009
/

Model_gen_editor_100

Sleeping

App Files Files Community

gaur3009 commited on Jul 21

Commit

483cf17

•

1 Parent(s): 81ee834

Update app.py

Browse files

Files changed (1) hide show

app.py +73 -110

app.py CHANGED Viewed

@@ -1,120 +1,83 @@
 import gradio as gr
-from PIL import Image, ImageOps, ImageEnhance, ImageFilter, ImageDraw
-import json
-def edit_image(image, grayscale, flip, rotate, brightness, contrast, color, crop_data, resize, blur, sharpness, draw_text, text_position, text_color, text_size):
-    img = Image.open(image)
-    if grayscale:
-        img = ImageOps.grayscale(img)
-    if flip:
-        img = ImageOps.flip(img)
-    if rotate:
-        img = img.rotate(rotate)
-    # Apply brightness
-    enhancer = ImageEnhance.Brightness(img)
-    img = enhancer.enhance(brightness)
-    # Apply contrast
-    enhancer = ImageEnhance.Contrast(img)
-    img = enhancer.enhance(contrast)
-    # Apply color
-    enhancer = ImageEnhance.Color(img)
-    img = enhancer.enhance(color)
-    # Apply crop
-    if crop_data:
-        crop = json.loads(crop_data)
-        img = img.crop((crop['x'], crop['y'], crop['x'] + crop['width'], crop['y'] + crop['height']))
-    # Apply resize
     if resize:
-        width, height = map(int, resize.split(","))
-        img = img.resize((width, height))
-    # Apply blur
-    if blur > 0:
-        img = img.filter(ImageFilter.GaussianBlur(blur))
-    # Apply sharpness
-    enhancer = ImageEnhance.Sharpness(img)
-    img = enhancer.enhance(sharpness)
-    # Draw text
-    if draw_text:
-        draw = ImageDraw.Draw(img)
-        x, y = map(int, text_position.split(","))
-        draw.text((x, y), draw_text, fill=text_color, font=None, anchor=None, spacing=4, align="left")
-    return img
-html_content = """
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <link href="https://unpkg.com/cropperjs/dist/cropper.css" rel="stylesheet">
-</head>
-<body>
-    <div>
-        <img id="image" style="max-width: 100%;">
-    </div>
-    <script src="https://unpkg.com/cropperjs"></script>
-    <script>
-        const image = document.getElementById('image');
-        const cropper = new Cropper(image, {
-            aspectRatio: NaN,
-            viewMode: 1,
-            autoCropArea: 1,
-            crop(event) {
-                const cropData = {
-                    x: event.detail.x,
-                    y: event.detail.y,
-                    width: event.detail.width,
-                    height: event.detail.height
-                };
-                document.querySelector('textarea[name="data"]').value = JSON.stringify(cropData);
-            }
-        });
-        // Function to load image into cropper
-        window.loadImage = function (src) {
-            image.src = src;
-            cropper.replace(src);
-        }
-    </script>
-</body>
-</html>
-"""
 interface = gr.Interface(
-    fn=edit_image,
     inputs=[
-        gr.Image(type="filepath", label="Upload Image"),
-        gr.Checkbox(label="Grayscale"),
-        gr.Checkbox(label="Flip Vertically"),
-        gr.Slider(minimum=0, maximum=360, step=1, value=0, label="Rotate Angle"),
-        gr.Slider(minimum=0.1, maximum=2, step=0.1, value=1, label="Brightness"),
-        gr.Slider(minimum=0.1, maximum=2, step=0.1, value=1, label="Contrast"),
-        gr.Slider(minimum=0.1, maximum=2, step=0.1, value=1, label="Color"),
-        gr.Textbox(label="Crop Data (JSON)", placeholder="{'x': 100, 'y': 100, 'width': 400, 'height': 400}"),
-        gr.Textbox(label="Resize (width, height)", placeholder="e.g., 800,600"),
-        gr.Slider(minimum=0, maximum=10, step=0.1, value=0, label="Blur"),
-        gr.Slider(minimum=0.1, maximum=2, step=0.1, value=1, label="Sharpness"),
-        gr.Textbox(label="Draw Text", placeholder="e.g., Hello World"),
-        gr.Textbox(label="Text Position (x, y)", placeholder="e.g., 100,100"),
-        gr.ColorPicker(label="Text Color"),
-        gr.Slider(minimum=10, maximum=100, step=1, value=30, label="Text Size")
     ],
-    outputs=gr.Image(),
-    live=True,
-    title="Advanced Image Editor",
-    description="Upload an image and apply various transformations including brightness, contrast, color adjustments, cropping, resizing, blurring, and adding text.",
 )
-# Adding custom HTML to the Gradio interface
-with gr.Blocks() as demo:
-    gr.HTML(html_content)
-    interface.render()
-demo.launch()

 import gradio as gr
+from PIL import Image, ImageOps, ImageDraw, ImageFont
+import numpy as np
+def apply_color(image, color):
+    color = Image.new('RGB', image.size, color)
+    colored_image = Image.blend(image.convert('RGB'), color, alpha=0.5)
+    return colored_image
+def add_design(image, design, position):
+    if design:
+        image.paste(design, position, design)
+    return image
+def resize_image(image, size):
+    if size:
+        return image.resize(size)
+    return image
+def rotate_image(image, angle):
+    if angle:
+        return image.rotate(angle, expand=True)
+    return image
+def crop_image(image, crop_coords):
+    if crop_coords:
+        return image.crop(crop_coords)
+    return image
+def add_text(image, text, position, font_size, font_color):
+    if text:
+        draw = ImageDraw.Draw(image)
+        font = ImageFont.load_default()
+        draw.text(position, text, fill=font_color, font=font)
+    return image
+def process_image(image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_coords):
+    if color:
+        image = apply_color(image, color)
+    if design:
+        image = add_design(image, design, design_position)
+    if text:
+        image = add_text(image, text, text_position, font_size, font_color)
     if resize:
+        image = resize_image(image, resize)
+    if rotate:
+        image = rotate_image(image, rotate)
+    if crop_coords:
+        image = crop_image(image, crop_coords)
+    return image
+def get_crop_coords(crop_points):
+    if crop_points:
+        x_coords = [point[0] for point in crop_points]
+        y_coords = [point[1] for point in crop_points]
+        return (min(x_coords), min(y_coords), max(x_coords), max(y_coords))
+    return None
+design_position = (100, 100)  # Default position for design
+text_position = (50, 50)      # Default position for text
 interface = gr.Interface(
+    fn=lambda image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_points: process_image(
+        image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, get_crop_coords(crop_points)
+    ),
     inputs=[
+        gr.Image(type="pil"),
+        gr.ColorPicker(label="Color"),
+        gr.Image(type="pil", optional=True, label="Design"),
+        gr.Point(label="Design Position"),
+        gr.Textbox(label="Text"),
+        gr.Point(label="Text Position"),
+        gr.Slider(10, 100, step=1, default=20, label="Font Size"),
+        gr.ColorPicker(label="Font Color"),
+        gr.Points(label="Resize", optional=True),
+        gr.Slider(0, 360, step=1, default=None, label="Rotate"),
+        gr.Points(label="Crop Area")
     ],
+    outputs=gr.Image(type="pil"),
+    live=True
 )
+interface.launch()