Update app.py

app.py

@@ -9,8 +9,13 @@ import json
 import numpy as np
 from ultralytics import YOLO
 import cv2
-from streamlit_img_label import st_img_label
-from streamlit_img_label.manage import ImageManager
+
+try:
+    from streamlit_img_label import st_img_label
+    from streamlit_img_label.manage import ImageManager
+except ImportError:
+    st.error("Required modules are not installed. Please install 'streamlit-img-label' and 'pascal-voc-writer'.")
+    st.stop()
 
 # Load CLIP model and tokenizer
 @st.cache_resource
@@ -139,7 +144,11 @@ def adjust_bounding_boxes(image, detections):
             "label": detection['category']
         })
     
-    adjusted_rects = st_img_label(image, box_color="red", rects=rects)
+    try:
+        adjusted_rects = st_img_label(image, box_color="red", rects=rects)
+    except Exception as e:
+        st.error(f"Error in st_img_label: {str(e)}")
+        return detections
     
     adjusted_detections = []
     for rect, detection in zip(adjusted_rects, detections):