test_import.py

"""Quick test to verify app imports correctly."""
import sys
print(f"Python: {sys.version}")

print("Testing import of gradio app...")
try:
    from seg_app.ui.gradio_app import demo
    print(f"✓ Gradio app imported: {type(demo)}")
except Exception as e:
    print(f"✗ Import failed: {e}")
    sys.exit(1)

print("\nTesting import of orchestrator...")
try:
    from seg_app.inference.orchestrator import get_available_models
    models = get_available_models()
    print(f"✓ Available models: {[m['id'] for m in models]}")
except Exception as e:
    print(f"✗ Import failed: {e}")
    sys.exit(1)

print("\nTesting model registry (without loading weights)...")
try:
    from seg_app.inference.model_registry import list_registered_models
    registered = list_registered_models()
    print(f"✓ Registered models: {registered}")
except Exception as e:
    print(f"✗ Import failed: {e}")
    sys.exit(1)

print("\n✓ All imports successful! Ready for deployment.")