Fix: gradio interface

app.py

@@ -8,7 +8,6 @@ from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
 
 def process_image(image, yolo_versions=["yolov5"]):
-    # Convert image from PIL to NumPy array
     image = np.array(image)
     image = cv2.resize(image, (640, 640))
 
@@ -28,36 +27,40 @@ sample_images = [
     os.path.join(os.getcwd(), "data/xai/sample2.jpg")
 ]
 
-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
-    )
-
-
-
-
-if __name__ == "__main__":
-    demo.launch()
+def load_sample(sample_name):
+    if sample_name and sample_name in sample_images:
+        return Image.open(sample_images[sample_name])
+    return None
+
+interface = gr.Interface(
+    fn=process_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload an Image"),
+        gr.CheckboxGroup(
+            choices=["yolov3", "yolov8s"],
+            value=["yolov5"],  # Set the default value (YOLOv5 checked by default)
+            label="Select Model(s)",
+        ),
+        gr.Dropdown(
+            choices=list(sample_images.keys()),
+            label="Select a Sample Image",
+            type="value",
+            interactive=True,
+        ),
+    ],
+    outputs=gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500),
+    title="Visualising the key image features that drive decisions with our explainable AI tool.",
+    description="XAI: Upload an image or select a sample to visualize object detection of your models.",
+)
+
+def main_logic(uploaded_image, selected_models, sample_selection):
+    # If the user selects a sample image, use that instead of the uploaded one
+    if sample_selection:
+        image = load_sample(sample_selection)
+    else:
+        image = uploaded_image
+
+    # Call the processing function
+    return process_image(image, selected_models)
+
+interface.launch()