Update app.py

app.py

@@ -89,42 +89,41 @@ def generate_transcript(detection_result, ocr_result):
         ocr_result
     )
 
-def generate_structural_dialogue(detection_result):
+def generate_structural_dialogue(detection_result, threshold=0.4):
     texts = detection_result.get("texts", [])
-    associations = detection_result.get(
-        "character_text_associations", []
+    characters = detection_result.get("characters", [])
+    scores = detection_result.get(
+        "text_character_matching_scores", []
     )
 
     dialogue_lines = []
 
-    # agrupar associações por texto
-    assoc_by_text = {}
-    for assoc in associations:
-        text_id = assoc["text_id"]
-        assoc_by_text.setdefault(text_id, []).append(assoc)
-
-    for text_id, text in enumerate(texts):
-        assoc_list = assoc_by_text.get(text_id, [])
-
-        if assoc_list:
-            # pega a associação mais forte
-            best = max(
-                assoc_list,
-                key=lambda x: x["score"]
-            )
-            line = (
-                f"Text {text_id} → "
-                f"Character {best['character_id']} "
-                f"(score: {best['score']:.2f})"
+    for text_id in range(len(texts)):
+        if text_id < len(scores) and scores[text_id]:
+            char_scores = scores[text_id]
+            best_char = max(
+                range(len(char_scores)),
+                key=lambda i: char_scores[i]
             )
+            best_score = char_scores[best_char]
+
+            if best_score >= threshold:
+                line = (
+                    f"Text {text_id} → "
+                    f"Character {best_char} "
+                    f"(score: {best_score:.2f})"
+                )
+            else:
+                line = f"Text {text_id} → Narration / Uncertain"
         else:
-            line = f"Text {text_id} → No character (narration/SFX)"
+            line = f"Text {text_id} → Narration / Uncertain"
 
         dialogue_lines.append(line)
 
     return "\n".join(dialogue_lines)
 
 
+
 # ===============================
 # UI
 # ===============================