app.py

import streamlit as st
from streamlit_player import st_player
from transformers import pipeline
from IPython.display import YouTubeVideo
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 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)      
    #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"):
      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 (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")
    

  elif (results[0]['label']=="sadness"):
    
    #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"):
    st_player("https://youtu.be/CmSKVW1v0xM")
    st.write("check out this [link](https://open.spotify.com/playlist/4yXfnhz0BReoVfwwYRtPBm)")

  elif (results[0]['label']=="love"):
    st_player("https://www.youtube.com/watch?v=XVhEm62Uqog")
  
  return results[0]['label']
  #return results

emo = st.text_input('This application detects the emotion in your text input and suggests a song that matches it. Please enter text below to try:')
st.write('Examples: Today, I felt good!, I feel energetic')
tester(emo)