Update app.py

app.py

@@ -1,77 +1,78 @@
 import gradio as gr
-import torch
-from torchvision import models, transforms
-from PIL import Image, ImageEnhance, ImageDraw
+from PIL import Image, ImageOps, ImageEnhance, ImageFilter, ImageDraw
 import numpy as np
 
-# Load pre-trained Faster R-CNN model with updated weights parameter
-model = models.detection.fasterrcnn_resnet50_fpn(weights=models.detection.FasterRCNN_ResNet50_FPN_Weights.COCO_V1)
-model.eval()
-
-# Image transformation
-transform = transforms.Compose([
-    transforms.ToTensor()
-])
-
-# Detection function
-def detect_dress(image):
-    image_tensor = transform(image).unsqueeze(0)
-    with torch.no_grad():
-        outputs = model(image_tensor)
+def edit_image(image, grayscale, flip, rotate, brightness, contrast, color, crop, resize, blur, sharpness, draw_text, text_position, text_color, text_size):
+    img = Image.open(image)
     
-    boxes = outputs[0]['boxes'].numpy()
-    labels = outputs[0]['labels'].numpy()
-    scores = outputs[0]['scores'].numpy()
+    if grayscale:
+        img = ImageOps.grayscale(img)
+    if flip:
+        img = ImageOps.flip(img)
+    if rotate:
+        img = img.rotate(rotate)
     
-    # Filter for dress class (e.g., class 1 in COCO for 'person' which might also work for clothing)
-    threshold = 0.5
-    dress_boxes = [box for box, label, score in zip(boxes, labels, scores) if label == 1 and score > threshold]
-
-    draw = ImageDraw.Draw(image)
-    for box in dress_boxes:
-        draw.rectangle(box.tolist(), outline="red", width=3)
+    # Apply brightness
+    enhancer = ImageEnhance.Brightness(img)
+    img = enhancer.enhance(brightness)
     
-    return image, dress_boxes
-
-# Image editing functions
-def crop_image(image, box):
-    return image.crop(box)
-
-def adjust_color(image, factor):
-    enhancer = ImageEnhance.Color(image)
-    return enhancer.enhance(factor)
-
-# Gradio interface function
-def process_image(image, edit_type, factor):
-    detected_image, boxes = detect_dress(image)
+    # Apply contrast
+    enhancer = ImageEnhance.Contrast(img)
+    img = enhancer.enhance(contrast)
     
-    if not boxes:
-        return detected_image, "No dresses detected."
-
-    if edit_type == "Crop":
-        box = boxes[0]
-        edited_image = crop_image(image, box)
-    elif edit_type == "Adjust Color":
-        edited_image = adjust_color(image, factor)
-    else:
-        edited_image = image
+    # Apply color
+    enhancer = ImageEnhance.Color(img)
+    img = enhancer.enhance(color)
+    
+    # Apply crop
+    if crop:
+        img = img.crop(crop)
+    
+    # Apply resize
+    if resize:
+        img = img.resize(resize)
+    
+    # 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)
+        draw.text(text_position, draw_text, fill=text_color, font=None, anchor=None, spacing=4, align="left")
+    
+    return img
 
-    return edited_image, "Edit applied."
+def get_crop_coordinates(crop_start, crop_end):
+    return (crop_start[0], crop_start[1], crop_end[0], crop_end[1])
 
-# Create Gradio interface
-iface = gr.Interface(
-    fn=process_image,
+interface = gr.Interface(
+    fn=edit_image,
     inputs=[
-        gr.Image(type="pil"),
-        gr.Radio(choices=["None", "Crop", "Adjust Color"]),
-        gr.Slider(0.5, 2.0, step=0.1, label="Factor")
-    ],
-    outputs=[
-        gr.Image(type="pil"),
-        gr.Textbox(label="Result")
+        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 (left, upper, right, lower)", placeholder="e.g., 100,100,400,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")
     ],
-    live=True
+    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."
 )
 
-# Launch the Gradio interface
-iface.launch()
+interface.launch()