Create app.py

app.py

@@ -0,0 +1,64 @@
+import gradio as gr
+from PIL import Image
+import os
+
+def add_logo(main_image, logo_size_percent=20, margin_percent=2):
+    """
+    Add logo to the top-right corner of the main image
+    
+    Args:
+        main_image: PIL Image object or file path
+        logo_size_percent: Size of logo as percentage of main image width (default 20%)
+        margin_percent: Margin from edges as percentage of main image width (default 2%)
+    """
+    if main_image is None:
+        return None
+    
+    # Load the logo
+    logo_path = "logo.png"
+    if not os.path.exists(logo_path):
+        return main_image  # Return original if logo not found
+    
+    # Open images
+    logo = Image.open(logo_path).convert("RGBA")
+    
+    # Convert main image to RGBA to support transparency
+    main_img = main_image.convert("RGBA")
+    
+    # Calculate logo size (maintain aspect ratio)
+    logo_width = int(main_img.width * logo_size_percent / 100)
+    logo_height = int(logo.height * (logo_width / logo.width))
+    
+    # Resize logo
+    logo = logo.resize((logo_width, logo_height), Image.Resampling.LANCZOS)
+    
+    # Calculate position (top-right corner with margin)
+    margin = int(main_img.width * margin_percent / 100)
+    x_position = main_img.width - logo_width - margin
+    y_position = margin
+    
+    # Create a copy of the main image
+    result = main_img.copy()
+    
+    # Paste logo with transparency
+    result.paste(logo, (x_position, y_position), logo)
+    
+    # Convert back to RGB for saving/display
+    result = result.convert("RGB")
+    
+    return result
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=add_logo,
+    inputs=[
+        gr.Image(type="pil", label="Upload Main Image")
+    ],
+    outputs=gr.Image(type="pil", label="Image with Logo"),
+    title="Logo Adder",
+    description="Upload an image to add logo.png to the top-right corner",
+    examples=None
+)
+
+if __name__ == "__main__":
+    iface.launch()