add special logic for modality

Author: jdkent

store/backend/neurostore/models/data.py

@@ -753,6 +753,13 @@ def validate_version(self, key, value):
 
 class PipelineStudyResult(BaseMixin, db.Model):
     __tablename__ = "pipeline_study_results"
+    __table_args__ = (
+        sa.Index(
+            "ix_pipeline_study_results__modality",
+            sa.text("(result_data -> 'Modality')"),
+            postgresql_using="gin",
+        ),
+    )
 
     config_id = db.Column(
         db.Text, db.ForeignKey("pipeline_configs.id", ondelete="CASCADE"), index=True

store/backend/neurostore/resources/data.py

@@ -956,6 +956,27 @@ def view_search(self, q, args):
             for idx, (field_path, operator, value) in enumerate(
                 filters["result_filters"]
             ):
+                normalized_field = field_path.replace("[]", "")
+                if (
+                    pipeline_name == "TaskExtractor"
+                    and normalized_field == "Modality"
+                    and operator == "="
+                ):
+                    modality_values = [
+                        val.strip() for val in value.split("|") if val.strip()
+                    ]
+                    if modality_values:
+                        pipeline_query = pipeline_query.filter(
+                            sae.or_(
+                                *[
+                                    PipelineStudyResultAlias.result_data["Modality"]
+                                    .contains([modality_value])
+                                    for modality_value in modality_values
+                                ]
+                            )
+                        )
+                        pipeline_subqueries.append(pipeline_query.subquery())
+                    continue
                 jsonpath = build_jsonpath(field_path, operator, value)
                 param_name = f"jsonpath_result_{pipeline_name}_{idx}"
                 pipeline_query = pipeline_query.filter(