import gradio as gr
import cv2
from fastai.vision.all import *
import imutils

# Definimos una función que se encarga de llevar a cabo las predicciones
def panoramica(img1,img2):

  lista = []
  
  pilImg1=PIL.Image.open(img1).convert('RGB') 
  open_cv_image = numpy.array(pilImg1) 
  lista.append(open_cv_image)
  
  pilImg2=PIL.Image.open(img2).convert('RGB') 
  open_cv_image2 = numpy.array(pilImg2) 
  lista.append(open_cv_image2 )

#  for i in imagenes:
#    lista.append(PILImage.create(i))
 
  
  stitcher = cv2.createStitcher() if imutils.is_cv3() else cv2.Stitcher_create()
  (estado, pan) = stitcher.stitch(lista)
  return pan
    
# Creamos la interfaz y la lanzamos. 
gr.Interface(fn=panoramica, inputs=[gr.inputs.Image(),gr.inputs.Image()], outputs=gr.outputs.Image(),examples=['edificio1.jpg','edificio2.jpg']).launch(share=False)