fixed bug in perform_object_detection function

app.py

@@ -95,27 +95,34 @@ def perform_object_detection(image, model_name, threshold):
     Returns:
     PIL.Image, str: The image with drawn bounding boxes and a string of detected objects.
     """
-    # Initialize the ObjectDetector
-    detector = ObjectDetector()
-    # Load the specified model
-    detector.load_model(model_name)
-    # Perform object detection
-    processed_image, detected_objects = detector.detect_objects(image, threshold)
-    return processed_image, detected_objects
+    
+    try:
+        # Perform object detection and draw bounding boxes
+        processed_image, detected_objects = detect_and_draw_objects(image, model_name, threshold)
+        return processed_image, detected_objects
+    except Exception as e:
+        # Print the error for debugging
+        print(f"Error in object detection: {e}")
+        return None, str(e)
+
 
 # Check if the 'Detect Objects' button was clicked
 if detect_button:
     if image is not None:
         # Open the uploaded image
-        image = Image.open(image)
-        # Display the original image
-        st.image(image, use_column_width=True, caption="Original Image")
-        
+        try:
+            image = Image.open(image)
+            # Display the original image
+            st.image(image, use_column_width=True, caption="Original Image")
+        except Exception as e:
+            st.error(f"Error loading image: {e}")
+            return
+
         # Perform object detection
         processed_image, detected_objects = perform_object_detection(image, detect_model, threshold)
         
-        # Display the image with detected objects
-        if isinstance(processed_image, Image.Image):
+        if processed_image:
+            # Display the image with detected objects
             st.image(processed_image, use_column_width=True, caption="Image with Detected Objects")
         else:
             st.error("Failed to process image for object detection.")
@@ -124,4 +131,3 @@ if detect_button:
         st.write(detected_objects)
     else:
         st.write("Please upload an image for object detection.")
-