| |
| """ |
| Test script for Sinhala Subtitle Hardcoder |
| Run this to test functionality before deploying to Hugging Face |
| """ |
|
|
| import os |
| import sys |
|
|
| def test_imports(): |
| """Test if all required packages are installed""" |
| print("Testing imports...") |
| try: |
| import gradio as gr |
| print("✅ Gradio imported successfully") |
| except ImportError: |
| print("❌ Gradio not found. Run: pip install gradio") |
| return False |
| |
| try: |
| import requests |
| print("✅ Requests imported successfully") |
| except ImportError: |
| print("❌ Requests not found. Run: pip install requests") |
| return False |
| |
| return True |
|
|
| def test_ffmpeg(): |
| """Test if FFmpeg is installed""" |
| print("\nTesting FFmpeg...") |
| import subprocess |
| try: |
| result = subprocess.run(['ffmpeg', '-version'], |
| capture_output=True, |
| text=True, |
| timeout=5) |
| if result.returncode == 0: |
| version = result.stdout.split('\n')[0] |
| print(f"✅ FFmpeg found: {version}") |
| return True |
| else: |
| print("❌ FFmpeg found but returned error") |
| return False |
| except FileNotFoundError: |
| print("❌ FFmpeg not found. Install it:") |
| print(" Ubuntu/Debian: sudo apt-get install ffmpeg") |
| print(" macOS: brew install ffmpeg") |
| print(" Windows: Download from https://ffmpeg.org/") |
| return False |
| except Exception as e: |
| print(f"❌ Error testing FFmpeg: {e}") |
| return False |
|
|
| def test_fonts(): |
| """Test if Sinhala fonts are available""" |
| print("\nTesting Sinhala fonts...") |
| import subprocess |
| try: |
| |
| result = subprocess.run(['fc-list', ':lang=si'], |
| capture_output=True, |
| text=True, |
| timeout=5) |
| if 'Noto' in result.stdout or 'Sinhala' in result.stdout: |
| print("✅ Sinhala fonts found") |
| fonts = [line for line in result.stdout.split('\n') if 'Sinhala' in line or 'Noto' in line] |
| for font in fonts[:3]: |
| print(f" - {font[:80]}...") |
| return True |
| else: |
| print("⚠️ No Sinhala fonts found. Install Noto fonts:") |
| print(" Ubuntu/Debian: sudo apt-get install fonts-noto-core fonts-noto-extra") |
| print(" macOS: brew install font-noto-sans") |
| return False |
| except FileNotFoundError: |
| print("⚠️ fontconfig not found, cannot check fonts") |
| return False |
| except Exception as e: |
| print(f"⚠️ Error checking fonts: {e}") |
| return False |
|
|
| def test_subtitle_encoding(): |
| """Test subtitle encoding handling""" |
| print("\nTesting subtitle encoding...") |
| test_content = """1 |
| 00:00:01,000 --> 00:00:04,000 |
| හෙලෝ, මෙය සිංහල උපසිරැසි පරීක්ෂණයකි |
| |
| 2 |
| 00:00:05,000 --> 00:00:08,000 |
| Unicode test: \u0D9A\u0DCA\u200D\u0DBB\u0DD2\u0DBA\u0DCF |
| """ |
| |
| import tempfile |
| try: |
| |
| with tempfile.NamedTemporaryFile(mode='w', encoding='utf-8', |
| suffix='.srt', delete=False) as f: |
| f.write(test_content) |
| test_file = f.name |
| |
| |
| with open(test_file, 'r', encoding='utf-8') as f: |
| content = f.read() |
| |
| |
| if '\u0D80' <= content[50] <= '\u0DFF': |
| print("✅ Subtitle encoding test passed") |
| os.unlink(test_file) |
| return True |
| else: |
| print("❌ Subtitle encoding test failed") |
| os.unlink(test_file) |
| return False |
| except Exception as e: |
| print(f"❌ Subtitle encoding test failed: {e}") |
| return False |
|
|
| def test_app_import(): |
| """Test if app.py can be imported""" |
| print("\nTesting app.py import...") |
| try: |
| sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) |
| import app |
| print("✅ app.py imported successfully") |
| return True |
| except Exception as e: |
| print(f"❌ Failed to import app.py: {e}") |
| return False |
|
|
| def main(): |
| """Run all tests""" |
| print("=" * 60) |
| print("Sinhala Subtitle Hardcoder - Environment Test") |
| print("=" * 60) |
| |
| tests = [ |
| ("Imports", test_imports), |
| ("FFmpeg", test_ffmpeg), |
| ("Sinhala Fonts", test_fonts), |
| ("Subtitle Encoding", test_subtitle_encoding), |
| ("App Import", test_app_import), |
| ] |
| |
| results = {} |
| for name, test_func in tests: |
| results[name] = test_func() |
| |
| print("\n" + "=" * 60) |
| print("Test Results Summary") |
| print("=" * 60) |
| |
| for name, passed in results.items(): |
| status = "✅ PASS" if passed else "❌ FAIL" |
| print(f"{name:20s}: {status}") |
| |
| print("=" * 60) |
| |
| if all(results.values()): |
| print("\n🎉 All tests passed! Ready to deploy to Hugging Face.") |
| print("\nNext steps:") |
| print("1. Create a new Space on Hugging Face") |
| print("2. Upload app.py, requirements.txt, packages.txt, README.md") |
| print("3. Wait for build to complete") |
| print("4. Test with real videos!") |
| return 0 |
| else: |
| print("\n⚠️ Some tests failed. Fix the issues above before deploying.") |
| failed = [name for name, passed in results.items() if not passed] |
| print(f"\nFailed tests: {', '.join(failed)}") |
| return 1 |
|
|
| if __name__ == "__main__": |
| sys.exit(main()) |
|
|
