Update cellemetry/services/sam.py

cellemetry/services/sam.py

@@ -31,7 +31,6 @@ def execute_segmentation(deps: AnalysisDeps, request: ComponentRequest) -> str:
     """
     t_start = time.time()
     
-    # SIMPLIFIED PROMPT: Color + Entity only (e.g., "green cell", "purple cell")
     text_prompt = f"{request.color} {request.entity}"
     print(f"\n[Engine] Segmenting: '{text_prompt}' ({len(request.bboxes)} boxes).")
 
@@ -64,7 +63,7 @@ def execute_segmentation(deps: AnalysisDeps, request: ComponentRequest) -> str:
     if not sam_input_boxes:
         return "No valid boxes provided."
 
-    safe_label = f"{request.color}_{request.entity}".replace(" ", "_").lower()
+    safe_label = f"{request.entity}".replace(" ", "_").lower()
     plot_filename = f"/tmp/out_{safe_label}.png"
     data_filename = f"/tmp/data_{safe_label}.npz"
     
@@ -181,19 +180,6 @@ def _save_plot(image, results, boxes, label, filename):
         
         ax.imshow(composite)
 
-    # 2. Draw Bounding Boxes
-    for box in boxes:
-        x_min, y_min, x_max, y_max = box
-        width = x_max - x_min
-        height = y_max - y_min
-        
-        rect = patches.Rectangle(
-            (x_min, y_min), width, height, 
-            linewidth=2, edgecolor='red', facecolor='none', linestyle='--'
-        )
-        ax.add_patch(rect)
-
-    ax.set_title(f"{label} (with Agent BBoxes)")
     ax.axis('off')
     
     # Save tightly to remove whitespace