| import os
|
| import io
|
| import sys
|
| import traceback
|
| from PIL import Image
|
| from cora_engine import CoraEngine
|
| from dotenv import load_dotenv
|
|
|
| def test_engine_directly():
|
| load_dotenv()
|
| print("Initializing CoraEngine...")
|
| try:
|
| engine = CoraEngine()
|
| except Exception:
|
| print("Failed to init engine:")
|
| traceback.print_exc()
|
| return
|
|
|
| prompt = "A Roman gladiator in the Colosseum, fine art"
|
| print(f"Generating for: '{prompt}'...")
|
|
|
| try:
|
| result = engine.generate_from_text(prompt)
|
| print(f"Generation result type: {type(result)}")
|
|
|
| if result:
|
| print(f"Image Mode: {result.mode}")
|
| print(f"Image Size: {result.size}")
|
| print(f"Image Format: {result.format}")
|
|
|
| img_bytes = io.BytesIO()
|
| result.save(img_bytes, format="PNG")
|
| print(f"✅ Successfully saved result to PNG. Size: {len(img_bytes.getvalue())} bytes")
|
|
|
|
|
| try:
|
| img_bytes.seek(0)
|
| re_opened = Image.open(img_bytes)
|
| re_opened.verify()
|
| print("✅ Buffer verification successful.")
|
| except Exception as inner_e:
|
| print(f"❌ BUFFER VERIFICATION FAILED: {inner_e}")
|
|
|
| except Exception as e:
|
| print("\n❌ CATCHING ERROR:")
|
| print(f"Error Message: {e}")
|
| print("\nFULL TRACEBACK:")
|
| traceback.print_exc()
|
|
|
| if __name__ == "__main__":
|
|
|
| sys.path.append(os.getcwd())
|
| test_engine_directly()
|
|
|
