biomedparse+biomedllama_3b_multimodal

app.py

@@ -385,11 +385,11 @@ def process_image(image_path, user_prompt, modality=None):
                 # First loop: collect prompts found in the analysis
                 found_prompts = []
                 for idx, prompt in enumerate(relevant_prompts):
-                    if prompt in detailed_analysis.lower():
-                        color = colors[idx % len(colors)]
-                        color_name = {(255,0,0): "red", (0,255,0): "green", (0,0,255): "blue", 
+                    color = colors[idx % len(colors)]
+                    color_name = {(255,0,0): "red", (0,255,0): "green", (0,0,255): "blue", 
                                     (255,255,0): "yellow", (255,0,255): "magenta"}[color]
-                        color_descriptions.append(f"The {prompt} is highlighted in {color_name} color")
+                    if prompt in detailed_analysis.lower():
+                        color_descriptions.append(f"The {prompt} is highlighted in {color_name} color for reference")
                         found_prompts.append(prompt)
                 
                 # Second loop: only process prompts found in analysis
@@ -418,7 +418,7 @@ def process_image(image_path, user_prompt, modality=None):
                 
                 if not results:
                     results = [image]  # Return original image if no overlays were created
-                detailed_analysis += "\n\nIn the segmentation analysis:"
+                detailed_analysis += ""
                 if color_descriptions:
                     detailed_analysis += " " + " ".join(color_descriptions) + "."
                 else: