test_gen.py

import sys
import os
from pathlib import Path

# Add project root to path
sys.path.append(str(Path(__file__).parent))

from src.summarization.note_generator import NoteGenerator
from src.utils.config import settings

def test_note_generator():
    print("Testing NoteGenerator initialization after fix...")
    try:
        # We need an API key for initialization if it's not in env
        api_key = os.getenv("GOOGLE_API_KEY", "dummy_key")
        gen = NoteGenerator(api_key=api_key)
        print("✅ Initialization successful")
        print(f"Model ID: {gen.model_id}")
    except Exception as e:
        print(f"❌ Initialization failed: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    test_note_generator()