#!/usr/bin/env python3
"""Test script for generate_topdown_asset tool."""
import asyncio
import json
import sys
import os
from pathlib import Path
# Add server directory to path
server_dir = Path(__file__).parent / "server"
sys.path.insert(0, str(server_dir))
# Set environment variable before importing
os.environ["BACKEND_TYPE"] = "comfyui"
# Import after setting env
import main
async def test_generate_topdown_asset():
"""Test the generate_topdown_asset tool with the provided payload."""
try:
print("Testing generate_topdown_asset with payload...", file=sys.stderr)
result = await main.generate_topdown_asset(
asset_type="prop",
prompt="ice crystal tower, frozen magical tower, blue glowing crystals, top-down view, game asset",
save_to_file=True,
seed=12348,
size=256
)
# Parse and pretty print the result
try:
parsed = json.loads(result)
print(json.dumps(parsed, indent=2))
except json.JSONDecodeError:
print("Raw result:", result)
except Exception as e:
print(f"Error during test: {e}", file=sys.stderr)
import traceback
traceback.print_exc(file=sys.stderr)
if __name__ == "__main__":
asyncio.run(test_generate_topdown_asset())
