Update object_description_generator.py

object_description_generator.py

@@ -514,10 +514,10 @@ class ObjectDescriptionGenerator:
                 confident_objects = [obj for obj in detected_objects
                                    if obj.get("confidence", 0) >= self.confidence_threshold_for_description]
                 
-                print(f"DEBUG: After confidence filtering (threshold={self.confidence_threshold_for_description}):")                
-                for class_name in ["car", "traffic light", "person", "handbag"]:                   
-                    class_objects = [obj for obj in confident_objects if obj.get("class_name") == class_name]
-                    print(f"DEBUG: {class_name}: {len(class_objects)} confident objects")
+                # print(f"DEBUG: After confidence filtering (threshold={self.confidence_threshold_for_description}):")                
+                # for class_name in ["car", "traffic light", "person", "handbag"]:                   
+                #     class_objects = [obj for obj in confident_objects if obj.get("class_name") == class_name]
+                #     print(f"DEBUG: {class_name}: {len(class_objects)} confident objects")
 
                 if not confident_objects:
                     no_confident_obj_msg = "While some elements might be present, no objects were identified with sufficient confidence for a detailed description."
@@ -557,10 +557,10 @@ class ObjectDescriptionGenerator:
                             if name not in objects_by_class:
                                 objects_by_class[name] = []
                             objects_by_class[name].append(obj)
-                            print(f"DEBUG: Before spatial deduplication:")
-                            for class_name in ["car", "traffic light", "person", "handbag"]:
-                                if class_name in objects_by_class:                                    
-                                    print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects before dedup")
+                            # print(f"DEBUG: Before spatial deduplication:")
+                            # for class_name in ["car", "traffic light", "person", "handbag"]:
+                            #     if class_name in objects_by_class:                                    
+                            #         print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects before dedup")
 
                     if not objects_by_class:
                         description_segments.append("No common objects were confidently identified for detailed description.")
@@ -617,10 +617,10 @@ class ObjectDescriptionGenerator:
 
                         objects_by_class = deduplicated_objects_by_class
                         
-                        print(f"DEBUG: After spatial deduplication:")
-                        for class_name in ["car", "traffic light", "person", "handbag"]:
-                            if class_name in objects_by_class:                                
-                                print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects after dedup")
+                        # print(f"DEBUG: After spatial deduplication:")
+                        # for class_name in ["car", "traffic light", "person", "handbag"]:
+                        #     if class_name in objects_by_class:                                
+                        #         print(f"DEBUG: {class_name}: {len(objects_by_class[class_name])} objects after dedup")
                             
                         sorted_object_groups = sorted(objects_by_class.items(), key=sort_key_object_groups)
 
@@ -629,8 +629,8 @@ class ObjectDescriptionGenerator:
                         for class_name, group_of_objects in sorted_object_groups:
                             count = len(group_of_objects)
                             
-                            if class_name in ["car", "traffic light", "person", "handbag"]:                                
-                                print(f"DEBUG: Final count for {class_name}: {count}")
+                            # if class_name in ["car", "traffic light", "person", "handbag"]:                                
+                            #     print(f"DEBUG: Final count for {class_name}: {count}")
                                 
                             if count == 0:
                                 continue
@@ -1282,4 +1282,4 @@ class ObjectDescriptionGenerator:
 
         except Exception as e:
             self.logger.error(f"Error updating configuration: {str(e)}")
-            raise ObjectDescriptionError(f"Failed to update configuration: {str(e)}") from e
+            raise ObjectDescriptionError(f"Failed to update configuration: {str(e)}") from e