#!/usr/bin/env python3
from app import app
from models import db, User
from assistant_routes import search_all_documents

with app.app_context():
    # Get first admin user
    user = User.query.filter_by(role='admin').first()

    if user:
        print(f"Testing search for user: {user.get_full_name()} (ID: {user.id})")
        result = search_all_documents(user, "vulnerabilities")
        print("\n" + "="*80)
        print(result)
        print("="*80)
    else:
        print("No admin user found")
        # Try any user
        user = User.query.first()
        if user:
            print(f"Testing with first user: {user.get_full_name()} (ID: {user.id})")
            result = search_all_documents(user, "vulnerabilities")
            print("\n" + "="*80)
            print(result)
            print("="*80)
        else:
            print("No users found")