from PIL import Image
import os

# Import the richer runner (with EXIF) from the Gradio wrapper
from app_gradio import run_process_with_exif

# Create a small synthetic test image (RGB gradient)
img = Image.new('RGB', (128, 128))
for y in range(128):
    for x in range(128):
        img.putpixel((x, y), (int(x*2), int(y*2), int((x+y)/2)))

# Call with explicit parameters to exercise many code paths
out_img, status, exif = run_process_with_exif(
    img,
    noise_std=0.02,
    clahe_clip=2.0,
    tile=8,
    cutoff=0.25,
    fstrength=0.9,
    awb=True,
    sim_camera=True,
    lut_file=None,
    lut_strength=0.1,
    awb_ref=None,
    fft_ref=None,
    seed=0,
    jpeg_cycles=1,
    jpeg_qmin=88,
    jpeg_qmax=96,
    vignette_strength=0.35,
    chroma_strength=1.2,
    iso_scale=1.0,
    read_noise=2.0,
    no_no_bayer=False,
    randomness=0.05,
    perturb=0.008,
    phase_perturb=0.08,
    radial_smooth=5,
    fft_mode="auto",
    fft_alpha=1.0,
    hot_pixel_prob=1e-6,
    banding_strength=0.0,
    motion_blur_kernel=1,
)

print('Status:', status)
print('EXIF (hex length):', len(exif) if exif else 0)
if out_img is not None:
    os.makedirs('test', exist_ok=True)
    out_path = os.path.join('test', 'test_gradio_output_debug.jpg')
    out_img.save(out_path)
    print('Wrote output to', out_path)
else:
    print('No output image generated')