Update src/emotion_detector.py

src/emotion_detector.py

@@ -1,11 +1,16 @@
 # emotion_detector.py
 import torchaudio
 from speechbrain.inference.classifiers import AudioClassifier
+import os
+import tempfile
 
-# Load pretrained model
+# Use a temp directory that's guaranteed to be writable
+temp_dir = os.path.join(tempfile.gettempdir(), "emotion_model")
+
+# Load pretrained model into temp directory
 classifier = AudioClassifier.from_hparams(
     source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP",
-    savedir="pretrained_models/emotion-recognition"
+    savedir=temp_dir
 )
 
 EMOTION_EMOJIS = {
@@ -16,9 +21,9 @@ EMOTION_EMOJIS = {
     "fearful": "😨"
 }
 
-def detect_emotion(audio_file):
-    signal, fs = torchaudio.load(audio_file)
-    prediction = classifier.classify_file(audio_file)
-    emotion = prediction[3]  # This is the predicted label
+def detect_emotion(audio_path):
+    signal, fs = torchaudio.load(audio_path)
+    prediction = classifier.classify_file(audio_path)
+    emotion = prediction[3]  # Predicted label
     emoji = EMOTION_EMOJIS.get(emotion.lower(), "❓")
     return emotion, emoji