from fastai.vision.all import load_learner
import gradio as gr

#import pathlib
#temp = pathlib.PosixPath
#pathlib.PosixPath = pathlib.WindowsPath

labels = ['Acinonyx jubatus', 'Ailurus fulgens', 'Antilope cervicapra', 'Aonyx cinereus', 'Arctictis binturong', 'Axis porcinus', 'Bos frontalis', 'Bos grunniens', 'Bos taurus', 'Boselaphus tragocamelus', 'Bubalus bubalis', 'Budorcas taxicolor', 'Callosciurus erythraeus', 'Canis aureus', 'Canis lupus', 'Capra falconeri', 'Capra sibirica', 'Caracal caracal', 'Catopuma temminckii', 'Cervus elaphus', 'Cuon alpinus', 'Cynopterus sphinx', 'Dicerorhinus sumatrensis', 'Elephas maximus', 'Equus hemionus', 'Equus kiang', 'Felis catus', 'Felis chaus', 'Felis manul', 'Felis silvestris', 'Funambulus palmarum', 'Funambulus pennantii', 'Gazella bennettii', 'Helarctos malayanus', 'Hemitragus jemlahicus', 'Herpestes edwardsi', 'Herpestes javanicus', 'Herpestes smithii', 'Herpestes urva', 'Herpestes vitticollis', 'Hipposideros ater', 'Hipposideros speoris', 'Hoolock hoolock', 'Hyaena hyaena', 'Kerivoula picta', 'Lepus nigricollis', 'Loris lydekkerianus', 'Lutra lutra', 'Lutrogale perspicillata', 'Lynx lynx', 'Macaca silenus', 'Manis crassicaudata', 'Martes flavigula', 'Martes foina', 'Mellivora capensis', 'Melursus ursinus', 'Moschiola indica', 'Muntiacus muntjak', 'Mustela altaica', 'Mustela erminea', 'Mustela sibirica', 'Naemorhedus goral', 'Naemorhedus griseus', 'Neofelis nebulosa', 'Nilgiritragus hylocrius', 'Nycticebus bengalensis', 'Ochotona roylei', 'Orcinus orca', 'Panthera leo', 'Panthera pardus', 'Panthera tigris', 'Pantholops hodgsonii', 'Pardofelis marmorata', 'Pipistrellus coromandra', 'Prionailurus bengalensis', 'Prionailurus rubiginosus', 'Prionailurus viverrinus', 'Pteropus faunulus', 'Pteropus giganteus', 'Pteropus hypomelanus', 'Pteropus vampyrus', 'Ratufa bicolor', 'Ratufa indica', 'Rhinoceros unicornis', 'Rhinolophus lepidus', 'Rucervus duvaucelii', 'Rucervus eldii', 'Rusa unicolor', 'Semnopithecus ajax', 'Sus scrofa', 'Tamiops mcclellandii', 'Tetracerus quadricornis', 'Trachypithecus phayrei', 'Trachypithecus pileatus', 'Tupaia belangeri', 'Uncia uncia', 'Ursus arctos', 'Ursus thibetanus', 'Viverra zibetha', 'Viverricula indica', 'Vulpes bengalensis', 'Vulpes vulpes', 'Ziphius cavirostris']

model = load_learner("mammals-of-india-v0.pkl")

def recognize_image(image):
    pred, idx, probs = model.predict(image)
    print(pred, idx, probs)
    return dict(zip(labels, map(float, probs)))


image = gr.inputs.Image(shape=(192, 192))
label = gr.outputs.Label(num_top_classes=5)

iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label)
iface.launch(inline=False)