feat: updated website

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import time
 import pandas as pd
 import plotly.express as px
+import plotly.graph_objects as go
 import matplotlib.pyplot as plt
 import numpy as np
 import lightgbm as lgb
@@ -452,31 +453,32 @@ elif page == "Customer Analysis":
                                 # Add 2024 to the years list
                                 years.append('2024')
 
-                                # Create an interactive bar chart using Plotly
-                                fig_sales_bar = px.bar(
+                                fig_sales_bar = go.Figure()
+                                # Add trace for actual sales
+                                fig_sales_bar.add_trace(go.Bar(
                                     x=years,
-                                    y=[sales_values, predicted_values],
-                                    labels={"x": "Year", "y": "Sales (€)"},
-                                    barmode='group',
-                                    title=f"Sales Over the Years for Customer {customer_code} (Actual vs Predicted)"
-                                )
+                                    y=sales_values,
+                                    name="Actual Sales",
+                                    marker_color='blue'
+                                ))
+
+                                # Add trace for predicted sales
+                                fig_sales_bar.add_trace(go.Bar(
+                                    x=years,
+                                    y=predicted_values,
+                                    name="Predicted Sales (Annualized for 2024)",
+                                    marker_color='orange'
+                                ))
 
-                                # Update the layout to make the bar chart interactive
+                                # Update layout
                                 fig_sales_bar.update_layout(
-                                    yaxis_title="Sales (€)",
+                                    title=f"Sales Over the Years for Customer {customer_code} (Actual vs Predicted)",
                                     xaxis_title="Year",
-                                    hovermode="x unified",
+                                    yaxis_title="Sales (€)",
+                                    barmode='group',
                                     height=600,
-                                    legend_title_text="Sales Type"
-                                )
-
-                                # Add traces for actual and predicted sales
-                                fig_sales_bar.add_trace(
-                                    px.bar(x=years, y=sales_values, name="Actual Sales", marker_color='blue').data[0]
-                                )
-
-                                fig_sales_bar.add_trace(
-                                    px.bar(x=years, y=predicted_values, name="Predicted Sales (Annualized for 2024)", marker_color='orange').data[0]
+                                    legend_title_text="Sales Type",
+                                    hovermode="x unified"
                                 )
 
                                 # Show the interactive bar chart in Streamlit