Update app.py

app.py

@@ -392,6 +392,14 @@ 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
+}
+
     
 with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
     with gr.Tab("Methodology"):
@@ -457,20 +465,20 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                     summarize_btn1.click(fn=lambda p: process_paragraph_1_sum(p), inputs=paragraph_1_dropdown, outputs=summary_textbox_1)
             
                     sentiment_btn1 = gr.Button("Classify Financial Tone from PDF 1")
-                    sentiment_textbox_1 = gr.Label(label="Classification for PDF 1")
+                    sentiment_textbox_1 = gr.Label(label="Classification for paragrapg from PDF 1")
             
                     # Classify the financial tone of the selected paragraph from PDF 1
                     sentiment_btn1.click(fn=lambda p: process_paragraph_1_sent(p), inputs=paragraph_1_dropdown, outputs=sentiment_textbox_1)
             
                     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")
+                        fin_spans_1 = gr.HighlightedText(label="Financial Tone Analysis for PDF 1",color_map=color_map)
                         
                         # 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)")
+                        fin_spans_1_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 1 (Bis)",color_map=color_map)
                 
                         # 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_)
@@ -492,19 +500,19 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                     summarize_btn2.click(fn=lambda p: process_paragraph_2_sum(p), inputs=paragraph_2_dropdown, outputs=summary_textbox_2)
             
                     sentiment_btn2 = gr.Button("Classify Financial Tone from PDF 2")
-                    sentiment_textbox_2 = gr.Label(label="Classification for PDF 2")
+                    sentiment_textbox_2 = gr.Label(label="Classification for paragraph from PDF 2")
             
                     # Classify the financial tone of the selected paragraph from PDF 2
                     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")
+                        fin_spans_2 = gr.HighlightedText(label="Financial Tone Analysis for PDF 2",color_map=color_map)
                 
                         # 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)")
+                        fin_spans_2_ = gr.HighlightedText(label="Financial Tone Analysis for PDF 2 (Bis)",color_map=color_map)
                 
                         # 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_)
@@ -532,10 +540,10 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
         
         with gr.Row():
             with gr.Column():
-                sentiment_results_pdf1 = gr.HighlightedText(label="Sentiment Analysis - PDF 1")
+                sentiment_results_pdf1 = gr.HighlightedText(label="Sentiment Analysis - PDF 1",color_map=color_map)
 
             with gr.Column():
-                sentiment_results_pdf2 = gr.HighlightedText(label="Sentiment Analysis - PDF 2")
+                sentiment_results_pdf2 = gr.HighlightedText(label="Sentiment Analysis - PDF 2",color_map=color_map)
                
         with gr.Accordion("Adverse growth trends"):
             with gr.Row():