| from PIL import Image | |
| import io | |
| import os | |
| # Import the run_process function from the Gradio wrapper | |
| from app_gradio import run_process | |
| # 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))) | |
| out_img, status = run_process(img) | |
| print('Status:', status) | |
| if out_img is not None: | |
| out_path = '/tmp/test_gradio_output.jpg' | |
| out_img.save(out_path) | |
| print('Wrote output to', out_path) | |
| else: | |
| print('No output image generated') | |