Update app.py

app.py

@@ -17,7 +17,10 @@ st.write("Sentiment classification: positive, neutral, negative")
 text = st.text_area("Enter the customer review", "")
 
 def sentiment_class(text):
-    print("Input text:", text)
+    # Check if the input text is empty or contains only whitespace
+    if not text.strip():
+        return None
+
     results = sentiment_classifier(text)[0]
     max_score = float('-inf')
     max_label = ''
@@ -28,22 +31,24 @@ def sentiment_class(text):
     return max_score, max_label
 
 def summarize_text(text):
-    print("Text for summarization:", text)
     results = summarizer(text)[0]['summary_text']
     return results
 
 # Perform sentiment analysis when the user clicks the "Classify Sentiment" button
 if st.button("Classify Sentiment"):
     # Check if the user has entered a review
-    if text is None or text.strip() == '':
+    if not text.strip():
         st.warning("Please enter a customer review first.")
     else:
         # Perform sentiment analysis on the input text
         sentiment_result = sentiment_class(text)
-        st.write("This review sentiment is", sentiment_result[1])
-        st.write("Prediction score is", sentiment_result[0])
-
-        # Perform text summarization when the review sentiment is classified as negative
-        if sentiment_result[1] == 'negative':
-            summary = summarize_text(text)
-            st.write("Review summary:", summary)
+        if sentiment_result is not None:
+            st.write("This review sentiment is", sentiment_result[1])
+            st.write("Prediction score is", sentiment_result[0])
+
+            # Perform text summarization when the review sentiment is classified as negative
+            if sentiment_result[1] == 'negative':
+                summary = summarize_text(text)
+                st.write("Review summary:", summary)
+        else:
+            st.warning("Please enter a non-empty customer review.")