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

parser = argparse.ArgumentParser()
parser.add_argument("--month", help="month", type=str)
args = parser.parse_args()

folder = '/nfs/NAS3/SABIOD/SITE/BOUSSOLE_nas1/'
ff = '8' if args.month == 'NOV' else '9'
monthfolder = 'DATA_CRC_'+args.month+'_0'+ff+'/DATA_CRC_'+args.month+'_0'+ff+'_WAV/'
fns = os.listdir(folder+monthfolder)
fns = [monthfolder+f for f in fns]

ok = []
for f in fns:
    try:
        sf.info(folder+f)
    except :
        continue
    ok.append(f)

fns = ok
model = models.get['stft_depthwise_32kHz'](64)
stdc='models/dw_64_32kHz.stdc'
preds = run(fns, stdc, model, folder, pool=True, batch_size=64, fe=32000)
preds.to_pickle('bousole'+args.month+stdc.split('/')[-1].rsplit('.',1)[0]+'.preds')