Update app.py

app.py

@@ -1,65 +1,86 @@
-from flask import Flask, request, jsonify
+import streamlit as st
 import requests
 
-app = Flask(__name__)
-
+# Define API details
 API_URL = "https://api-inference.huggingface.co/models/ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition"
 headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
 
+# Function to query the Hugging Face API
 def query_api(audio_file):
     with open(audio_file, "rb") as f:
         response = requests.post(API_URL, headers=headers, data=f)
     return response.json()
 
+# Function to compute the neg_rate from API response
 def compute_neg_rate(emotion_data):
-    neg_emotions = ["anger", "disgust", "fear", "sadness"]
+    neg_emotions = ["anger", "disgust", "fear", "sadness"]  # Example negative emotions
     neg_rate = sum([emotion['score'] for emotion in emotion_data if emotion['label'].lower() in neg_emotions])
-    return neg_rate * 100
+    return neg_rate * 100  # Scale to a percentage
 
+# Depression scale logic
 def analyze_depression(neg_rate):
-    response = ""
+    responses = []
     if neg_rate <= 30.0:
         if neg_rate <= 5.0:
-            response = "just chill and relax"
+            responses.append("just chill and relax")
         else:
-            response = "***   MILD DEPRESSION  ***"
+            responses.append("***   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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: Everything is a-okay! There's absolutely nothing wrong 😁😁\n\t\tYou're probably cuddling a fluffy kitten right now")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: You are a bit frustrated and disappointed\n\t\tBut you're easily distracted and cheered with little effort 😁😁")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: Things are bothering you but you're coping up\n\t\tYou might be overtired and hungry\n\t\tThe emotional equivalent of a headache 😁😁")
     elif neg_rate > 30.0 and neg_rate <= 60.0:
-        response = "***   MODERATE DEPRESSION  ***"
+        responses.append("***   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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: Today is slightly a bad day for you.\n\t\tYou still have the skills to get through it, but be gentle with yourself\n\t\tUse self-care strategies 😁😁")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: Your mental health is starting to impact your everyday life.\n\t\tEasy things are becoming difficult")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: You are not able to do things the way you usually do them due to your mental health.\n\t\tImpulsive and compulsive thoughts might be difficult to cope with")
     elif neg_rate > 60.0 and neg_rate <= 100.0:
-        response = "***   SEVERE DEPRESSION  ***"
+        responses.append("***   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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: You are losing interest in activities that used to be enjoyable.\n\t\tYou should definitely seek help\n\t\tThis is becoming serious ☹☹")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: You can't ignore your struggles now, It's HIGH time!.\n\t\tYou may have issues sleeping, eating, having fun, socializing, and work/study\n\t\tYour mental health is affecting almost all parts of your life ☹☹")
         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)
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: You are at a critical point!!!.\n\t\tYou aren't functioning anymore and need urgent help.\n\t\tYou may be a risk to yourself or others if left untreated ☹☹")
         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
+            responses.append(f"\nDepression Scale Detected: {round(neg_rate) / 10}\n")
+            responses.append("My Response: The worst mental and emotional distress possible.\n\t\tYou can't imagine things getting any better now and you might think it's all over for you (SUICIDE).\n\t\tContact a crisis line or get started with treatment immediately. ☹☹")
+    return "\n".join(responses)
+
+# Streamlit app
+def main():
+    st.title("Emotion Detection and Depression Scale Analysis")
+
+    audio_file = st.file_uploader("Upload an audio file", type=["wav"])
+    if audio_file:
+        st.audio(audio_file, format="audio/wav")
+
+        if st.button("Analyze"):
+            # Save uploaded file to a temporary file
+            with open("temp_audio.wav", "wb") as f:
+                f.write(audio_file.read())
 
-@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})
+            # Query API and compute neg_rate
+            emotion_data = query_api("temp_audio.wav")
+            neg_rate = compute_neg_rate(emotion_data)
 
-@app.route("/")
-def home():
-    return "Emotion Recognition and Depression Analysis API"
+            # Analyze and display results
+            result = analyze_depression(neg_rate)
+            st.text(result)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=8000)
+    main()