#!/usr/bin/env python3
"""Test batch summarization"""
from app import app
from models import db, User, Case
import sys

with app.app_context():
    # Get user
    user = User.query.filter_by(role='admin').first()
    if not user:
        print("No admin user found")
        sys.exit(1)

    # Get case 6
    case = Case.query.filter_by(id=6).first()
    if not case:
        print("Case 6 not found")
        sys.exit(1)

    print(f"Testing batch summarization for:")
    print(f"  User: {user.get_full_name()}")
    print(f"  Case: {case.case_name}\n")

    # Try importing the function
    try:
        from assistant_routes import batch_summarize_documents
        print("✅ Successfully imported batch_summarize_documents\n")

        # Try running it
        print("Calling batch_summarize_documents...")
        result = batch_summarize_documents(user, case)
        print("\nResult:")
        print(result)

    except Exception as e:
        print(f"❌ Error: {e}")
        import traceback
        traceback.print_exc()