Update index.py

index.py

@@ -30,7 +30,7 @@ print(unique_domains)
 unique_topics = df['Topic'].unique()
 print(unique_topics)
 
-df = df.rename(columns={df.columns[4]: "Veículos de notícias"})
+# df = df.rename(columns={df.columns[4]: "Veículos de notícias"})
 
 df['FinBERT_label'] = df['FinBERT_label'].astype(str)
 df['FinBERT_label'].replace({
@@ -41,7 +41,7 @@ df['FinBERT_label'].replace({
 
 
 
-counts = df.groupby(['date', 'Topic', 'Veículos de notícias', 'FinBERT_label']).size().reset_index(name='count')
+counts = df.groupby(['date', 'Topic', 'domain_folder_name', 'FinBERT_label']).size().reset_index(name='count')
 counts['count'] = counts['count'].astype('float64')
 counts['rolling_mean_counts'] = counts['count'].rolling(window=30, min_periods=2).mean()
 df_pos = counts[[x in ['positive'] for x in counts.FinBERT_label]]
@@ -149,8 +149,9 @@ def update_output(selected_topic, selected_domain, start_date, end_date):
 
     #create line graphs based on filtered dataframes
     line_fig_1 = px.line(df_filtered, x="date", y="normalised results",
-                     color='Veículos de notícias', title="O gráfico mostra a evolução temporal de sentimento dos títulos de notícias. Numa escala de -1 (negativo) a 1 (positivo), sendo 0 (neutro).")
+                     color='domain_folder_name', title="O gráfico mostra a evolução temporal de sentimento dos títulos de notícias. Numa escala de -1 (negativo) a 1 (positivo), sendo 0 (neutro).")
 
+    # Veículos de notícias
     #set x-axis title and y-axis title in line graphs
     line_fig_1.update_layout(
                    xaxis_title='Data',