Update app.py

app.py

@@ -97,10 +97,19 @@ def process_images(flat_img, persp_img, json_file):
         roi_corners_flat = get_rotated_rect_corners(roi_x, roi_y, roi_w, roi_h, roi_rot_deg)
         roi_corners_persp = cv2.perspectiveTransform(roi_corners_flat.reshape(-1,1,2), H).reshape(-1,2)
 
-        # Draw ROI & detector label on perspective image only
+        # Draw ROI
         persp_out = persp_img.copy()
         cv2.polylines(persp_out, [roi_corners_persp.astype(int)], True, (0,255,0), 3)
-        cv2.putText(persp_out, det, (10,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,255,255), 2)
+
+        # Draw attractive detector label box
+        text = det
+        font = cv2.FONT_HERSHEY_SIMPLEX
+        scale = 1
+        thickness = 2
+        text_size = cv2.getTextSize(text, font, scale, thickness)[0]
+        text_x, text_y = 10, 30
+        cv2.rectangle(persp_out, (text_x-5, text_y-25), (text_x + text_size[0]+5, text_y+5), (0,0,0), -1)
+        cv2.putText(persp_out, text, (text_x, text_y), font, scale, (0,255,255), thickness)
 
         # Convert to RGB for Gradio
         persp_rgb = cv2.cvtColor(persp_out, cv2.COLOR_BGR2RGB)
@@ -117,10 +126,10 @@ iface = gr.Interface(
         gr.File(type="filepath", label="JSON File")
     ],
     outputs=[
-        gr.Gallery(label="Detector Results (Perspective Images with ROI & Label)")
+        gr.Gallery(label="Detector Results (Perspective Images with ROI & Detector Label)")
     ],
     title="Feature Detection ROI Projection",
-    description="Shows SIFT, ORB, BRISK, AKAZE, KAZE detector results on the Perspective image only, with ROI and detector label."
+    description="Shows SIFT, ORB, BRISK, AKAZE, KAZE detector results on the Perspective image only, with ROI and attractive detector label."
 )
 
 iface.launch()