#!/usr/bin/python
from subprocess import Popen
import os
import sys
from face_client import FaceClient

api_key = '34a84a7835bf24df2d84b4bded84e838'
api_secret = '5bc9e8c5a9e3a2d916abbe3659a1b3f8'
url_base = 'http://74.95.195.225/static/img/jon/users/'
client = FaceClient(api_key, api_secret)

def train(ns,dataset,dirs):
    f = open('train.log', 'w')
    for d in dirs:
        tids = []
        count = 0
        user = d.rstrip('/').split('/')[-1]
        for line in open(d+'/'+dataset+'-train', 'r'):
            pic = line.split(',')[-2].split('/')[-1].split('.')[0]+'.jpeg'
            if pic != "face.jpeg":
                count += 1
                url = url_base+user+'/train/'+pic
                try:
                    response = client.faces_detect(url)
                    sys.stderr.write(str(response['photos'][0]['tags'])+"\n")
                    for photo in response['photos']:
                        if len(photo['tags']) > 0:
                            tids += [photo['tags'][0]['tid']]
                            f.write("\t".join([user, pic, "face"])+"/n")
                        else:
                            f.write("\t".join([user, pic, "no-face"])+"/n")
                except:
                    sys.stderr.write("Unexpexted error:"+str(sys.exc_info()[0]))
                    pass
        saved = client.tags_save(tids = ',' . join(tids), uid = user+'@'+ns)
        sys.stderr.write(str(saved)+"\n")
        trained = client.faces_train(user+'@'+ns)
        sys.stderr.write(str(trained)+"\n")
        f.write("\t".join([user, str(len(saved['saved_tags'])), str(trained)])+"/n")
        k = trained.keys()
        k.remove(u'status')
        ntrain = trained[k[0]][0][u'training_set_size']

        print ",".join([user,str(count),str(len(saved['saved_tags'])),str(ntrain)])


if __name__ == "__main__":
    train(sys.argv[1],sys.argv[2],sys.argv[3:])