1

app.py

@@ -6,11 +6,19 @@ import numpy as np
 import tempfile
 import os
 import warnings
+from pydub import AudioSegment
 
 warnings.filterwarnings("ignore")
 
 app = FastAPI()
 
+def convert_mp3_to_wav(mp3_path):
+    # Convert MP3 to WAV
+    sound = AudioSegment.from_mp3(mp3_path)
+    wav_path = mp3_path.replace(".mp3", ".wav")
+    sound.export(wav_path, format="wav")
+    return wav_path
+
 def extract_audio_features(audio_file_path):
     # Load the audio file using soundfile
     waveform, sample_rate = sf.read(audio_file_path)
@@ -24,7 +32,6 @@ def extract_audio_features(audio_file_path):
     mfccs = np.mean(np.abs(np.fft.fft(waveform)[:13]), axis=0)  # Simplified MFCC-like features
 
     # Placeholder for speech rate and fundamental frequency
-    # Speech rate and pitch extraction would require more complex DSP techniques or external libraries.
     speech_rate = 4.0  # Arbitrary placeholder value for speech rate
     f0 = np.mean(np.abs(np.diff(waveform))) * sample_rate / (2 * np.pi)  # Rough pitch estimate
 
@@ -77,27 +84,31 @@ async def analyze_stress(
     # Handle audio file analysis
     if file or file_path:
         if file:
-            if not file.filename.endswith(".wav"):
-                raise HTTPException(status_code=400, detail="Only .wav files are supported.")
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
+            if not (file.filename.endswith(".wav") or file.filename.endswith(".mp3")):
+                raise HTTPException(status_code=400, detail="Only .wav and .mp3 files are supported.")
+            with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[-1]) as temp_file:
                 temp_file.write(await file.read())
-                temp_wav_path = temp_file.name
+                temp_audio_path = temp_file.name
         else:
-            if not file_path.endswith(".wav"):
-                raise HTTPException(status_code=400, detail="Only .wav files are supported.")
+            if not (file_path.endswith(".wav") or file_path.endswith(".mp3")):
+                raise HTTPException(status_code=400, detail="Only .wav and .mp3 files are supported.")
             if not os.path.exists(file_path):
                 raise HTTPException(status_code=400, detail="File path does not exist.")
-            temp_wav_path = file_path
+            temp_audio_path = file_path
+
+        # Convert MP3 to WAV if needed
+        if temp_audio_path.endswith(".mp3"):
+            temp_audio_path = convert_mp3_to_wav(temp_audio_path)
 
         try:
-            result = analyze_voice_stress(temp_wav_path)
+            result = analyze_voice_stress(temp_audio_path)
             return JSONResponse(content=result)
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))
         finally:
             # Clean up temporary files
             if file:
-                os.remove(temp_wav_path)
+                os.remove(temp_audio_path)
 
     # Handle text analysis
     elif text: