Create app.py

app.py

@@ -0,0 +1,65 @@
+from flask import Flask, request, jsonify
+import requests
+
+app = Flask(__name__)
+
+API_URL = "https://api-inference.huggingface.co/models/ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition"
+headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
+
+def query_api(audio_file):
+    with open(audio_file, "rb") as f:
+        response = requests.post(API_URL, headers=headers, data=f)
+    return response.json()
+
+def compute_neg_rate(emotion_data):
+    neg_emotions = ["anger", "disgust", "fear", "sadness"]
+    neg_rate = sum([emotion['score'] for emotion in emotion_data if emotion['label'].lower() in neg_emotions])
+    return neg_rate * 100
+
+def analyze_depression(neg_rate):
+    response = ""
+    if neg_rate <= 30.0:
+        if neg_rate <= 5.0:
+            response = "just chill and relax"
+        else:
+            response = "***   MILD DEPRESSION  ***"
+        if neg_rate <= 10.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: Everything is a-okay! There's absolutely nothing wrong 😁😁".format(round(neg_rate) / 10)
+        elif neg_rate > 10.0 and neg_rate <= 20.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: You are a bit frustrated and disappointed\n\tBut you're easily distracted and cheered with little effort 😁😁".format(round(neg_rate) / 10)
+        elif neg_rate > 20.0 and neg_rate <= 30.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: Things are bothering you but you're coping up\n\tYou might be overtired and hungry\n\tThe emotional equivalent of a headache 😁😁".format(round(neg_rate) / 10)
+    elif neg_rate > 30.0 and neg_rate <= 60.0:
+        response = "***   MODERATE DEPRESSION  ***"
+        if neg_rate > 30.0 and neg_rate <= 40.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: Today is slightly a bad day for you.\n\tYou still have the skills to get through it, but be gentle with yourself\n\tUse self-care strategies 😁😁".format(round(neg_rate) / 10)
+        elif neg_rate > 40.0 and neg_rate <= 50.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: Your mental health is starting to impact your everyday life.\n\tEasy things are becoming difficult".format(round(neg_rate) / 10)
+        elif neg_rate > 50.0 and neg_rate <= 60.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: You are not able to do things the way you usually do them due to your mental health.\n\tImpulsive and compulsive thoughts might be difficult to cope with".format(round(neg_rate) / 10)
+    elif neg_rate > 60.0 and neg_rate <= 100.0:
+        response = "***   SEVERE DEPRESSION  ***"
+        if neg_rate > 60.0 and neg_rate <= 70.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: You are losing interest in activities that used to be enjoyable.\n\tYou should definitely seek help\n\tThis is becoming serious ☹☹".format(round(neg_rate) / 10)
+        elif neg_rate > 70.0 and neg_rate <= 80.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: You can't ignore your struggles now, It's HIGH time!.\n\tYou may have issues sleeping, eating, having fun, socializing, and work/study\n\tYour mental health is affecting almost all parts of your life ☹☹".format(round(neg_rate) / 10)
+        elif neg_rate > 80.0 and neg_rate <= 90.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: You are at a critical point!!!.\n\tYou aren't functioning anymore and need urgent help.\n\tYou may be a risk to yourself or others if left untreated ☹☹".format(round(neg_rate) / 10)
+        elif neg_rate > 90.0 and neg_rate <= 100.0:
+            response += "\nDepression Scale Detected: {:.1f}\nMy Response: The worst mental and emotional distress possible.\n\tYou can't imagine things getting any better now and you might think it's all over for you (SUICIDE).\n\tContact a crisis line or get started with treatment immediately. ☹☹".format(round(neg_rate) / 10)
+    return response
+
+@app.route("/analyze", methods=["POST"])
+def analyze():
+    audio_file = request.files['file']
+    emotion_data = query_api(audio_file)
+    neg_rate = compute_neg_rate(emotion_data)
+    response = analyze_depression(neg_rate)
+    return jsonify({"response": response})
+
+@app.route("/")
+def home():
+    return "Emotion Recognition and Depression Analysis API"
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8000)