Create app.py

app.py

@@ -0,0 +1,59 @@
+import gradio as gr
+import numpy as np
+from PIL import Image, ImageDraw
+
+def create_canvas(width, height, bg_color):
+    # Create a new image with the specified size and background color
+    img = Image.new('RGB', (width, height), color=bg_color)
+    return np.array(img)
+
+def add_shape(image, shape_type, x, y, size, color):
+    img = Image.fromarray(image)
+    draw = ImageDraw.Draw(img)
+    
+    if shape_type == "square":
+        draw.rectangle([x, y, x+size, y+size], fill=color)
+    elif shape_type == "circle":
+        draw.ellipse([x, y, x+size, y+size], fill=color)
+    elif shape_type == "rectangle":
+        draw.rectangle([x, y, x+size, y+size//2], fill=color)
+    
+    return np.array(img)
+
+def update_canvas(image, width, height, bg_color, shape_type, x, y, size, color):
+    # Create new canvas if size changed
+    if image.shape[0] != height or image.shape[1] != width:
+        image = create_canvas(width, height, bg_color)
+    
+    # Add shape to the canvas
+    image = add_shape(image, shape_type, x, y, size, color)
+    
+    return image
+
+# Define the Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# Image Generator")
+    
+    with gr.Row():
+        width = gr.Slider(256, 1024, 256, step=1, label="Canvas Width")
+        height = gr.Slider(256, 1024, 256, step=1, label="Canvas Height")
+        bg_color = gr.ColorPicker(label="Background Color", value="#ffffff")
+    
+    with gr.Row():
+        shape_type = gr.Dropdown(["square", "circle", "rectangle"], label="Shape Type")
+        x = gr.Slider(0, 1000, 0, step=1, label="X Position")
+        y = gr.Slider(0, 1000, 0, step=1, label="Y Position")
+        size = gr.Slider(10, 200, 50, step=1, label="Shape Size")
+        color = gr.ColorPicker(label="Shape Color")
+    
+    canvas = gr.Image(shape=(256, 256), image_mode="RGB")
+    
+    add_btn = gr.Button("Add Shape")
+    
+    add_btn.click(
+        update_canvas,
+        inputs=[canvas, width, height, bg_color, shape_type, x, y, size, color],
+        outputs=canvas
+    )
+
+demo.launch()