Create app.py

app.py

@@ -0,0 +1,55 @@
+import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image, ImageOps
+
+# Initialize global variables
+design_image = None
+target_image = None
+
+def load_images(design_path, target_path):
+    global design_image, target_image
+    design_image = Image.open(design_path).convert("RGBA")
+    target_image = Image.open(target_path).convert("RGBA")
+    return design_image, target_image
+
+def overlay_images(drag_info):
+    if not drag_info or not design_image or not target_image:
+        return target_image
+    
+    # Extract drag position and scale
+    x = drag_info['x']
+    y = drag_info['y']
+    scale = drag_info['scale']
+    
+    # Resize the design image
+    resized_design = design_image.resize(
+        (int(design_image.width * scale), int(design_image.height * scale)),
+        Image.ANTIALIAS
+    )
+    
+    # Overlay design onto target
+    overlay = Image.new('RGBA', target_image.size, (255, 255, 255, 0))
+    overlay.paste(resized_design, (x, y), resized_design)
+    combined_image = Image.alpha_composite(target_image, overlay)
+    
+    return combined_image
+
+# Gradio Interface
+with gr.Blocks() as editor:
+    gr.Markdown("## Drag-and-Drop Image Editor")
+    
+    with gr.Row():
+        with gr.Column():
+            design_path = gr.Image(label="Design Image")
+            target_path = gr.Image(label="Target Image")
+            load_btn = gr.Button("Load Images")
+        with gr.Column():
+            drag_info = gr.JSON(label="Drag Info", value={"x": 0, "y": 0, "scale": 1.0})
+            drag_area = gr.Canvas(tool="drag", label="Drag Design Here", width=600, height=400)
+            overlay_result = gr.Image(label="Overlay Result")
+    
+    load_btn.click(load_images, inputs=[design_path, target_path], outputs=[drag_info, drag_area])
+    drag_area.edit(overlay_images, inputs=drag_info, outputs=overlay_result)
+
+editor.launch()