Update app.py

app.py

@@ -71,6 +71,23 @@ def unified_matching_plot2(image0, image1, kpts0, kpts1, mkpts0, mkpts1, color,
 
     return img
 
+def stitch_images(images):
+    """Stitches a list of images vertically."""
+    if not images:
+        return Image.new('RGB', (100, 100), color='gray')
+
+    max_width = max([img.width for img in images])
+    total_height = sum(img.height for img in images)
+
+    composite = Image.new('RGB', (max_width, total_height))
+
+    y_offset = 0
+    for img in images:
+        composite.paste(img, (0, y_offset))
+        y_offset += img.height
+
+    return composite
+
 # Main functions
 def detect_and_crop(target_image, query_image, threshold=0.5, nms_threshold=0.3):
     target_sizes = torch.Tensor([target_image.size[::-1]])
@@ -187,20 +204,21 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Upload a Query Image (Poster)"),
         gr.Image(type="pil", label="Upload a Target Image (Media)"),
         gr.Slider(minimum=0, maximum=100, step=1, value=50, label="Threshold"),
-        gr.CheckboxGroup(choices=[0.33, 0.66, 1.0], value=[0.33, 0.66, 1.0], label="Scale Factors")
+        gr.CheckboxGroup(choices=["0.33", "0.66", "1.0"], value=["0.33", "0.66", "1.0"], label="Scale Factors"),
     ],
     outputs=[
-        gr.JSON(label="Result")
+        gr.Label(label="Object Presence"),
+        gr.Image(type="pil", label="Detected Bounding Boxes"),
     ],
-    title="Object Detection in Image",
+    title="Object Detection in Images",
     description="""
-    **Instructions:**
-
-    1. **Upload a Query Image (Poster)**: Select an image file that contains the object you want to detect.
-    2. **Upload a Target Image (Media)**: Select an image file where you want to detect the object.
-    3. **Set Threshold**: Adjust the slider to set the threshold for object detection.
-    4. **Set Scale Factors**: Select the scale factors for image pyramid.
-    5. **View Results**: The result will show whether the object is present in the image along with additional details.
+    This application allows you to check if an object in a query image (poster) is present in a target image (media).
+    Steps:
+    1. Upload a Query Image (Poster)
+    2. Upload a Target Image (Media)
+    3. Set Threshold
+    4. Set Scale Factors
+    5. View Results
     """
 )