updated

app.py

@@ -1,5 +1,5 @@
 import os
-os.environ[ 'NUMBA_CACHE_DIR' ] = '/tmp/'
+os.environ['NUMBA_CACHE_DIR'] = '/tmp/'
 
 from fastapi import FastAPI, WebSocket, WebSocketDisconnect, Request
 from fastapi.middleware.cors import CORSMiddleware
@@ -13,7 +13,6 @@ import logging
 import io
 from pydub import AudioSegment
 from typing import List
-import asyncio
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -22,7 +21,7 @@ app = FastAPI()
 
 @app.get("/", response_class=HTMLResponse)
 async def get(request: Request):
-    logger.info("Saving the index page")
+    logger.info("Serving the index page")
     with open("templates/index.html") as f:
         html_content = f.read()
     return HTMLResponse(content=html_content, status_code=200)
@@ -40,36 +39,23 @@ app.add_middleware(
 )
 
 is_detecting = False
-detection_thread = None
 
 model = joblib.load('models/xgb_test.pkl')
-# s
+
 class ConnectionManager:
     def __init__(self):
         self.active_connections: List[WebSocket] = []
-        self.audio_buffers = {}
 
     async def connect(self, websocket: WebSocket):
         await websocket.accept()
         self.active_connections.append(websocket)
-        self.audio_buffers[websocket] = b''  # Initialize buffer for each connection
 
     def disconnect(self, websocket: WebSocket):
         self.active_connections.remove(websocket)
-        del self.audio_buffers[websocket]  # Clean up buffer
 
     async def send_message(self, websocket: WebSocket, message: str):
         await websocket.send_text(message)
 
-    def add_to_buffer(self, websocket: WebSocket, data: bytes):
-        self.audio_buffers[websocket] += data  # Accumulate data in the buffer
-
-    def get_buffer(self, websocket: WebSocket) -> bytes:
-        return self.audio_buffers[websocket]
-
-    def clear_buffer(self, websocket: WebSocket):
-        self.audio_buffers[websocket] = b''  # Clear the buffer
-
 manager = ConnectionManager()
 
 def extract_features(audio):
@@ -89,19 +75,20 @@ def extract_features(audio):
     combined_features = np.hstack([mfccs, chroma, contrast, centroid])
     return combined_features
 
-async def process_audio_data(websocket: WebSocket):
-    audio_data = manager.get_buffer(websocket)
+async def process_audio_data(audio_data):
     try:
-        # Convert audio data from webm/ogg to wav format using pydub
-        audio_segment = AudioSegment.from_file(io.BytesIO(audio_data), format="webm")
-        logging.info("audio_segment worked")
+        # Prepend header to the audio data
+        with open("header.webm", 'rb') as source_file:
+            header_data = source_file.read(1024)
+        
+        full_audio_data = header_data + audio_data
+
+        # Convert audio data from webm to wav format using pydub
+        audio_segment = AudioSegment.from_file(io.BytesIO(full_audio_data), format="webm")
         wav_io = io.BytesIO()
         audio_segment.export(wav_io, format="wav")
-        logging.info("audio_segment wav worked")
         wav_io.seek(0)
-        logging.info("wav io worked")
         audio, sr = sf.read(wav_io, dtype='float32')
-        logging.info("sf worked")
     except Exception as e:
         logger.error(f"Failed to read audio data: {e}")
         return
@@ -115,8 +102,7 @@ async def process_audio_data(websocket: WebSocket):
     is_fake = prediction[0]
     result = 'fake' if is_fake else 'real'
     
-    await manager.send_message(websocket, result)
-    manager.clear_buffer(websocket)
+    return result
 
 @app.post("/start_detection")
 async def start_detection():
@@ -138,8 +124,9 @@ async def websocket_endpoint(websocket: WebSocket):
     try:
         while True:
             data = await websocket.receive_bytes()
-            manager.add_to_buffer(websocket, data)
-            await process_audio_data(websocket)
+            result = await process_audio_data(data)
+            if result:
+                await manager.send_message(websocket, result)
     except WebSocketDisconnect:
         manager.disconnect(websocket)
 

templates/index.html

@@ -90,14 +90,12 @@
                     logMessage('WebSocket connection closed', 'info');
                 };
 
-                mediaRecorder.start(3000);  // Start recording with 3-second intervals
+                mediaRecorder.start(5000);  // Start recording with 3-second intervals
             } catch (error) {
                 logMessage(`Error: ${error.message}`, 'error');
             }
         }
 
-
-
         async function stopDetection() {
             if (!isDetecting) {
                 logMessage('Detection is not running...', 'info');
@@ -138,3 +136,4 @@
     </script>
 </body>
 </html>
+