j

main.py

@@ -27,10 +27,6 @@ async def get(request: Request):
 def health_check():
     return {"status": "ok"}
 
-# @app.get("/")
-# def read_root():
-#     return {"status": "ok"}
-
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -39,9 +35,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-duration = 2
-sample_rate = 16000
-
 is_detecting = False
 detection_thread = None
 
@@ -82,57 +75,29 @@ def extract_features(audio):
     combined_features = np.hstack([mfccs, chroma, contrast, centroid])
     return combined_features
 
-async def audio_callback(indata, frames, time, status):
-    if status:
-        print(status)
-
-    audio_data = indata[:, 0]
-    print(f"Audio data: {audio_data[:10]}... (length: {len(audio_data)})")
-    logger.info(f"Audio data: {audio_data[:10]}... (length: {len(audio_data)})")
-    
-    features = extract_features(audio_data)
+async def process_audio_data(audio_data):
+    audio_np = np.frombuffer(audio_data, dtype=np.float32)
+    features = extract_features(audio_np)
     features = features.reshape(1, -1)
     prediction = model.predict(features)
     is_fake = prediction[0]
 
-    print(f"Prediction: {is_fake}")
-    logger.info(f"Prediction: {is_fake}")
-
     result = 'fake' if is_fake else 'real'
     
-    print(f"Detected {result} audio")
-    logger.info(f"Detected {result} audio")
-
     await manager.send_message(result)
 
-def detect_fake_audio():
-    global is_detecting
-    try:
-        with sd.InputStream(callback=lambda indata, frames, time, status: asyncio.run(audio_callback(indata, frames, time, status)), channels=1, samplerate=sample_rate, blocksize=int(sample_rate * duration)):
-            print("Listening...")
-            logger.info("Listening...")
-            while is_detecting:
-                sd.sleep(duration * 1000)
-    except Exception as e:
-        print(f"Exception: {str(e)}")
-        logger.info(f"Exception: {str(e)}")
-
 @app.post("/start_detection")
 async def start_detection():
-    global is_detecting, detection_thread
+    global is_detecting
 
     if not is_detecting:
         is_detecting = True
-        detection_thread = threading.Thread(target=detect_fake_audio)
-        detection_thread.start()
     return JSONResponse(content={'status': 'detection_started'})
 
 @app.post("/stop_detection")
 async def stop_detection():
-    global is_detecting, detection_thread
+    global is_detecting
     is_detecting = False
-    if detection_thread:
-        detection_thread.join()
     return JSONResponse(content={'status': 'detection_stopped'})
 
 @app.websocket("/ws")
@@ -140,9 +105,10 @@ async def websocket_endpoint(websocket: WebSocket):
     await manager.connect(websocket)
     try:
         while True:
-            await websocket.receive_text()
+            data = await websocket.receive_bytes()
+            await process_audio_data(data)
     except WebSocketDisconnect:
         manager.disconnect(websocket)
 
 if __name__ == '__main__':
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)

templates/index.html

@@ -81,7 +81,7 @@
                 const result = await response.json();
                 logMessage(`Detection started: ${result.status}`, 'info');
 
-                websocket = new WebSocket(`wss://${window.location.host}/ws`);
+                websocket = new WebSocket(`ws://${window.location.host}/ws`);
                 websocket.onmessage = function(event) {
                     const data = event.data;
                     logMessage(`Detected ${data} audio`, data);