Update app.py

app.py

@@ -31,8 +31,8 @@ def summarize_text(text):
     resp = summarizer(text)
     return resp[0]['summary_text']
 
-def text_to_sentiment(text):
-    sentiment = fin_model(text)[0]["label"]
+def text_to_sentiment(text, all_score=False):
+    sentiment = fin_model(text, return_all_scores=all_score))[0]["label"]
     return sentiment
 
 def fin_ext(text):
@@ -80,8 +80,13 @@ def process_paragraph_1_sent(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
                     selected_paragraph = stored_paragraphs_1[paragraph_index]
-                    results = text_to_sentiment(selected_paragraph)
-                    return results
+                    results = text_to_sentiment(selected_paragraph, True)
+                    if isinstance(results, list) and isinstance(results[0], list):
+                        # We unpack the list of dictionaries to get all labels
+                        output = {result['label']: result['score'] for result in results[0]}
+                    else:
+                        output = {"Error": "Unexpected output format"}
+                    return output
                 except (IndexError, ValueError):
                     return {"Error": "Unexpected output format"}
 def process_paragraph_1_sent_tone(paragraph):
@@ -112,8 +117,13 @@ def process_paragraph_2_sent(paragraph):
                 try:
                     paragraph_index = int(paragraph.split(':')[0].replace('Paragraph ', '')) - 1
                     selected_paragraph = stored_paragraphs_2[paragraph_index]
-                    results = text_to_sentiment(selected_paragraph)
-                    return results
+                    results = text_to_sentiment(selected_paragraph, True)
+                    if isinstance(results, list) and isinstance(results[0], list):
+                        # We unpack the list of dictionaries to get all labels
+                        output = {result['label']: result['score'] for result in results[0]}
+                    else:
+                        output = {"Error": "Unexpected output format"}
+                    return output
                 except (IndexError, ValueError):
                     return {"Error": "Unexpected output format"}
 def process_paragraph_2_sent_tone(paragraph):