Fix: sample option selection

app.py

@@ -28,37 +28,36 @@ sample_images = [
     os.path.join(os.getcwd(), "data/xai/sample2.jpg")
 ]
 
-interface = gr.Interface(
-    fn=process_image,
-    inputs=[
-        gr.Radio(
-            label="Select Sample Image or Upload Image", 
-            choices=["None"] + sample_images,  # Add 'None' option for uploading image
-            value="None",  # Default option is 'None'
-            type="value",  # We pass the file path or 'None' here
-        ),
-        gr.Image(
-            label="Upload Image", 
-            type="pil",  # Type 'pil' to work with PIL images
-            visible=False  # Initially hidden, will only be visible if 'None' is selected
-        ),
-        gr.CheckboxGroup(
-            choices=["yolov5", "yolov8s"],
-            value=["yolov5"],  # Default model selected
-            label="Select Model(s)",
-            type="value",  # Ensure the value is passed as a list of selected models
-        ),
-    ],
-    outputs=gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500),
-    title="Explainable AI for YOLO Models",
-    description="Select a sample image or upload your own image and choose a model to visualize YOLO object detection with Grad-CAM.",
-    examples=sample_images,  # Example images to display
-    live=True,
-)
+with gr.Blocks() as demo:
+    gr.Markdown("### Image Upload and Selection Interface")
+    
+    # Checkbox for selecting mode
+    mode_checkbox = gr.Checkbox(label="Select Mode", value=False)
+    
+    # Image input for uploading or selecting sample images
+    image_input = gr.Image(type="filepath", label="Upload or Select an Image", tool="editor")
+    
+    # Dropdown for selecting sample images
+    sample_selector = gr.Dropdown(choices=sample_images, label="Or select a sample image", multiselect=False)
+    
+    # Combine the image input and sample selector
+    def combine_inputs(uploaded_image, selected_sample):
+        return uploaded_image if uploaded_image else selected_sample
+    
+    combined_image = gr.Button("Combine Inputs")
+    
+    # Output display
+    output_text = gr.Textbox(label="Output")
+    
+    # Define actions on button click
+    combined_image.click(
+        lambda img_path: process_image(img_path, mode_checkbox.value),
+        inputs=[image_input, sample_selector],
+        outputs=output_text
+    )
+
 
-# Override the input image function to provide fallback
-interface.inputs[0].update(value=None)
 
 
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()