Update app.py

app.py

@@ -8,29 +8,31 @@ owner_pets = {
 }
 
 # Define your function
-def your_function(owner_name, pet_name):
+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 in pets:
+    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, pet_name):
-    image_path = your_function(owner_name, pet_name)
+def your_gradio_function(owner_name):
+    image_path = your_function(owner_name)
     return image_path
 
 # Create the interface
-interface = gr.Interface(fn=your_gradio_function, inputs=["text", "text"], outputs="image")
+interface = gr.Interface(fn=your_gradio_function, inputs=["text"], outputs="image")
 
 # Launch the interface
 interface.launch()
 
 if __name__ == "__main__":
-    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")
+    interface.input("Owner Name").send_value("Emily")
     output = interface.run().result()
-    print(f"Image for Emily and David: {output}")
+    print(f"Image for Emily: {output}")