biomedparse+biomedllama_3b_multimodal

app.py

@@ -384,13 +384,15 @@ def process_image(image_path, user_prompt, modality=None):
                 color_descriptions = []
                 # First loop: collect prompts found in the analysis
                 found_prompts = []
+                count = 0
                 for idx, prompt in enumerate(relevant_prompts):
-                    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]
                     if prompt in detailed_analysis.lower():
+                        color = colors[count % 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 for reference")
                         found_prompts.append(prompt)
+                        count += 1
                 
                 # Second loop: only process prompts found in analysis
                 for idx, prompt in enumerate(found_prompts):