from _arango import db
from _llm import LLM
from print_color import *
relations = list(db.aql.execute('for doc in relations return doc', count=True))

for relation in relations:
    interrogation = db.collection('interrogations').get(relation['interrogation'])
    if not interrogation:
        print_red(relation)
        continue
    for i in ['to', 'from']:
        if 'name' not in interrogation:
            interrogation['name'] = interrogation['person']
            db.collection('interrogations').update(interrogation, check_rev=False)
        if relation[i] == interrogation['name']:
            relation[i] = interrogation['person_id']
        elif relation[i] == interrogation['person_mentioned_as']:
            relation[i] = interrogation['person_id']
    for k, v in relation.items():
        print_rainbow(k, v)
    print()
    for k, v in interrogation.items():
        if k == 'text' or k == 'formatted_text':
            continue
        print_rainbow(k, v)
    db.collection('relations').update(relation, check_rev=False)