debug_translation.py

#!/usr/bin/env python3
"""
Debug script to test translation functionality
"""

import os
import asyncio
import sys
from pathlib import Path

# Add the current directory to Python path
sys.path.insert(0, str(Path(__file__).parent))

from translator import DocumentTranslator

async def test_translation():
    """Test the translation system"""
    print("🧪 Testing Document Translation System...")
    
    # Check API key
    api_key = os.getenv('OPENROUTER_API_KEY')
    if not api_key:
        print("❌ OPENROUTER_API_KEY not found!")
        print("Set it with: export OPENROUTER_API_KEY='your_key_here'")
        return
    
    print(f"✅ API key found (length: {len(api_key)})")
    
    # Initialize translator
    translator = DocumentTranslator()
    
    if not translator.is_ready():
        print("❌ Translator not ready")
        return
    
    print("✅ Translator initialized")
    
    # Test model availability
    models = await translator.get_available_models()
    print(f"✅ Available models: {len(models)}")
    for model in models:
        print(f"   - {model['name']}: {model['id']}")
    
    # Test basic translation
    test_text = "Hello, this is a test sentence for translation."
    print(f"\n🔤 Testing basic translation...")
    print(f"Original: {test_text}")
    
    try:
        translated = await translator.translate_text(
            test_text, 
            "google/gemini-2.5-pro-exp-03-25",
            "en", 
            "ar"
        )
        print(f"Translated: {translated}")
        
        if translated != test_text:
            print("✅ Basic translation working!")
        else:
            print("⚠️  Translation returned original text - check API key and credits")
            
    except Exception as e:
        print(f"❌ Translation test failed: {e}")
    
    print("\n🎯 Translation system test complete!")

if __name__ == "__main__":
    asyncio.run(test_translation())