#!/usr/bin/env python3
import sys
import matplotlib.pyplot as plt

file_path = sys.argv[1]

with open(file_path, "r") as f: 
    lines = f.readlines()

loss = []

key_word = "constrast_loss: "
for line in lines:
    if key_word in line:
        loss.append(line.split(key_word)[-1].split("|")[0])


X = range(len(loss))
plt.plot(X, loss)
plt.savefig(file_path + "_plot.png")