Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import inflect
 from transformers import pipeline
 
 st.set_page_config(page_title="Text to Sentiment Analysis Audio", page_icon="🦜")
@@ -17,12 +18,20 @@ def analyze_sentiment(text):
     return sentiment, probability
 
 def format_sentiment_text(sentiment, probability):
-    return f"The sentiment of the text is {sentiment} with a probability of {probability:.2f}."
-
-def preprocess_text(text):
-    words = text.split()
-    processed_words = [p.number_to_words(word) if word.isdigit() else word for word in words]
-    return ' '.join(processed_words)
+    return f"The sentence is {sentiment} with a probability of {probability:.2f}."
+
+def number_to_words(number):
+    p = inflect.engine()
+    # Split the number into integer and decimal parts
+    integer_part, decimal_part = divmod(number, 1)
+    # Convert integer part to words
+    words = p.number_to_words(int(integer_part))
+    # Handle the decimal part
+    if decimal_part > 0:
+        decimal_part = round(decimal_part * 100)  # Convert to two decimal places
+        decimal_words = p.number_to_words(int(decimal_part))
+        words += f" point {decimal_words.replace('-', ' ')}"
+    return words
 
 def text_to_audio(text):
     audio = text_audio_pipeline(text)
@@ -39,8 +48,8 @@ if input_text:
 
     # Stage 2: Convert text to audio
     st.text('Generating audio...')
-    sentiment_text_for_audio = f"The sentiment of the input text is {sentiment}, and the probability is {probability:.2f}."
-    sentiment_text_for_audio = preprocess_text(sentiment_text_for_audio)
+    prob=number_to_words(probability)
+    sentiment_text_for_audio = f"The sentence is {sentiment}, with a probability of {prob:.2f}."
     audio_data = text_to_audio(sentiment_text_for_audio)
 
     # Play button