import gradio as gr

# Assuming you have a dictionary of pet names and their image URLs or file paths
pet_images = {
    "David": "Users/EmilyWitko/desktop/David AI/david.jpg",
    "Kikou": "Users/EmilyWitko/desktop/David AI/kikou.jpg",
    # Add as many pets as you have
}

# Define your Gradio interface here
def your_function(pet_name):
    # Handle the input and produce the image path or URL
    image_path = pet_images.get(pet_name, "Users/EmilyWitko/desktop/David AI/david.jpg")  # Default if pet not found
    return image_path

# Create the interface
iface = gr.Interface(fn=your_function, inputs="text", outputs="image")

# Launch the interface
iface.launch()