import gradio as gr
from glob import glob
from PIL import Image
from explain import get_results, reproduce


def classify_and_explain(image):
    reproduce()
    # This function will classify the image and return a list of image paths
    list_of_images = get_results(img_for_testing=image)
    return list_of_images


def get_examples():
    return [Image.open(i) for i in glob("samples/*")]


demo = gr.Interface(
    fn=classify_and_explain, inputs="image", outputs="gallery", examples=get_examples()
)
demo.launch()