gatherimages.py

from datetime import date, datetime
from random import random
from time import sleep, time

import os
import config
import tinder_api as api
import features as feat
# import cv2
from PIL import Image
import urllib
from io import StringIO
import config
import json
import foldercount
import progressbar

if api.authverif() == True:
        print("Gathering images from nearby users for training...")
        # print(api.get_self())
        y = 1
        repeats = []

        # Create images folder if necessary
        if not os.path.isdir('./images'):
            os.makedirs('./images')

        while foldercount.numfile("./images/") < 1000:
            sleep(0.5)
            print(foldercount.numfile('./images/'))

            search =  api.get_recs_v2()
            try:
                ppl= search['data']['results']

                for i in range(len(ppl)):
                    
                    if ppl[i]['user']['_id'] in repeats:
                        pass
                    else:
                        pics = feat.get_photos(ppl[i]['user'])
                        path = './images/'
                        for j in pics:
                            x = foldercount.numfile(path)+1
                            file = urllib.request.urlopen(j)
                            img = Image.open(file)
                            img.save(path+'userpic'+str(x)+".jpg",'JPEG')
                        repeats.append(ppl[i]['user']['_id'])
            except KeyError:
                print("Key error...")
            
        print("Gathering images for predictions")

        # Create prediction folder if necessary
        if not os.path.isdir('./prediction'):
            os.makedirs('./prediction')

        while foldercount.numfile("./prediction/") < 100:
            sleep(0.5)

            search =  api.get_recs_v2()
            ppl= search['data']['results']

            for i in range(len(ppl)):
                
                if ppl[i]['user']['_id'] in repeats:
                    pass
                else:
                    pics = feat.get_photos(ppl[i]['user'])
                    path = './prediction/'
                    for j in pics:
                        x = foldercount.numfile(path)+1
                        file = urllib.request.urlopen(j)
                        img = Image.open(file)
                        img.save(path+'userpic'+str(x)+".jpg",'JPEG')
                    repeats.append(ppl[i]['user']['_id'])

            



else:
        print("Something went wrong. You were not authorized.")