from _arango import arango
from _llm import LLM
db = arango.db

db.create_collection("rumors_relations", edge=True)

q = 'for doc in rumors return doc'
rumors = list(db.aql.execute(q))
persons = list(db.collection("persons").all())
persons_string = "\n".join([i["name"] for i in persons])
llm = LLM(chat=False)

for rumor in rumors:

    heard_from = rumor["heard_from"]

    prompt = f'Nedan är en lista med personer:\n\n{persons_string}\n\nVem av dessa tror du att {heard_from} är?\n\nHär är mer kontext som det hjälper:\n\n{rumor['sexual_summary']} \n\nSvara med namn och hur personen nämns i texten på formen "namn;hur personen nämns\n".'
    response = llm.generate(prompt)

    db.collection("rumors_relations").insert(
        {
            "_from": person["_id"],
            "_to": rumor["_id"],
            "context": "rumor",
        }
    )