#!/usr/bin/env python3
"""Test GoogleFlowImageProvider with Nano Banana Pro model."""
import logging
import sys
import os

sys.path.insert(0, os.path.dirname(__file__))

from dotenv import load_dotenv
load_dotenv()

logging.basicConfig(level=logging.INFO, format="%(levelname)s %(name)s: %(message)s")

from providers.google_flow_provider import GoogleFlowImageProvider

print("=" * 60)
print("Testing GoogleFlowImageProvider (Nano Banana Pro)")
print("=" * 60)

with GoogleFlowImageProvider() as provider:
    print(f"Model: {provider.model}")
    print(f"Aspect Ratio: {provider.aspect_ratio}")
    print()
    
    path = provider.generate(
        prompt="A cute orange cat sitting on a sunny windowsill, photorealistic, warm lighting, 4K",
        output_path="output/test_nano_banana.png",
    )
    print(f"\nImage saved: {path} ({path.stat().st_size} bytes)")