import sys
import pickle
from numpy import argmax
from spectrumizer import getFreqSpec
from train import class_labels

if len(sys.argv) < 4:
    print "Usage:\nnetwork dataset .wav-file"
    sys.exit(0)

nn = pickle.load(open(sys.argv[1], 'r'))

ds = pickle.load(open(sys.argv[2], 'r'))

freqSpec, freqs = getFreqSpec(sys.argv[3], ds.num_freq_bins*2)

ds.normalizeSpec(freqSpec)

answer = nn.activate(freqSpec)

print answer

print "Is this a \"" + class_labels[argmax(answer)] + "\" sound?"
import sys
import cPickle
from spectrumizer import getFreqSpec, plotFreqSpec

try:
    filename = sys.argv[1]
    fft_length = int(sys.argv[2])
except IndexError:
    sys.stderr.write('Usage: sound file, fft length\n')
    sys.exit(1)


spec, freqs = getFreqSpec(filename, fft_length)
plotFreqSpec(spec, freqs)