Fix tag-driven H3 ref routing

dumas_h3_longvideos.py

@@ -4635,6 +4635,31 @@ def resolve_prompt_refs(text, ref_list, include_named=True):
     return rewritten, refs, dropped
 
 
+def resolve_tag_driven_prompt_refs(text, ref_list):
+    """Tag-driven prompt refs for mixed tagged/untagged scripts.
+
+    Once any shot in the script uses `<Picture N>`, untagged shots should keep the
+    handoff path instead of quietly becoming ref-conditioned just because they name
+    a character or location. Named refs still supplement explicitly tagged shots so
+    their semantic context and actual image list stay aligned.
+    """
+    rewritten, tagged_refs, dropped = resolve_tagged_refs(text, ref_list)
+    if not tagged_refs:
+        return rewritten, [], dropped
+    refs = list(tagged_refs)
+    seen_slots = {
+        slot_number
+        for slot_number in picture_tags(text)
+        if 1 <= slot_number <= len(ref_list or []) and _reference_image(ref_list[slot_number - 1]) is not None
+    }
+    for slot_number, ref in _named_refs_for_text(rewritten, ref_list, kinds=("character", "location")):
+        if slot_number in seen_slots:
+            continue
+        seen_slots.add(slot_number)
+        refs.append(ref)
+    return rewritten, refs, dropped
+
+
 def resolve_shot_references(text, ref_list, ref_mode="auto ref2v", shot_index=0, handoff=None):
     """Compatibility wrapper for older tests and helper code.
 
@@ -6823,7 +6848,7 @@ class H3LongVideos:
                 for shot_index, gen in enumerate(gens):
                     shot_mode = beat_ref_mode_directive(beats[shot_index] if shot_index < len(beats) else "") or ref_mode
                     if shot_mode in ("where tagged", "auto ref2v") and any_tags_anywhere:
-                        if resolve_prompt_refs(gen, ref_slots)[1]:
+                        if resolve_tag_driven_prompt_refs(gen, ref_slots)[1]:
                             on.append(shot_index + 1)
                             tagged_used = True
                     else:
@@ -6918,7 +6943,7 @@ class H3LongVideos:
                 # The prompt itself says where each reference belongs: the shot whose
                 # text names <Picture N> gets image N, renumbered to match what that
                 # shot actually carries. Every untagged shot keeps its handoff.
-                gen_prompt, shot_refs, dropped = resolve_prompt_refs(gen_prompt, ref_slots)
+                gen_prompt, shot_refs, dropped = resolve_tag_driven_prompt_refs(gen_prompt, ref_slots)
                 for n in dropped:
                     if n not in ref_missing:
                         ref_missing.append(n)

tests/test_dumas_h3_longvideos.py

@@ -458,6 +458,39 @@ class DumasH3LongVideosHelperTests(unittest.TestCase):
         self.assertEqual([self.module._reference_image(ref) for ref in references], ["img2"])
         self.assertEqual(dropped, [])
 
+    def test_resolve_tag_driven_prompt_refs_keeps_untagged_shot_on_handoff(self):
+        refs = [
+            {"kind": "character", "image": "img1", "name": "Mara"},
+            {"kind": "location", "image": "img2", "name": "Hangar"},
+        ]
+
+        text, references, dropped = self.module.resolve_tag_driven_prompt_refs(
+            "Mara waits in the hangar.",
+            refs,
+        )
+
+        self.assertEqual(text, "Mara waits in the hangar.")
+        self.assertEqual(references, [])
+        self.assertEqual(dropped, [])
+
+    def test_resolve_tag_driven_prompt_refs_keeps_named_refs_on_tagged_shot(self):
+        refs = [
+            {"kind": "character", "image": "img1", "name": "Mara"},
+            {"kind": "location", "image": "img2", "name": "Hangar"},
+        ]
+
+        text, references, dropped = self.module.resolve_tag_driven_prompt_refs(
+            "Mara waits in the hangar near <Picture 2>.",
+            refs,
+        )
+
+        self.assertEqual(text, "Mara waits in the hangar near <Picture 1>.")
+        self.assertEqual(
+            [self.module._reference_image(ref) for ref in references],
+            ["img2", "img1"],
+        )
+        self.assertEqual(dropped, [])
+
     def test_resolve_shot_references_uses_named_characters_without_picture_tags(self):
         refs = [
             {"kind": "character", "image": "img1", "name": "Mara"},