Update app.py

app.py

@@ -3,6 +3,8 @@ import asyncio
 import tempfile
 import torch
 import torchaudio
+
+torch.set_num_threads(2)  # جلوی رقابت بی‌فایده‌ی رشته‌ها روی CPU محدود رو می‌گیره
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
 from faster_whisper import WhisperModel
@@ -199,7 +201,7 @@ def run_transcription(audio_path):
 
 def run_diarization(audio_path):
     """فقط تشخیص گوینده - بلاک‌کننده، توی یه thread جدا اجرا میشه"""
-    return diarize_pipeline(audio_path)
+    return diarize_pipeline(audio_path, max_speakers=8)
 
 
 def run_alignment_pass(segments_raw, align_waveform, diarization):
@@ -234,11 +236,16 @@ def run_alignment_pass(segments_raw, align_waveform, diarization):
     return segments
 
 
-async def run_with_heartbeat(websocket, blocking_fn, *args, status, detail):
+class ProcessingTimeout(Exception):
+    pass
+
+
+async def run_with_heartbeat(websocket, blocking_fn, *args, status, detail, max_seconds=240):
     """
     یه تابع سنگین و بلاک‌کننده رو توی یه thread جدا اجرا می‌کنه، و تا وقتی تموم بشه،
     هر چند ثانیه یه پیام "هنوز کار می‌کنم" روی همون status می‌فرسته - این هم جلوی
     قطع‌شدن اتصال به‌خاطر سکوت طولانی رو می‌گیره، هم UI رو زنده نگه می‌داره.
+    اگه از max_seconds بیشتر طول بکشه، به‌جای قفل ابدی، خطای مشخص میده.
     """
     task = asyncio.create_task(asyncio.to_thread(blocking_fn, *args))
     elapsed = 0
@@ -248,6 +255,11 @@ async def run_with_heartbeat(websocket, blocking_fn, *args, status, detail):
             await asyncio.wait_for(asyncio.shield(task), timeout=HEARTBEAT_SEC)
         except asyncio.TimeoutError:
             elapsed += HEARTBEAT_SEC
+            if elapsed >= max_seconds:
+                task.cancel()
+                raise ProcessingTimeout(
+                    f"مرحله‌ی «{detail}» بیشتر از {max_seconds} ثانیه طول کشید و متوقف شد."
+                )
             await websocket.send_json({
                 "status": status,
                 "detail": f"{detail} ({elapsed} ثانیه)",
@@ -359,6 +371,12 @@ async def transcribe(websocket: WebSocket):
 
     except WebSocketDisconnect:
         print("WebSocket disconnected by client")
+    except ProcessingTimeout as e:
+        print(f"TIMEOUT: {e}")
+        try:
+            await websocket.send_json({"status": "error", "message": str(e)})
+        except Exception:
+            pass
     except Exception as e:
         print(f"ERROR: {type(e).__name__}: {e}")
         import traceback