Update app.py

app.py

@@ -38,7 +38,6 @@ def text_to_sentiment(text, all_score=False, label = True):
         return fin_model(text, return_all_scores=all_score)
 def fin_ext(text):
     results = fin_model(split_in_sentences(text))
-    print(make_spans(text, results))
     return make_spans(text, results)
 def fin_ext_bis(text):
     results = fin_model_bis(split_in_sentences(text))
@@ -402,6 +401,12 @@ color_map = {
     "Negative": "red"   # Red for negative
 }
 
+color_map1 = {
+    "positive": "green",  # Green for positive
+    "neutral": "blue",   # Blue for neutral
+    "negative": "red"   # Red for negative
+}
+
     
 with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
     with gr.Tab("Methodology"):
@@ -480,7 +485,7 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                         analyze_btn1.click(fn=lambda p: process_paragraph_1_sent_tone(p), inputs=paragraph_1_dropdown, outputs=fin_spans_1)
                 
                         analyze_btn1_ = gr.Button("With ProsusAI/finbert")
-                        fin_spans_1_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 1 (Bis)",color_map=color_map, show_legend=True)
+                        fin_spans_1_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 1 (Bis)",color_map=color_map1, show_legend=True)
                 
                         # Analyze financial tone using ProsusAI/finbert
                         analyze_btn1_.click(fn=lambda p: process_paragraph_1_sent_tone_bis(p), inputs=paragraph_1_dropdown, outputs=fin_spans_1_)
@@ -514,7 +519,7 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                         analyze_btn2.click(fn=lambda p: process_paragraph_2_sent_tone(p), inputs=paragraph_2_dropdown, outputs=fin_spans_2)
                 
                         analyze_btn2_ = gr.Button("With ProsusAI/finbert")
-                        fin_spans_2_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 2 (Bis)",color_map=color_map, show_legend=True)
+                        fin_spans_2_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 2 (Bis)",color_map=color_map1, show_legend=True)
                 
                         # Analyze financial tone using ProsusAI/finbert for PDF 2
                         analyze_btn2_.click(fn=lambda p: process_paragraph_2_sent_tone_bis(p), inputs=paragraph_2_dropdown, outputs=fin_spans_2_)