Update app.py

app.py

@@ -1,26 +1,20 @@
-import streamlit as st
+import streamlit.components.v1 as components
 from streamlit_player import st_player
 from transformers import pipeline
-from IPython.display import YouTubeVideo
+import streamlit as st
 import random 
 
 
-import streamlit as st
-import streamlit.components.v1 as components
-
 def tester(text):
   #classifier = pipeline("sentiment-analysis", model='arpanghoshal/EmoRoBERTa')
   #classifier = pipeline("sentiment-analysis", model='cardiffnlp/twitter-roberta-base-emotion')
   #classifier = pipeline("sentiment-analysis", 'j-hartmann/emotion-english-distilroberta-base')
   classifier = pipeline("sentiment-analysis", model='bhadresh-savani/distilbert-base-uncased-emotion')
-  
- 
   results = classifier(text)
-  #st.write(results[0]['label'])
-  
+
   
-  if (results[0]['label']=="joy"):
-    a = random.randint(1,7)
+  if (results[0]['label']=="joy"): #songs for joy emotion
+    a = random.randint(1,7)  #shuffle
     if a == 1: #Rick Astley - Never Gonna Give You Up
       components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/dQw4w9WgXcQ?&autoplay=1" title="YouTube video player" frameborder="0"       allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
     elif a == 2: #WALK THE MOON - Shut Up and Dance
@@ -35,15 +29,10 @@ def tester(text):
       components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/zA_deMWtz8g?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
     elif a == 7: #The Itchyworms - Penge Naman Ako N'yan
       components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/cYLhKaSY6is?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
-    #elif a == 8:
-    #  components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/I_izvAbhExY?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
-    #elif a == 9:
-    #  components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/I_izvAbhExY?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)    
-    #elif a == 10:
-    #  components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/I_izvAbhExY?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)          
+        
       
       
-    elif (results[0]['label']=="anger"):
+    elif (results[0]['label']=="anger"): #songs for anger emotion
       components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/dQw4w9WgXcQ?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
     
     
@@ -56,15 +45,26 @@ def tester(text):
     st_player("https://www.youtube.com/watch?v=iyEUvUcMHgE?&autoplay=1")
     
 
-  elif (results[0]['label']=="sadness"):
+  elif (results[0]['label']=="sadness"): #songs for sadness emotion
+    a = random.randint(1,7)
+    if a == 1: #Join the Club - Nobela
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/SZADEHP2yFg?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
+    elif a ==2: #Pagsamo - Arthur Nery
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/Hx4-GMWWb7Q?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
+    elif a ==3: #SpongeCola - Jeepney
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/YE6bbpEY4nU?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
+    elif a ==4: #The Walters -- I Love You So
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/NwFVSclD_uc?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
+    elif a ==2: #Pagsamo - Arthur Nery
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/Hx4-GMWWb7Q?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
+    elif a ==2: #Pagsamo - Arthur Nery
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/Hx4-GMWWb7Q?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
+    elif a ==2: #Pagsamo - Arthur Nery
+      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/Hx4-GMWWb7Q?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)      
+
+    
     
-    #video = YouTubeVideo("1k8craCGpgs")
-    #display(video)
-    #st_player("https://www.youtube.com/watch?v=BZsXcc_tC-o")
-    # embed streamlit docs in a streamlit app
-    #components.html("""<iframe width=560" height="315" src="https://www.youtube.com/embed/NwFVSclD_uc?&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
     
-    components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/videoseries?list=PLMC9KNkIncKtPzgY-5rmhvj7fax8fdxoj&autoplay=1" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>""",width=560,height=325)
     
     
   elif (results[0]['label']=="surprise"):