Update app.py

app.py

@@ -6,17 +6,40 @@ from langchain.vectorstores import FAISS
 import tempfile 
 from streamlit_chat import message
 import streamlit as st
+import nltk
+from nltk.sentiment import SentimentIntensityAnalyzer
+
+def get_sentiment(text):
+    sid = SentimentIntensityAnalyzer()
+    sentiment_scores = sid.polarity_scores(text)
+    compound_score = sentiment_scores['compound']
+    if compound_score >= 0.05:
+        return 'positive'
+    elif compound_score <= -0.05:
+        return 'negative'
+    else:
+        return 'neutral'
+
+def add_sentiment_emoji(text, sentiment):
+    emoji_mapping = {
+        'positive': '😄',
+        'negative': '😞',
+        'neutral': '😐'
+    }
+    emoji = emoji_mapping.get(sentiment, '')
+    return f"{text} {emoji}"
 
 import os 
 import sys
 import pandas as pd
 
+
 def conversational_chat(query):    
-    result = chain({"question": query, 
-    "chat_history": st.session_state['history']})
+    result = chain({"question": query, "chat_history": st.session_state['history']})
     st.session_state['history'].append((query, result["answer"]))
-
-    return result["answer"]
+    sentiment = get_sentiment(result["answer"])
+    answer_with_emoji = add_sentiment_emoji(result["answer"], sentiment)
+    return answer_with_emoji
 
 
 user_api_key = st.sidebar.text_input(
@@ -73,4 +96,7 @@ if user_api_key is not None and user_api_key.strip() != "":
             with response_container:
                 for i in range(len(st.session_state['generated'])):
                     message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="big-smile")
-                    message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+                    message(st.session_state["generated"][i], key=str(i), avatar_style="thumbs")
+
+else:
+    st.text("Please enter your OpenAI API key above.")