bug fix

app.py

@@ -73,24 +73,29 @@ def generate_plot(df):
     
     # Plot scatter plots for each group
     axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='grey')
+    axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[0, 0].set_title('Article Group 1')
+    axs[0, 0].xticks(rotation=45)
+    axs[0, 0].tight_layout()
     
     axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '7', 'sales'], color='grey')
+    axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[0, 1].set_title('Article Group 2')
+    axs[0, 0].xticks(rotation=45)
+    axs[0, 0].tight_layout()
     
     axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '1', 'sales'], color='grey')
+    axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[1, 0].set_title('Article Group 3')
+    axs[0, 0].xticks(rotation=45)
+    axs[0, 0].tight_layout()
     
     axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '6', 'sales'], color='grey')
-    axs[1, 1].set_title('Article Group 4')
-
-    axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
-    axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
-    axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
+    axs[1, 1].set_title('Article Group 4')
+    axs[0, 0].xticks(rotation=45)
+    axs[0, 0].tight_layout()
 
-    plt.xticks(rotation=45)
-    plt.tight_layout()
     return fig, axs
 
 @st.cache_data