from stockfish import Stockfish
import sys



stockfish = Stockfish("/home/tun08104/senior_design/stockfish_dwld/stockfish/src/stockfish")  
fen_file = sys.argv[1]
output_file = sys.argv[2]

# Engine options
stockfish.update_engine_parameters({
    "Threads": 4,   # increase if allowed
    "Hash": 512     # MB, increase if allowed
})

counter = 0
with open(fen_file) as fin, open(output_file, "w") as fout:
    for line in fin:
        fen = line.strip()
        if fen == "":
            continue
        counter += 1
        print(f"Processing FEN #{counter}")
        stockfish.set_fen_position(fen)
        eval = stockfish.get_evaluation()
        score = eval["value"]
        fout.write(f"{fen} {score}\n")