Spaces:

kowalsky
/

fake_audio_detection

Runtime error

App Files Files Community

kowalsky commited on Aug 27, 2024

Commit

8a4ec28

1 Parent(s): 3d055a1

added not real time detection option

Browse files

Files changed (2) hide show

main.py +33 -1
templates/index.html +63 -4

main.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 os.environ['NUMBA_CACHE_DIR'] = '/tmp/'
-from fastapi import FastAPI, WebSocket, WebSocketDisconnect, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
@@ -138,6 +138,38 @@ async def stop_detection():
     is_detecting = False
     return JSONResponse(content={'status': 'detection_stopped'})
 @app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
     await manager.connect(websocket)

 import os
 os.environ['NUMBA_CACHE_DIR'] = '/tmp/'
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect, Request, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
     is_detecting = False
     return JSONResponse(content={'status': 'detection_stopped'})
+@app.post("/upload_audio/")
+async def upload_audio(file: UploadFile = File(...)):
+    try:
+        # Read the uploaded file
+        audio_data = await file.read()
+        # Convert the audio file to the appropriate format for processing
+        audio_segment = AudioSegment.from_file(io.BytesIO(audio_data), format=file.filename.split('.')[-1])
+        wav_io = io.BytesIO()
+        audio_segment.export(wav_io, format="wav")
+        wav_io.seek(0)
+        audio, sr = sf.read(wav_io, dtype='float32')
+        # If audio has more than one channel, average them
+        if audio.ndim > 1:
+            audio = np.mean(audio, axis=1)
+        # Extract features
+        features = extract_features(audio)
+        features = scaler.transform(features.reshape(1, -1))
+        # Predict using the loaded model
+        prediction = model.predict(features)
+        is_fake = prediction[0]
+        result = 'fake' if is_fake else 'real'
+        return JSONResponse(content={'status': 'success', 'result': result})
+    except Exception as e:
+        logger.error(f"Failed to process audio file: {e}")
+        return JSONResponse(content={'status': 'error', 'message': str(e)}, status_code=500)
 @app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):
     await manager.connect(websocket)

templates/index.html CHANGED Viewed

@@ -74,6 +74,26 @@
             border-radius: 5px;
             display: none;
         }
     </style>
 </head>
 <body>
@@ -86,6 +106,13 @@
     <div id="console"></div>
     <div id="notification">Fake Call Detected!</div>
     <script>
         let isDetecting = false;
         const startButton = document.getElementById('startButton');
@@ -131,8 +158,8 @@
                     }
                 };
-                const response = await fetch(`${window.location.origin}/start_detection`, {
-                //const response = await fetch(`http://localhost:7860/start_detection`, {
                     method: 'POST',
                 });
@@ -143,8 +170,8 @@
                 const result = await response.json();
                 logMessage(`Status: ${result.status}`, 'info');
-                websocket = new WebSocket(`wss://${window.location.host}/ws`);
-                // websocket = new WebSocket(`ws://localhost:7860/ws`);
                 websocket.onmessage = async function(event) {
                     const data = event.data;
                     if (data === 'global-fake') {
@@ -202,6 +229,38 @@
         startButton.addEventListener('click', startDetection);
         stopButton.addEventListener('click', stopDetection);
     </script>
 </body>
 </html>

             border-radius: 5px;
             display: none;
         }
+        #fileUploadSection {
+            margin-top: 30px;
+            text-align: center;
+        }
+        #uploadButton {
+            padding: 10px 20px;
+            font-size: 16px;
+            cursor: pointer;
+            border: none;
+            border-radius: 5px;
+            background-color: #28a745;
+            color: #fff;
+            transition: background-color 0.3s;
+        }
+        #uploadButton:hover {
+            background-color: #218838;
+        }
+        #fileInput {
+            margin-bottom: 10px;
+        }
     </style>
 </head>
 <body>
     <div id="console"></div>
     <div id="notification">Fake Call Detected!</div>
+    <div id="fileUploadSection">
+        <h2>Upload Audio for Detection</h2>
+        <input type="file" id="fileInput" accept="audio/*">
+        <button id="uploadButton">Upload and Detect</button>
+        <p id="uploadResult"></p>
+    </div>
     <script>
         let isDetecting = false;
         const startButton = document.getElementById('startButton');
                     }
                 };
+                // const response = await fetch(`${window.location.origin}/start_detection`, {
+                const response = await fetch(`http://localhost:7860/start_detection`, {
                     method: 'POST',
                 });
                 const result = await response.json();
                 logMessage(`Status: ${result.status}`, 'info');
+                // websocket = new WebSocket(`wss://${window.location.host}/ws`);
+                websocket = new WebSocket(`ws://localhost:7860/ws`);
                 websocket.onmessage = async function(event) {
                     const data = event.data;
                     if (data === 'global-fake') {
         startButton.addEventListener('click', startDetection);
         stopButton.addEventListener('click', stopDetection);
+        const fileInput = document.getElementById('fileInput');
+        const uploadButton = document.getElementById('uploadButton');
+        const uploadResult = document.getElementById('uploadResult');
+        uploadButton.addEventListener('click', async () => {
+            const file = fileInput.files[0];
+            if (!file) {
+                uploadResult.textContent = 'Please select a file first.';
+                return;
+            }
+            const formData = new FormData();
+            formData.append('file', file);
+            try {
+                // const response = await fetch(`${window.location.origin}/upload_audio/`, {
+                const response = await fetch(`http://localhost:7860/upload_audio/`, {
+                    method: 'POST',
+                    body: formData,
+                });
+                if (!response.ok) {
+                    throw new Error('Network response was not ok');
+                }
+                const result = await response.json();
+                uploadResult.textContent = `Detected audio is: ${result.result}`;
+            } catch (error) {
+                uploadResult.textContent = `Error: ${error.message}`;
+            }
+        });
     </script>
 </body>
 </html>