Voice et bot modif

core/whisper_worker.py

@@ -0,0 +1,58 @@
+import os
+import logging
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+from typing import Optional
+
+logger = logging.getLogger('Superviseur')
+
+class WhisperWorker:
+    """Handles local STT transcription using faster-whisper."""
+    
+    def __init__(self, model_size="base", device="cpu"):
+        self.model_size = model_size
+        self.device = device
+        self.model = None
+        self.executor = ThreadPoolExecutor(max_workers=2)
+        self.initialized = False
+
+    def initialize(self):
+        """Load the model into RAM/VRAM."""
+        try:
+            from faster_whisper import WhisperModel
+            logger.info(f"◈ SYSTEM: Initializing Whisper ({self.model_size}) on {self.device}...")
+            
+            # On tente le chargement. Si device="cuda" échoue (AMD), on bascule sur CPU.
+            try:
+                self.model = WhisperModel(self.model_size, device=self.device, compute_type="int8")
+            except Exception as e:
+                if self.device != "cpu":
+                    logger.warning(f"◈ GPU failure ({e}). Falling back to CPU.")
+                    self.device = "cpu"
+                    self.model = WhisperModel(self.model_size, device="cpu", compute_type="int8")
+                else:
+                    raise e
+
+            self.initialized = True
+            logger.info(f"◈ SYSTEM: Whisper Model Loaded on {self.device}. Ear Protocol Active.")
+        except Exception as e:
+            logger.error(f"Failed to initialize Whisper: {e}")
+            self.initialized = False
+
+    async def transcribe(self, audio_path: str) -> Optional[str]:
+        """Transcribe an audio file asynchronously."""
+        if not self.initialized or not self.model:
+            return None
+
+        def _run():
+            try:
+                segments, info = self.model.transcribe(audio_path, beam_size=5)
+                text = " ".join([segment.text for segment in segments]).strip()
+                logger.debug(f"◈ Whisper: Transcription result: '{text}' (lang: {info.language})")
+                return text
+            except Exception as e:
+                logger.error(f"Transcription error: {e}")
+                return None
+
+        loop = asyncio.get_event_loop()
+        return await loop.run_in_executor(self.executor, _run)