#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
API endpoint для парсера судебных документов
Использование: python3 court_parser_api.py "текст документа"
"""

import json
import sys
import os
from court_document_parser import CourtDocumentParser


def main():
    """API endpoint для вызова из n8n через SSH"""
    
    # Создаем парсер
    document_parser = CourtDocumentParser()
    
    try:
        # Получаем текст документа из аргумента командной строки
        if len(sys.argv) < 2:
            error_result = {
                'error': 'Не указан текст документа',
                'status': 'error'
            }
            print(json.dumps(error_result, ensure_ascii=False, indent=2))
            return 1
        
        # Берем текст из первого аргумента
        document_text = sys.argv[1]
        
        # Создаем объект документа
        input_data = {"combinedText": document_text}
        
        # Парсим документ
        results = document_parser.parse_documents([input_data])
        
        # Выводим результаты в stdout
        print(json.dumps(results, ensure_ascii=False, indent=2))
        
        return 0
        
    except Exception as e:
        error_result = {
            'error': f'Ошибка обработки: {str(e)}',
            'status': 'error'
        }
        print(json.dumps(error_result, ensure_ascii=False, indent=2))
        return 1


if __name__ == '__main__':
    sys.exit(main())