create app.py

app.py

@@ -0,0 +1,62 @@
+import streamlit as st
+from transformers import pipeline
+
+
+def main():
+    st.title("Sentiment analysis")
+
+    st.header("Add comment")
+    input = st.text_input("Enter a new comment:")
+    if st.button("Add"):
+        add_input_text(input)
+        result_list(input)
+        
+    display_comments()
+
+
+def add_input_text(input):
+    if input:
+        if 'input_text' not in st.session_state:
+            st.session_state.input_text = []
+        st.session_state.input_text.append(input)
+
+
+def result_list(input):
+    if input:
+        if 'result_list' not in st.session_state:
+            st.session_state.result_list = []
+        pipe = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
+        sentiment = pipe(input)
+        result = sentiment[0]['label']
+        st.session_state.result_list.append(result)
+
+
+def display_comments():
+    if 'result_list' in st.session_state:
+        st.header("Filter by Type")
+        filter_option = st.selectbox("Select type:", ["All", "Positive", "Negative"])
+
+
+        if filter_option == "All":
+            st.header(f"{len(st.session_state.result_list)} comments")
+        elif filter_option == "Positive":
+            st.header(f"{st.session_state.result_list.count('positive')} comments")        
+        elif filter_option == "Negative":
+            st.header(f"{st.session_state.result_list.count('negative')} comments")        
+
+        for id,result in enumerate(st.session_state.result_list):
+            if filter_option == "All":
+                # st.header(f"{len(st.session_state.result_list)} comments")
+                if result == 'positive':
+                    st.success(st.session_state.input_text[id])
+                else:
+                    st.error(st.session_state.input_text[id])
+            elif filter_option == "Positive" and result == 'positive':
+                # st.header(f"{st.session_state.result_list.count('positive')} comments")
+                st.success(st.session_state.input_text[id])
+            elif filter_option == "Negative" and result == 'negative':
+                st.error(st.session_state.input_text[id])
+
+
+if __name__ == "__main__":
+    main()