Update index.py

index.py

@@ -286,11 +286,16 @@ def update_output(selected_topic, selected_domain, start_date, end_date):
         # Fill missing values with zeros
         merged_df['count'].fillna(0, inplace=True)
         merged_df['rolling_mean_counts'].fillna(0, inplace=True)
-    
+
+        # Define colors for each label
+        label_colors = {'positivo': '#039a4d', 'neutro': '#3c03f4', 'negativo': '#ca3919'}
+        
         # Create line graph...
         line_fig_2 = px.line(merged_df, x="date", y="count", color="FinBERT_label",
                        line_group="FinBERT_label", title="Sentimento ao longo do tempo",
-                       labels={"count": "Número de notícias", "date": "Date"})
+                       labels={"count": "Número de notícias", "date": "Date"},
+                             color_discrete_sequence=[label_colors[label] for label in all_combinations.index]
+                            )
     
     
         # Update layout...
@@ -308,8 +313,7 @@ def update_output(selected_topic, selected_domain, start_date, end_date):
         # Calculate percentage of each label
         label_percentages_all = (label_counts_all / label_counts_all.sum()) * 100
         
-        # Define colors for each label
-        label_colors = {'positivo': '#039a4d', 'neutro': '#3c03f4', 'negativo': '#ca3919'}
+        
         # Plot general pie chart
         pie_chart_1 = px.pie(
             values=label_percentages_all,