#!/usr/bin/env python3
"""
TEST: First 5 images to green screen
"""
import os
from pathlib import Path
from rembg import remove
from PIL import Image
# Find first 5 images
all_images = []
for dir_path in ["assets/PHASE_PACKS/0_DEMO", "assets/PHASE_PACKS/1_FAZA_1"]:
if Path(dir_path).exists():
all_images.extend(Path(dir_path).rglob("*.png"))
# Take first 5
test_images = sorted([str(p) for p in all_images if '_backup' not in str(p)])[:5]
OUTPUT_DIR = "test_5_green"
CHROMA_GREEN = (0, 255, 0, 255)
os.makedirs(OUTPUT_DIR, exist_ok=True)
print("🟢 TEST: First 5 images → Green Screen")
print("=" * 50)
results = []
for i, img_path in enumerate(test_images):
name = os.path.basename(img_path)
print(f"\n[{i+1}/5] Processing: {name}")
# Open and process
img = Image.open(img_path)
img_no_bg = remove(img)
# Create green background version
green_bg = Image.new('RGBA', img_no_bg.size, CHROMA_GREEN)
green_bg.paste(img_no_bg, (0, 0), img_no_bg)
green_rgb = green_bg.convert('RGB')
# Save
out_name = f"{i+1}_{name}"
green_rgb.save(os.path.join(OUTPUT_DIR, out_name), 'PNG')
results.append({'name': name, 'out': out_name, 'path': img_path})
print(f" ✅ Saved: {out_name}")
# Create simple HTML
html = '''
Test 5 Green
🟢 Test: First 5 Images with Green Background
'''
for r in results:
html += f'''
{r['name']}
'''
html += '''
'''
with open(os.path.join(OUTPUT_DIR, "index.html"), 'w') as f:
f.write(html)
print("\n" + "=" * 50)
print(f"✅ Done! Open: {OUTPUT_DIR}/index.html")