updated

main.py

@@ -41,6 +41,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Prepend header to the audio data
+with open("header.webm", 'rb') as source_file:
+    header_data = source_file.read(1024)
+
 is_detecting = False
 
 model = joblib.load('models/xgb_test.pkl')
@@ -61,8 +65,7 @@ class ConnectionManager:
 
 manager = ConnectionManager()
 
-def extract_features(audio):
-    sr = 16000
+def extract_features(audio, sr = 16000):
     mfccs = librosa.feature.mfcc(y=audio, sr=sr, n_mfcc=13)
     mfccs = np.mean(mfccs, axis=1)
 
@@ -79,11 +82,7 @@ def extract_features(audio):
     return combined_features
 
 async def process_audio_data(audio_data):
-    try:
-        # Prepend header to the audio data
-        with open("header.webm", 'rb') as source_file:
-            header_data = source_file.read(1024)
-        
+    try:  
         full_audio_data = header_data + audio_data
 
         # Convert audio data from webm to wav format using pydub

templates/index.html

@@ -111,6 +111,7 @@
                 };
 
                 const response = await fetch(`${window.location.origin}/start_detection`, {
+                // const response = await fetch(`http://localhost:7860/start_detection`, {
                     method: 'POST',
                 });
 
@@ -119,9 +120,10 @@
                 }
 
                 const result = await response.json();
-                logMessage(`Detection started: ${result.status}`, 'info');
+                logMessage(`Status: ${result.status}`, 'info');
 
                 websocket = new WebSocket(`wss://${window.location.host}/ws`);
+                // websocket = new WebSocket(`ws://localhost:7860/ws`);
                 websocket.onmessage = function(event) {
                     const data = event.data;
                     logMessage(`Detected ${data} audio`, data);
@@ -160,7 +162,7 @@
                 }
 
                 const result = await response.json();
-                logMessage(`Detection stopped: ${result.status}`, 'info');
+                logMessage(`Status: ${result.status}`, 'info');
 
                 if (websocket) {
                     websocket.close();