Update app.py

app.py

@@ -477,7 +477,10 @@ with gr.Blocks(theme='gradio/soft',js=js_func) as demo:
                         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.click(fn=lambda p: process_paragraph_1_sent_tone(p), inputs=paragraph_1_dropdown, outputs=fin_spans_1.update(
+                            color_map=color_map,  # Keep color map
+                            show_legend=True      # Keep legend
+                        ))
                 
                         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)