Update app.py

app.py

@@ -1,9 +1,9 @@
 import os
 import tempfile
-import asyncio
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
 from faster_whisper import WhisperModel
+from pipeline.postprocess import process
 
 app = FastAPI()
 
@@ -43,41 +43,38 @@ async def transcribe(websocket: WebSocket):
         # مرحله ۳ — transcription
         await websocket.send_json({"status": "transcribing"})
 
-        segments, info = model.transcribe(
+        segments_raw, info = model.transcribe(
             tmp_path,
             language="en",
             word_timestamps=True,
             vad_filter=True,
         )
 
-        # مرحله ۴ — ساخت SRT
+        # مرحله ۴ — تبدیل به لیست قابل پردازش
         await websocket.send_json({"status": "processing"})
 
-        srt_entries = []
-        index = 1
-
-        for segment in segments:
-            words = list(segment.words)
+        segments = []
+        for seg in segments_raw:
+            words = list(seg.words)
             if not words:
                 continue
-
-            start = words[0].start
-            end = words[-1].end
-            text = segment.text.strip()
-
-            srt_entries.append({
-                "index": index,
-                "start": start,
-                "end": end,
-                "text": text,
-                "words": [{"word": w.word, "start": w.start, "end": w.end} for w in words]
+            segments.append({
+                "text": seg.text.strip(),
+                "start": words[0].start,
+                "end": words[-1].end,
+                "words": [
+                    {"word": w.word, "start": w.start, "end": w.end}
+                    for w in words
+                ],
             })
-            index += 1
 
-        # مرحله ۵ — ارسال نتیجه
+        # مرحله ۵ — Post-Processing (اون ۲۰٪ طلایی)
+        srt_content = process(segments)
+
+        # مرحله ۶ — ارسال نتیجه
         await websocket.send_json({
             "status": "done",
-            "segments": srt_entries,
+            "srt": srt_content,
             "language": info.language,
             "duration": info.duration,
         })