main.py

from Model.tree_gen import GenerationWrapper
from Model.prompts import Baseparser
from Model.llm import GroqLLM, OpenaiLLM

if __name__ == "__main__":
    base_answer = open("Data/bhoomi_q1_mock10.txt").read()
    base_question = open("Data/q1_mock10.txt").read()

    base_answer_ = Baseparser.parse(base_answer)
    base_question_ = Baseparser.parse(base_question)

    generator = GenerationWrapper(claim_context=base_answer_, rebuttal_context=base_question_,
                                   tree_save_path=f"Trees/test_G.pkl",
                                   llm_mode="openai", openai_model = "gpt-4o")

    G = generator.create_graph(base_answer=base_answer_)

    print(G.show())