Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from PIL import Image, ImageDraw
+import numpy as np
+import cv2  # Add OpenCV to handle image processing
 import sys
 import os
 import time
@@ -54,8 +56,18 @@ def analyze_image(input_image):
 
     try:
         logging.info("Performing fracture detection.")
+
+        # Convert PIL image to NumPy array for OpenCV
+        img_array = np.array(input_image)  # Convert PIL image to NumPy array
+        if len(img_array.shape) == 2:  # Grayscale image
+            img_array = cv2.cvtColor(img_array, cv2.COLOR_GRAY2BGR)
+        elif img_array.shape[2] == 4:  # RGBA image
+            img_array = cv2.cvtColor(img_array, cv2.COLOR_RGBA2BGR)
+        else:
+            img_array = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR)
+
         # Perform fracture detection
-        detections = detector.detect_fractures(temp_image_path, conf_threshold=0.25)
+        detections = detector.detect_fractures(img_array, conf_threshold=0.25)
 
         # Mark the fracture areas on the image
         marked_image = mark_fracture_area(input_image.copy(), detections)