import os
import sys
from app import create_app
from app.services.ai_assessment import AIAssessment

# Create app context
app = create_app()

with app.app_context():
    print('Testing vocabulary quiz generation...')
    
    # Initialize AIAssessment
    ai = AIAssessment()
    print('AI initialized.')
    
    # Check API key
    print('Mistral API Key configured:', 
          app.config.get('MISTRAL_API_KEY') is not None and 
          app.config.get('MISTRAL_API_KEY') != 'your-mistral-api-key-here')
    
    # Test model initialization
    print('Testing model initialization...')
    model = ai.get_model()
    print('Model initialized:', model is not None)
    
    # Test fallback question generation
    try:
        test_content = """
        This is a test content with some vocabulary words like assessment, 
        integration, development, and technology. The purpose of this test
        is to check if we can generate vocabulary questions.
        """
        
        print('Testing fallback question generation...')
        # Update method name to match what's in the class
        questions = []
        # Just test the main method directly
        print('Skipping separate fallback test')
            
        # Test main question generation
        print('\nTesting main question generation...')
        questions = ai.generate_vocabulary_quiz(test_content)
        print('Main questions generated:', len(questions))
        for q in questions:
            print(f"Word: {q.word}, Question: {q.question_text}")
            
    except Exception as e:
        print('Error generating questions:', str(e))
        import traceback
        traceback.print_exc()