| |
| """Quick test of Kie.ai Nano Banana Pro image generation.""" |
| import os |
| import sys |
|
|
| from pathlib import Path |
| sys.path.insert(0, str(Path(__file__).resolve().parent)) |
|
|
| from dotenv import load_dotenv |
| load_dotenv() |
|
|
| from app.kie_image import generate_image_sync |
|
|
| def main(): |
| print("Testing Kie.ai Nano Banana Pro...") |
| if not os.environ.get("KIE_API_KEY") and not os.environ.get("KIE_AI_API_KEY"): |
| print("ERROR: KIE_API_KEY not set in .env") |
| sys.exit(1) |
| url, err = generate_image_sync( |
| prompt="A minimalist product photo of a ceramic vase on a white background, soft lighting", |
| width=1024, |
| height=1024, |
| ) |
| if err: |
| print(f"FAILED: {err}") |
| sys.exit(1) |
| print(f"SUCCESS: {url}") |
|
|
| if __name__ == "__main__": |
| main() |
|
|
