Update app.py

app.py

@@ -397,9 +397,9 @@ function refresh() {
 
 # Define custom colors for the labels
 color_map = {
-    "POSITIVE": "#00FF00",  # Green for positive
-    "NEUTRAL": "#0000FF",   # Blue for neutral
-    "NEGATIVE": "#FF0000"   # Red for negative
+    "Positive": "green",  # Green for positive
+    "Neutral": "blue",   # Blue for neutral
+    "Negative": "red"   # Red for negative
 }
 
     
@@ -474,13 +474,13 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
             
                     with gr.Accordion("Analyze Financial Tone on each sentence"):
                         analyze_btn1 = gr.Button("With FinBERT-tone")
-                        fin_spans_1 = gr.HighlightedText(label="Financial Tone Analysis for PDF 1",color_map=color_map)
+                        fin_spans_1 = gr.HighlightedText(label="Financial Tone Analysis for PDF 1",color_map=color_map, show_legend=True)
                         
                         # Analyze financial tone on each sentence using FinBERT-tone
                         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)
+                        fin_spans_1_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 1 (Bis)",color_map=color_map, 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_)
@@ -508,13 +508,13 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                     sentiment_btn2.click(fn=lambda p: process_paragraph_2_sent(p), inputs=paragraph_2_dropdown, outputs=sentiment_textbox_2)
                     with gr.Accordion("Analyze Financial Tone on each sentence"):
                         analyze_btn2 = gr.Button("With FinBERT-tone")
-                        fin_spans_2 = gr.HighlightedText(label="Financial Tone Analysis for PDF 2",color_map=color_map)
+                        fin_spans_2 = gr.HighlightedText(label="Financial Tone Analysis for PDF 2",color_map=color_map, show_legend=True)
                 
                         # Analyze financial tone on each sentence using FinBERT-tone for PDF 2
                         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)
+                        fin_spans_2_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 2 (Bis)",color_map=color_map, 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_)