Update app.py

app.py

@@ -6,80 +6,41 @@ import random
 
 
 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)
 
-  
   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
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/6JCLY0Rlx6Q?&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: #Bee Gees - Stayin' Alive
-      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 == 4: #Lil Nas X - Industry Baby
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/4aNUdvctOok?&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 == 5: #Earth, Wind & Fire - Let's Groove
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/Lrle0x_DHBM?&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 == 6: #BTS - 「Dynamite」 / Guitar Cover
-      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)      
-        
-      
-      
+    with open('joyplaylist.txt') as f:
+      contents = f.read()
+    components.html(contents,width=560,height=325)   
+
   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)
-    
-    
-    
+    with open('angryplaylist.txt') as f:
+      contents = f.read()
+    components.html(contents,width=560,height=325)
     
   elif (results[0]['label']=="disgust"):
     st_player("https://www.youtube.com/watch?v=zWq2TT3ieGE")
     
   elif (results[0]['label']=="fear"):
-    st_player("https://www.youtube.com/watch?v=iyEUvUcMHgE?&autoplay=1")
+    with open('fearplaylist.txt') as f:
+      contents = f.read()
+    components.html(contents,width=560,height=325)
     
-
   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 ==5: #Coldplay - The Scientist
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/RB-RcX5DS5A?&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 ==6: #Wiz Khalifa - See You Again
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/cPyovQwFmhE?&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: #Ricky Montgomery - Mr Loverman
-      components.html("""<iframe width="560" height="315" src="https://www.youtube.com/embed/MxXM3H2VGy4?&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)      
+    with open('sadplaylist.txt') as f:
+      contents = f.read()
+    components.html(contents,width=560,height=325)
 
-    
-    
-    
-    
-    
   elif (results[0]['label']=="surprise"):
-    st_player("https://youtu.be/CmSKVW1v0xM")
-    st.write("check out this [link](https://open.spotify.com/playlist/4yXfnhz0BReoVfwwYRtPBm)")
+    st.write("gulat ka noh")
 
   elif (results[0]['label']=="love"):
-    #st_player("https://www.youtube.com/watch?v=XVhEm62Uqog")
-    
-    with open('sadplaylist.txt') as f:
+    with open('loveplaylist.txt') as f:
       contents = f.read()
     components.html(contents,width=560,height=325)
   
   return results[0]['label']
-  #return results