Spaces:
Runtime error
Runtime error
| #!/usr/bin/env python3 | |
| """ | |
| Test script to verify deployment configuration | |
| """ | |
| import os | |
| import sys | |
| def test_local_mode(): | |
| """Test local mode configuration""" | |
| print("🧪 Testing LOCAL mode configuration...") | |
| # Set local mode | |
| os.environ["DEPLOYMENT_MODE"] = "local" | |
| try: | |
| from config import is_local_mode, is_modal_mode, get_cache_dir | |
| assert is_local_mode() == True, "Should be in local mode" | |
| assert is_modal_mode() == False, "Should not be in modal mode" | |
| cache_dir = get_cache_dir() | |
| assert "gradio_mcp_cache" in cache_dir, f"Cache dir should be local: {cache_dir}" | |
| print("✅ Local mode configuration OK") | |
| return True | |
| except Exception as e: | |
| print(f"❌ Local mode test failed: {e}") | |
| return False | |
| def test_modal_mode(): | |
| """Test modal mode configuration""" | |
| print("🧪 Testing MODAL mode configuration...") | |
| # Set modal mode | |
| os.environ["DEPLOYMENT_MODE"] = "modal" | |
| try: | |
| # Clear config module cache to reload with new env var | |
| if 'config' in sys.modules: | |
| del sys.modules['config'] | |
| from config import is_local_mode, is_modal_mode, get_cache_dir | |
| assert is_local_mode() == False, "Should not be in local mode" | |
| assert is_modal_mode() == True, "Should be in modal mode" | |
| cache_dir = get_cache_dir() | |
| assert cache_dir == "/root/cache", f"Cache dir should be modal: {cache_dir}" | |
| print("✅ Modal mode configuration OK") | |
| return True | |
| except Exception as e: | |
| print(f"❌ Modal mode test failed: {e}") | |
| return False | |
| def test_gpu_adapters(): | |
| """Test GPU adapters""" | |
| print("🧪 Testing GPU adapters...") | |
| try: | |
| from gpu_adapters import transcribe_audio_adaptive_sync | |
| # This should not crash, even if endpoints are not configured | |
| result = transcribe_audio_adaptive_sync( | |
| "test_file.mp3", | |
| "turbo", | |
| None, | |
| "srt", | |
| False | |
| ) | |
| # Should return error result but not crash | |
| assert "processing_status" in result or "error_message" in result, "Should return valid result structure" | |
| print("✅ GPU adapters OK") | |
| return True | |
| except Exception as e: | |
| print(f"❌ GPU adapters test failed: {e}") | |
| return False | |
| def test_imports(): | |
| """Test all imports work correctly""" | |
| print("🧪 Testing imports...") | |
| try: | |
| # Test config imports | |
| from config import DeploymentMode, get_deployment_mode | |
| # Test MCP tools imports | |
| from mcp_tools import get_mcp_server | |
| # Test app imports (should work in both modes) | |
| from app import create_app, main, get_app | |
| print("✅ All imports OK") | |
| return True | |
| except Exception as e: | |
| print(f"❌ Import test failed: {e}") | |
| return False | |
| def test_hf_spaces_mode(): | |
| """Test Hugging Face Spaces mode""" | |
| print("🧪 Testing HF Spaces mode...") | |
| try: | |
| # Clear deployment mode to simulate HF Spaces | |
| old_mode = os.environ.get("DEPLOYMENT_MODE") | |
| if "DEPLOYMENT_MODE" in os.environ: | |
| del os.environ["DEPLOYMENT_MODE"] | |
| # Clear config module cache | |
| if 'config' in sys.modules: | |
| del sys.modules['config'] | |
| if 'app' in sys.modules: | |
| del sys.modules['app'] | |
| from app import get_app | |
| app = get_app() | |
| assert app is not None, "Should create app for HF Spaces" | |
| # Restore environment | |
| if old_mode: | |
| os.environ["DEPLOYMENT_MODE"] = old_mode | |
| print("✅ HF Spaces mode OK") | |
| return True | |
| except Exception as e: | |
| print(f"❌ HF Spaces test failed: {e}") | |
| return False | |
| def main(): | |
| """Run all tests""" | |
| print("🚀 Running deployment configuration tests...\n") | |
| tests = [ | |
| test_imports, | |
| test_local_mode, | |
| test_modal_mode, | |
| test_hf_spaces_mode, | |
| test_gpu_adapters, | |
| ] | |
| passed = 0 | |
| total = len(tests) | |
| for test in tests: | |
| try: | |
| if test(): | |
| passed += 1 | |
| except Exception as e: | |
| print(f"❌ Test {test.__name__} crashed: {e}") | |
| print() | |
| print(f"📊 Test Results: {passed}/{total} passed") | |
| if passed == total: | |
| print("🎉 All tests passed! Deployment configuration is ready.") | |
| return 0 | |
| else: | |
| print("⚠️ Some tests failed. Please check the configuration.") | |
| return 1 | |
| if __name__ == "__main__": | |
| sys.exit(main()) |