import numpy as np
import matplotlib.pyplot as plt

print('Insira o nome do arquivo de output:')
output = str(input())

data = np.loadtxt(output)

l_obs = data[:,0]
f_obs = data[:, 1]
f_syn = data[:, 2]

fig, ax = plt.subplots()
ax.plot(l_obs, f_obs, color='blue', lw='1', label='Observed')
ax.plot(l_obs, f_syn, color='red', lw='1', label='Synthetic')
ax.legend(loc='upper right')

ax.set_xlim([3600, 7000])
ax.set_title('Spectrum of galaxy')

fileName = output + '.png'

fig.savefig(fileName)
plt.show()