from module import Topic_Aware_Rag, Data_Init, Extract_Triple,Score

def main():

    data, template = Data_Init("data/AstroKGC_test/Paper_Triple/AstroTriple.json", "data/AstroKGC_test/Template_Triple/AstroTem.json")

    Rag_input = Topic_Aware_Rag(data, template)

    output = Extract_Triple(Rag_input)

    Triple = [item[1] for item in data]

    print(Score(output,Triple))

if __name__ == "__main__":
    main()