Update pipeline/postprocess.py

pipeline/postprocess.py

@@ -167,6 +167,31 @@ def split_long_segment(seg: dict) -> list:
     return split_long_segment(seg1) + split_long_segment(seg2)
 
 
+def wrap_text(text: str, max_chars: int = 42) -> str:
+    """اگه متن بیشتر از max_chars بود، به دو خط تقسیم کن"""
+    if len(text) <= max_chars:
+        return text
+
+    words = text.split()
+    best_break = -1
+    best_diff  = float('inf')
+
+    current_len = 0
+    for i, word in enumerate(words):
+        current_len += len(word) + (1 if i > 0 else 0)
+        diff = abs(current_len - len(text) / 2)
+        if diff < best_diff:
+            best_diff  = diff
+            best_break = i + 1
+
+    if best_break <= 0 or best_break >= len(words):
+        return text
+
+    line1 = " ".join(words[:best_break])
+    line2 = " ".join(words[best_break:])
+    return f"{line1}\n{line2}"
+
+
 def build_srt(segments: list) -> str:
     lines = []
     index = 1
@@ -176,7 +201,8 @@ def build_srt(segments: list) -> str:
             continue
         start_tc = format_timestamp(seg["start"])
         end_tc   = format_timestamp(seg["end"])
-        lines.append(f"{index}\n{start_tc} --> {end_tc}\n{text}")
+        wrapped  = wrap_text(text)
+        lines.append(f"{index}\n{start_tc} --> {end_tc}\n{wrapped}")
         index += 1
     return "\n\n".join(lines) + "\n"