import os

# os.environ['CUDA_VISIBLE_DEVICES']='3'

from ToT.task import *

import json

import random

from tqdm import tqdm

import sys

from contextlib import contextmanager

@contextmanager

def redirect_stdout_to_file(filename):

sys.stdout = original_stdout

with open('', 'r') as f: # Your question files

lines = f.readlines()

model = 'qwen'

reasoning = open('data/reasoning/tot-'+ model + '.json', 'a+', encoding='utf-8')

output_file = 'tot-output-' + model + '.log'

count = 0

# 打印抽取的行

for line in tqdm(lines):

question = json.loads(line)['question']

with redirect_stdout_to_file(output_file):

task = ToT_Task(question, model, model)

output = task.run()

print(output)

reasoning.write(json.dumps({'question': question, 'solution': output[0]['solution']}, ensure_ascii=False) + '\n')

if count % 10 == 0:

reasoning.flush()

reasoning.close()

# print(output[0]['solution'])

# print(output[0]['summary'])