Update output.py

output.py

@@ -7,13 +7,14 @@ Aphasia classification inference (cleaned).
 - Adds predict_from_chajson(json_path, ...) helper
 """
 
-import json
+import json as _json
 import os
 import math
 from dataclasses import dataclass
 from typing import Dict, List, Optional, Tuple
 from collections import defaultdict
 
+
 import numpy as np
 import torch
 import torch.nn as nn
@@ -594,6 +595,22 @@ def predict_from_chajson(model_dir: str, chajson_path: str, output_file: Optiona
         pass
     return out
 
+def format_result(pred: dict, style: str = "json") -> str:
+    """Back-compat formatter. 'pred' is the dict returned by predict_*."""
+    if style == "json":
+        return _json.dumps(pred, ensure_ascii=False, indent=2)
+    # simple text summary
+    if isinstance(pred, dict) and "summary" in pred:
+        s = pred["summary"]
+        lines = [
+            f"Total sentences: {pred.get('total_sentences', 0)}",
+            f"Avg confidence: {s.get('average_confidence', 'N/A')}",
+            f"Avg fluency: {s.get('average_fluency_score', 'N/A')}",
+            f"Most common: {s.get('most_common_prediction', 'N/A')}",
+        ]
+        return "\n".join(lines)
+    return str(pred)
+
 
 # ---------- CLI ----------