Spaces:
Runtime error
Runtime error
# Import necessary library | |
import gradio | |
# Define the nested dictionary of owners and their respective pets | |
owner_pets = { | |
"Emily": {"David": "./David.jpg"}, | |
"Sasha": {"Kikou": "./Kikou.jpg"}, | |
# Add more owners and their pets as needed | |
} | |
import os | |
for owner, owned_pets in owner_pets.items(): | |
for pet, img_path in owned_pets.items(): | |
assert os.path.isfile(img_path), f"File '{img_path}' does not exist." | |
# Define your function | |
def your_function(owner_name, pet_name=None): | |
pets = owner_pets.get(owner_name) | |
if pets is None: | |
# Return the first pet image if no pet name was provided | |
return next((val for sublist in owner_pets.values() for val in sublist), "./download.jpeg") | |
pet_image = None | |
if pets and (pet_name is None or pet_name in pets): | |
pet_image = pets[pet_name] | |
return pet_image or "./download.jpeg" | |
# Define your Gradio interface | |
def your_gradio_function(owner_name): | |
print(f"Received owner name: {owner_name}") | |
image_path = your_function(owner_name) | |
print(f"Returned image path: {image_path}") | |
return image_path | |
# Create the interface | |
interface = gradio.Interface(fn=your_gradio_function, inputs=["text"], outputs="image") | |
# Launch the interface | |
interface.launch() | |
if __name__ == "__main__": | |
interface.input("Owner Name").send_value("Emily") | |
output = interface.run().result() | |
print(f"Image for Emily: {output}") |