Add debug image

app.py

@@ -60,11 +60,12 @@ def extract_flip7_value(card_img):
 def detect_cards_and_sum(image, game):
     try:
         if image is None:
-            return 0, 0, "No image provided."
+            return 0, 0, "No image provided.", None
 
         if not isinstance(image, np.ndarray):
             image = np.array(image.convert("RGB"))
         img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+        display_img = img.copy()
 
         gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
         blur = cv2.GaussianBlur(gray, (5,5), 0)
@@ -94,14 +95,20 @@ def detect_cards_and_sum(image, game):
                     detected_values.append(value)
                     total += value
 
+                    # Draw bounding box + label
+                    cv2.polylines(display_img, [approx], True, (0,255,0), 3)
+                    x, y = approx[0][0]
+                    cv2.putText(display_img, str(value), (int(x), int(y)-10),
+                                cv2.FONT_HERSHEY_SIMPLEX, 1.2, (0,0,255), 3)
+
         if not detected_values:
-            return 0, 0, "No valid card values detected."
+            return 0, 0, "No valid card values detected.", cv2.cvtColor(display_img, cv2.COLOR_BGR2RGB)
 
-        print(detected_values)
-        return len(detected_values), total, None
+        return len(detected_values), total, None, cv2.cvtColor(display_img, cv2.COLOR_BGR2RGB)
 
     except Exception as e:
-        return 0, 0, f"Error: {str(e)}"
+        return 0, 0, f"Error: {str(e)}", None
+
 
 
 # Gradio UI
@@ -114,12 +121,13 @@ with gr.Blocks() as demo:
     num_cards = gr.Number(label="Number of Cards Detected")
     total = gr.Number(label="Total Sum")
     error = gr.Textbox(label="Error Message")
+    annotated = gr.Image(type="numpy", label="Detected Cards with Values")
 
     run_btn = gr.Button("Process")
     run_btn.click(
         fn=detect_cards_and_sum,
         inputs=[image_input, game_choice],
-        outputs=[num_cards, total, error]
+        outputs=[num_cards, total, error, annotated]
     )
 
 if __name__ == "__main__":