Update index.py

index.py

@@ -154,7 +154,46 @@ def update_output(selected_topic, start_date, end_date):
     #set label format on y-axis in line graphs
     line_fig_1.update_xaxes(tickformat="%b %d<br>%Y")
 
-    return line_fig_1
+    # filter dataframes based on updated data range
+    mask_2 = ((df_pos["Topic"] == selected_topic) & (df_pos["domain_folder_name"] == selected_domain) & (df_pos['date'] >= start_date) & (df_pos['date'] <= end_date))
+    mask_3 = ((df_neu["Topic"] == selected_topic) & (df_neu["domain_folder_name"] == selected_domain) & (df_neu['date'] >= start_date) & (df_neu['date'] <= end_date))
+    mask_4 = ((df_neg["Topic"] == selected_topic) & (df_neg["domain_folder_name"] == selected_domain) & (df_neg['date'] >= start_date) & (df_neg['date'] <= end_date))
+    df2_filtered = df_pos.loc[mask_2]
+    df3_filtered = df_neu.loc[mask_3]
+    df4_filtered = df_neg.loc[mask_4]
+
+#create line graphs based on filtered dataframes
+    line_fig_2 = px.line(df2_filtered, x="date", y="rolling_mean_counts", line_group="FinBERT_label",
+                     title="Positive")
+    line_fig_3 = px.line(df3_filtered, x="date", y="rolling_mean_counts", line_group="FinBERT_label",
+                     title="Neutral")
+    line_fig_4 = px.line(df4_filtered, x="date", y="rolling_mean_counts", line_group="FinBERT_label",
+                     title="Negative")
+
+#set x-axis title and y-axis title in line graphs
+    line_fig_2.update_layout(
+                   xaxis_title='Data',
+                   yaxis_title='Número de notícias com sentimento positivo')
+    line_fig_3.update_layout(
+                   xaxis_title='Data',
+                   yaxis_title='Número de notícias com sentimento neutro')
+    line_fig_4.update_layout(
+                   xaxis_title='Data',
+                   yaxis_title='Número de notícias com sentimento negativo')
+
+#set label format on y-axis in line graphs
+    line_fig_2.update_xaxes(tickformat="%b %d<br>%Y")
+    line_fig_3.update_xaxes(tickformat="%b %d<br>%Y")
+    line_fig_4.update_xaxes(tickformat="%b %d<br>%Y")
+
+#set label format on y-axis in line graphs
+    line_fig_2.update_traces(line_color='#1E88E5')
+    line_fig_3.update_traces(line_color='#004D40')
+    line_fig_4.update_traces(line_color='#D81B60')
+
+    return line_fig_1, line_fig_2, line_fig_3, line_fig_4
+
+    # return line_fig_1