Update app.py

app.py

@@ -2,8 +2,6 @@ import streamlit as st
 from streamlit_player import st_player
 from transformers import pipeline
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
 
 
 def tester(text):
@@ -12,6 +10,28 @@ def tester(text):
   classifier = pipeline("sentiment-analysis", 'j-hartmann/emotion-english-distilroberta-base')
   results = classifier(text)
   st.write(results[0]['label'])
+  
+  if (results[0]['label']=="anger"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+    
+  elif (results[0]['label']=="disgust"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+    
+  elif (results[0]['label']=="fear"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+    
+  elif (results[0]['label']=="joy"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+    
+  elif (results[0]['label']=="sadness"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+    
+  elif (results[0]['label']=="surprise"):
+    st_player("https://youtu.be/CmSKVW1v0xM")
+  
+  
+  return results[0]['label']
+  #return results
 
 emo = st.text_input('Enter text here:')
 tester(emo)