Update app.py

app.py

@@ -17,10 +17,6 @@ st.write("Sentiment classification: positive, neutral, negative")
 text = st.text_area("Enter the customer review", "")
 
 def sentiment_class(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 = ''
@@ -31,24 +27,21 @@ def sentiment_class(text):
     return max_score, max_label
 
 def summarize_text(text):
-    results = summarizer(text)[0]['summary_text']
-    return results
+    results = summarizer(text, max_length=100, min_length=5, do_sample=False)
+    return results[0]['summary_text']
 
 # 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 not text.strip():
+    # Check if the user has entered review
+    if text is None or text.strip() == '':
         st.warning("Please enter a customer review first.")
     else:
         # Perform sentiment analysis on the input text
         sentiment_result = sentiment_class(text)
-        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.")
+        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] == 'LABEL_2':  # Change this to match the negative label in your model
+            summary = summarize_text(text)
+            st.write("Review summary: ", summary)