import models
import os
import pandas as pd
import numpy as np
from run_model import run
import soundfile as sf
import argparse

stdc='models/stft_depthwise_64_r0.stdc'
folder = '/nfs/NASDELL/SABIOD/SITE/BOMBYX/'
monthfolder = '/BOMBYX2017/'
subs = ['BOMBYX3-11','BOMBYX3-14', 'BOMBYX3-15', 'BOMBYX3-4']
preds = pd.DataFrame()
model = models.get['stft_depthwise'](64)

for s in subs:
    monthfolder = '/BOMBYX2017/'+s+'/'
    fns = os.listdir(folder+monthfolder)
    fns = [monthfolder+f+'/'+f.replace('-',"")+'_BOMBYX_3.wav' for f in fns if not ('octave' in f or 'spectro' in f)]
    preds  = preds.append(run(fns, stdc, model, folder, pool=True, batch_size=128))


outfn = 'BOMBYX2017'+stdc.split('/')[-1].rsplit('.',1)[0]+'.preds'
print(outfn)
#preds = run(fns, stdc, model, folder, pool=True, batch_size=128)
preds.to_pickle(outfn)