Files
novafarma/scripts/utils/add_greenscreen_kai.py
2026-01-25 12:20:50 +01:00

43 lines
1.2 KiB
Python

#!/usr/bin/env python3
"""
Add Green Screen Background to Kai
Adds alpha 24 green (#00FF00) background to transparent image
"""
from PIL import Image
import os
# Input/Output
input_file = "/Users/davidkotnik/repos/novafarma/assets/slike/glavna_referenca/kai1.png"
output_file = "/Users/davidkotnik/repos/novafarma/assets/slike/glavna_referenca/kai_greenscreen.png"
print("🎨 ADDING GREEN SCREEN BACKGROUND TO KAI\n")
# Load image
print(f"📂 Loading: {os.path.basename(input_file)}")
img = Image.open(input_file)
# Get size
width, height = img.size
print(f"📐 Size: {width}x{height}px")
# Create green background (Chroma Key Green #00FF00)
green_bg = Image.new('RGBA', (width, height), (0, 255, 0, 255))
# Composite: green background + original image on top
if img.mode == 'RGBA':
# Image has alpha channel - composite it on green
result = Image.alpha_composite(green_bg, img)
else:
# No alpha - convert and place on green
img_rgba = img.convert('RGBA')
result = Image.alpha_composite(green_bg, img_rgba)
# Save
result.save(output_file)
print(f"\n✅ SUCCESS!")
print(f"📁 Saved: {os.path.basename(output_file)}")
print(f"🎨 Background: Chroma Key Green (#00FF00)")
print(f"📍 Location: {output_file}")