Update index.py

index.py

@@ -152,81 +152,82 @@ def update_output(selected_topic, selected_domain, start_date, end_date):
     mask_1 = ((df["Topic"] == selected_topic) & (df['date'] >= start_date) & (df['date'] <= end_date))
     df_filtered = df.loc[mask_1]
     print(df_filtered.shape)
-    #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).")
-
-    # Veículos de notícias
-    #set x-axis title and y-axis title in line graphs
-    line_fig_1.update_layout(
-                   xaxis_title='Data',
-                   yaxis_title='Classificação de Sentimento')
-
-    #set label format on y-axis in line graphs
-    line_fig_1.update_xaxes(tickformat="%b %d<br>%Y")
-
-    # Bar Graph start
-    grouped_df = df_filtered.groupby(['date', 'Veículos de notícias']).size().reset_index(name='occurrences')
+    if len(df_filtered)>0:
+        #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).")
     
-    # Sort DataFrame by 'period' column
-    grouped_df = grouped_df.sort_values(by='date')
+        # Veículos de notícias
+        #set x-axis title and y-axis title in line graphs
+        line_fig_1.update_layout(
+                       xaxis_title='Data',
+                       yaxis_title='Classificação de Sentimento')
     
-    # Create a list of all unique media
-    all_media = df_filtered['domain_folder_name'].unique()
+        #set label format on y-axis in line graphs
+        line_fig_1.update_xaxes(tickformat="%b %d<br>%Y")
     
-    # Create a date range from Jan/2000 to the last month in the dataset
-    date_range = pd.date_range(start=df_filtered['date'].min().date(), end=df_filtered['date'].max().date(), freq='MS')
+        # Bar Graph start
+        grouped_df = df_filtered.groupby(['date', 'Veículos de notícias']).size().reset_index(name='occurrences')
+        
+        # Sort DataFrame by 'period' column
+        grouped_df = grouped_df.sort_values(by='date')
+        
+        # Create a list of all unique media
+        all_media = df_filtered['domain_folder_name'].unique()
+        
+        # Create a date range from Jan/2000 to the last month in the dataset
+        date_range = pd.date_range(start=df_filtered['date'].min().date(), end=df_filtered['date'].max().date(), freq='MS')
+        
+        # Create a MultiIndex with all combinations of date_range and all_media
+        idx = pd.MultiIndex.from_product([date_range, all_media], names=['date', 'Veículos de notícias'])
+        
+        # Reindex the DataFrame to include all periods and media
+        grouped_df = grouped_df.set_index(['date', 'Veículos de notícias']).reindex(idx, fill_value=0).reset_index()
     
-    # Create a MultiIndex with all combinations of date_range and all_media
-    idx = pd.MultiIndex.from_product([date_range, all_media], names=['date', 'Veículos de notícias'])
+        bar_fig_1 = px.bar(grouped_df, x='date', y='occurrences', color='Veículos de notícias',
+                 labels={'date': 'Período', 'occurrences': 'Número de notícias', 'Veículos de notícias': 'Portal'},
+                 title='Número de notícias por período de tempo')
+        bar_fig_1.update_xaxes(tickformat="%b %d<br>%Y")
+        # Bar Graph ends
+                           
+        # 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]
     
-    # Reindex the DataFrame to include all periods and media
-    grouped_df = grouped_df.set_index(['date', 'Veículos de notícias']).reindex(idx, fill_value=0).reset_index()
-
-    bar_fig_1 = px.bar(grouped_df, x='date', y='occurrences', color='Veículos de notícias',
-             labels={'date': 'Período', 'occurrences': 'Número de notícias', 'Veículos de notícias': 'Portal'},
-             title='Número de notícias por período de tempo')
-    bar_fig_1.update_xaxes(tickformat="%b %d<br>%Y")
-    # Bar Graph ends
-                       
-    # 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, bar_fig_1, line_fig_2, line_fig_3, line_fig_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, bar_fig_1, line_fig_2, line_fig_3, line_fig_4
 
     # return line_fig_1