Update app.py

app.py

@@ -166,6 +166,17 @@ async def predict(files: List[UploadFile] = File(...)):
             # Preprocess the audio file
             waveform = preprocess_audio(temp_audio_path)
 
+            rms = torch.sqrt(torch.mean(waveform**2)).item()
+            volume = "normal"
+
+            if rms < 0.0005:
+                volume = "silent"
+            elif rms < 0.01:
+                volume = "augmented"
+                waveform = waveform * 10  # Apply gain boost
+
+            logger.info(f"Volume state for {file.filename}: {volume}")
+
             # Perform inference
             label, confidence = infer(model_list, waveform)
 
@@ -181,6 +192,7 @@ async def predict(files: List[UploadFile] = File(...)):
                 "label": "Genuine" if label == 0 else "Spoof",
                 "confidence": confidence,
                 "status": "success",
+                "volume":volume
             }
             responses.append(response)