reset to biomedparse+biomedllama_3b_multimodal

app.py

@@ -384,15 +384,13 @@ 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):
-                    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", 
+                    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_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):