sentiment analysis app

app.py

@@ -0,0 +1,17 @@
+from engine import SentimentAnalyzer
+import streamlit as st
+
+
+# Load the sentiment analysis model from Hugging Face
+sentiment_analysis = SentimentAnalyzer()
+
+# Define the Streamlit app interface
+st.title("User Sentiment Analysis")
+
+sentence = st.text_input("Enter a sentence:")
+
+# Perform sentiment analysis on the input sentence
+if sentence:
+    label = sentiment_analysis.get_sentiment(sentence)
+    # Display the sentiment analysis result to the user
+    st.write(f"Sentiment analysis result: {label}")

engine.py

@@ -0,0 +1,49 @@
+from transformers import pipeline
+
+
+class SentimentAnalyzer:
+    """Class for analyzing the sentiment of sentences
+    """
+
+    def __init__(self) -> None:
+        """initializes the class with sentiment analysis pipeline using the distilbert-base-uncased-finetuned-sst-2-english model
+        """
+        self.analyzer = pipeline(
+            "sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
+
+    def score_sentiment(self, sentence: str) -> float:
+        """Uses the analyzer to analyze the sentiment of the provided sentence
+
+        Parameters
+        ----------
+        sentence : str
+            a short sentence to be analyzed
+
+        Returns
+        -------
+        float
+            score of the sentiment from 0 to 1. Below 0.5 is negative, above is positive. 0.5 is neutral
+        """
+        return self.analyzer(sentence)[0]
+
+    def get_sentiment(self, sentence: str) -> str:
+        """returns the label of the sentiment provided
+
+        Parameters
+        ----------
+        sentence : str
+            a short sentence to be analyzed
+
+        Returns
+        -------
+        str
+            label of the sentiment wether it is positive, negative, or neutral
+        """
+        sentiment_score = self.score_sentiment(sentence)
+        return sentiment_score['label']
+
+
+if __name__ == "__main__":
+    sentence = "I love you"
+    sentiment_analyzer = SentimentAnalyzer()
+    print(sentiment_analyzer.get_sentiment(sentence))

requirements.txt

@@ -0,0 +1,53 @@
+altair==5.2.0
+attrs==23.2.0
+blinker==1.7.0
+cachetools==5.3.3
+certifi==2023.7.22
+charset-normalizer==3.3.2
+click==8.1.7
+filelock==3.13.1
+fsspec==2024.3.1
+gitdb==4.0.11
+GitPython==3.1.42
+huggingface-hub==0.21.4
+idna==3.4
+Jinja2==3.1.3
+jsonschema==4.21.1
+jsonschema-specifications==2023.12.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.2.1
+numpy==1.26.4
+packaging==23.2
+pandas==2.2.1
+pillow==10.2.0
+protobuf==4.25.3
+pyarrow==15.0.2
+pydeck==0.8.1b0
+Pygments==2.17.2
+python-dateutil==2.9.0.post0
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.34.0
+regex==2023.12.25
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.0
+safetensors==0.4.2
+six==1.16.0
+smmap==5.0.1
+streamlit==1.32.2
+sympy==1.12
+tenacity==8.2.3
+tokenizers==0.15.2
+toml==0.10.2
+toolz==0.12.1
+torch==2.2.1
+tornado==6.4
+tqdm==4.66.2
+transformers==4.39.1
+typing_extensions==4.10.0
+tzdata==2024.1
+urllib3==2.0.7