Update app.py

app.py

@@ -13,14 +13,31 @@ def get_response(question):
 
     return assistant_answer.content
 
+def get_sentiment(user_input, nlp):
+    result = nlp(user_input)
+    sentiment = ""
+    if (result[0]['label'] == '1 star'):
+        sentiment = 'very negative'
+    elif (result[0]['label'] == '2 star'):
+        sentiment = 'negative'
+    elif (result[0]['label'] == '3 stars'):
+        sentiment = 'neutral'
+    elif (result[0]['label'] == '4 stars'):
+        sentiment = 'positive'
+    else:
+        sentiment = 'very positive'
 
+    prob = result[0]['score']
+
+    return sentiment, prob
 
 
 
 # open ai
 chat = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
 
-
+# hugging-face model
+nlp = pipeline(task='sentiment-analysis', model='nlptown/bert-base-multilingual-uncased-sentiment')
 
 
 
@@ -44,6 +61,54 @@ if user_input := st.chat_input("Say something"):
     # add assistant input to history
     st.session_state.messages.append({"role": "assistant", "content": assistant_input})
 
+
+
+    # sentiment analysis
+    sentiment, prob = get_sentiment(user_input, nlp)
+    if sentiment == "very negative":
+        rain( 
+        emoji="😭", 
+        font_size=20,  # the size of emoji 
+        falling_speed=3,  # speed of raining 
+        animation_length="infinite",  # for how much time the animation will happen 
+        )         
+    elif sentiment == "negative":
+        rain( 
+        emoji="🥺", 
+        font_size=20,  # the size of emoji 
+        falling_speed=3,  # speed of raining 
+        animation_length="infinite",  # for how much time the animation will happen 
+        ) 
+    elif sentiment == "neutral":
+        rain( 
+        emoji="😐", 
+        font_size=20,  # the size of emoji 
+        falling_speed=3,  # speed of raining 
+        animation_length="infinite",  # for how much time the animation will happen 
+        ) 
+    elif sentiment == "positive":
+        rain( 
+        emoji="😊", 
+        font_size=20,  # the size of emoji 
+        falling_speed=3,  # speed of raining 
+        animation_length="infinite",  # for how much time the animation will happen 
+        ) 
+    elif sentiment == "very positive":
+        rain( 
+        emoji="🤩", 
+        font_size=20,  # the size of emoji 
+        falling_speed=3,  # speed of raining 
+        animation_length="infinite",  # for how much time the animation will happen 
+        ) 
+
+
+
+
+
+
+
+    
+
 # display chat history
 for message in st.session_state.messages:
      with st.chat_message(message["role"]):