Update app.py

app.py

@@ -1,25 +1,36 @@
 import gradio as gr
 
-# Assuming you have a dictionary of pet names and their image URLs or file paths
-pet_images = {
-    "David": "./david.jpg",
-    "Kikou": "./kikou.jpg",
-    # Add as many pets as you have
+# 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
 }
 
-# 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, "./default_pet.jpg")  # Use a valid default image path
+# Define your function
+def your_function(owner_name, pet_name):
+    pets = owner_pets.get(owner_name)
+    
+    pet_image = None
+    if pets and 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, pet_name):
+    image_path = your_function(owner_name, pet_name)
     return image_path
 
 # Create the interface
-iface = gr.Interface(fn=your_function, inputs="text", outputs="image")
+interface = gr.Interface(fn=your_gradio_function, inputs=["text", "text"], outputs="image")
 
 # Launch the interface
-iface.launch()
+interface.launch()
 
 if __name__ == "__main__":
-    iface.input("Pet Name").send_value("Kikou")  # Set the initial value for the input field
-    output = iface.run().result()
-    print(f"Image for Kikou: {output}")
+    interface.inputs([{"label": "Owner Name", "type": "text"}, {"label": "Pet Name", "type": "text"}])
+    interface.input("Owner Name").send_value("John Doe")
+    interface.input("Pet Name").send_value("Doggo")
+    output = interface.run().result()
+    print(f"Image for Emily and David: {output}")