from _llm import LLM
from _arango import arango
from print_color import *


llm = LLM(chat=False)
interrogations = list(arango.db.collection("interrogations").all())

for interrogation in interrogations:
    text = interrogation['text']
    prompt = f'Vad sägs om lördagskvällen i texten nedan? \n\n"""{text}""" Jag vill veta vad som sägs i texten om lördagskvällen. Var noga med prsonre, namn och platser.'

    answer = llm.generate(prompt)
    print_blue(interrogation['person'])
    print(answer, '\n')