Spaces:
Sleeping
Sleeping
| import os | |
| import streamlit as st | |
| from langchain.chat_models import ChatOpenAI | |
| from langchain.schema import (AIMessage,HumanMessage,SystemMessage) | |
| from transformers import pipeline | |
| from streamlit_extras.let_it_rain import rain | |
| def get_response(question): | |
| st.session_state.sessionMessages.append(HumanMessage(content=question)) | |
| assistant_answer = chat(st.session_state.sessionMessages ) | |
| st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content)) | |
| 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 stars'): | |
| 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=1) | |
| # hugging-face model | |
| nlp = pipeline(task='sentiment-analysis', model='nlptown/bert-base-multilingual-uncased-sentiment') | |
| st.set_page_config(page_title="Home Assistant", page_icon=":robot:") | |
| st.header("Knock knock, It's meeee the JOKER!") | |
| if "sessionMessages" not in st.session_state: | |
| st.session_state.sessionMessages = [SystemMessage(content="You have an evil personality like Joker from Batman")] | |
| if "messages" not in st.session_state: | |
| st.session_state.messages = [] | |
| if user_input := st.chat_input("Say something"): | |
| assistant_input = get_response(user_input) | |
| sentiment, prob = get_sentiment(user_input, nlp) | |
| sentiment_analysis = f" (sentiment:{sentiment},score:{prob})" | |
| # add user input to history | |
| st.session_state.messages.append({"role": "user", "content": user_input}) | |
| # add assistant input to history | |
| st.session_state.messages.append({"role": "assistant", "content": assistant_input}) | |
| # sentiment analysis | |
| 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"]): | |
| st.markdown(message["content"]) |