#!/usr/bin/env python3
import sys
import json
from deeppavlov import build_model, configs

def classify_dispute(text):
    # Загружаем базовую модель классификации из DeepPavlov
    classifier = build_model(configs.classifiers.torch_classification_model, download=True)
    result = classifier([text])
    # Предполагается, что результат — это список, содержащий метку класса
    return result[0]

def main():
    if len(sys.argv) < 2:
        print("Usage: classify_dispute.py \"<текст>\"")
        sys.exit(1)
    
    input_text = " ".join(sys.argv[1:])
    essence = classify_dispute(input_text)
    output = {"суть_спора": essence}
    print(json.dumps(output, ensure_ascii=False, indent=2))

if __name__ == '__main__':
    main()