Update app.py

app.py

@@ -52,7 +52,34 @@ def run_generate(sequence_images):
     if not sequence_images:
         return "", gr.DownloadButton(visible=False)
 
-    images = [s[1] for s in sequence_images]
+    PRIORITY_KEYWORDS = [
+    "flair", "tirm",            # T2 FLAIR — highest priority
+    "dwi", "diff",              # DWI
+    "t1_se_tra_320+fs",         # T1 + Contrast (has 'C' in name)
+    "t1",                       # T1
+    "t2_tse_tra", "t2_tra",     # T2 axial
+    ]
+
+    # Cap at 5
+    selected = sorted_sequences[:5]
+
+    # Score each sequence by priority
+    def get_priority(name_img_tuple):
+        name = name_img_tuple[0].lower()
+        for i, keyword in enumerate(PRIORITY_KEYWORDS):
+            if keyword in name:
+                return i
+        return 999  # unknown sequences go last
+
+    # Sort by priority and take top 4
+    sorted_sequences = sorted(sequence_images, key=get_priority)
+    selected = sorted_sequences[:4]
+
+    print("Sequences sent to model:")
+    for name, _ in selected:
+        print(f"  - {name}")
+
+    images = [img for _, img in selected]
     report = generate_report(images)
 
     return report, gr.DownloadButton(visible=True)