import os
import DBHelper
import Facial_Recognition_Wrapper


def start():
    # Downloads all the user and thief photos from database to the project folder first or updates them.
    count = 0
    users = DBHelper.db.child("Users").get()
    try:
        for x in users.each():
            count = +1
            for y in range(20):
                if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count)):
                    os.makedirs("Photos_of_Users/User_" + str(count))
                DBHelper.download_user_photo("User_" + str(count) + "/" + str(y) + ".jpg")
    except:
        print("No Users are registered.")
    count = 0
    try:
        for x in users.each():
            count = +1
            for y in range(20):
                if not os.path.isdir("Photos_of_Thieves/Thief_" + str(count)):
                    os.makedirs("Photos_of_Thieves/Thief_" + str(count))
                DBHelper.download_thief_photo("Thief_" + str(count) + "/" + str(y) + ".jpg")
    except:
        print("No Thieves for now.")
    Facial_Recognition_Wrapper.training_recognizer("Fisher")
    Facial_Recognition_Wrapper.face_recognition_inference("Fisher")