Fix: interface layout

app.py

@@ -94,26 +94,39 @@ with gr.Blocks() as interface:
                 type="value",
                 value="Sample 1",  # Set default selection
             )
+            
             # Display the selected or uploaded image
             image_display = gr.Image(
                 value=load_sample_image("Sample 1"),  # Load default sample image
                 label="Selected/Uploaded Image",
             )
 
-            # File upload for custom images (placed below the sample selection)
-            uploaded_image = gr.Image(type="numpy", label="Upload an Image")
+            # Buttons for each sample image to upload custom image
+            upload_button_1 = gr.File(label="Upload Image for Sample 1", file_count="single")
+            upload_button_2 = gr.File(label="Upload Image for Sample 2", file_count="single")
+            upload_button_3 = gr.File(label="Upload Image for Sample 3", file_count="single")
 
-            # Update display when sample selection changes
+            # When a sample is selected, update the image display accordingly
             sample_selection.change(
                 fn=load_sample_image,
                 inputs=sample_selection,
                 outputs=image_display,
             )
 
-            # Update display when an image is uploaded
-            uploaded_image.change(
-                fn=lambda img: img,
-                inputs=uploaded_image,
+            # Handle file uploads for each custom image upload button
+            upload_button_1.change(
+                fn=lambda file: cv2.imread(file.name)[:, :, ::-1] if file else None,  # Read and display uploaded image
+                inputs=upload_button_1,
+                outputs=image_display,
+            )
+            upload_button_2.change(
+                fn=lambda file: cv2.imread(file.name)[:, :, ::-1] if file else None,
+                inputs=upload_button_2,
+                outputs=image_display,
+            )
+            upload_button_3.change(
+                fn=lambda file: cv2.imread(file.name)[:, :, ::-1] if file else None,
+                inputs=upload_button_3,
                 outputs=image_display,
             )