add more debug log

app.py

@@ -12,19 +12,22 @@ def detect_traffic_signs(image):
     :param image: PIL Image or numpy array
     :return: numpy array of processed image
     """
+    print(f"Received image type: {type(image)}")
     # Convert PIL to numpy if necessary
     if hasattr(image, 'convert'):
         image = np.array(image)
-    
+        print(f"Converted PIL to numpy array, shape: {image.shape}")
+
     # Convert RGB to BGR for OpenCV
     image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
-    
+    print(f"Converted to BGR, shape: {image.shape}")
+
     # Perform detection
     result_image = detector.detect(image)
-    
+
     # Convert back to RGB for Gradio
     result_image = cv2.cvtColor(result_image, cv2.COLOR_BGR2RGB)
-    
+
     return result_image
 
 # Create Gradio interface

model.py

@@ -45,21 +45,25 @@ class TrafficSignDetector:
         :param image: numpy array of the image
         :return: image with drawn bounding boxes
         """
+        print(f"Input image shape: {image.shape}")
         results = self.model(image, conf=self.conf_threshold)
-        
+        print(f"Number of results: {len(results)}")
+
         for result in results:
             boxes = result.boxes
+            print(f"Number of boxes in this result: {len(boxes)}")
             for box in boxes:
                 # Get bounding box coordinates
                 x1, y1, x2, y2 = box.xyxy[0].cpu().numpy().astype(int)
                 conf = box.conf[0].cpu().numpy()
                 cls = int(box.cls[0].cpu().numpy())
-                
+                print(f"Detected: {self.classes[cls]} with conf {conf:.2f} at ({x1},{y1})-({x2},{y2})")
+
                 # Draw bounding box
                 cv2.rectangle(image, (x1, y1), (x2, y2), self.box_color, self.thickness)
-                
+
                 # Draw label
                 label = f"{self.classes[cls]}: {conf:.2f}"
                 cv2.putText(image, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, self.text_color, 2)
-        
+
         return image