feat: updated app.py

app.py

@@ -239,6 +239,7 @@ elif page == "Customer Analysis":
                     st.write(f"Shape of euros_proveedor: {euros_proveedor.shape}")
 
                     # Get percentage of units sold for each manufacturer
+                    customer_df = df[df["CLIENTE"] == str(customer_code)]
                     all_manufacturers = customer_data.iloc[:, 1:].T  # Exclude CLIENTE column
                     all_manufacturers.index = all_manufacturers.index.astype(str)
 
@@ -305,16 +306,21 @@ elif page == "Customer Analysis":
                         st.warning("No combined top manufacturers found.")
 
                     # Customer sales 2021-2024 (if data exists)
-                    sales_columns = ['VENTA_2021', 'VENTA_2022', 'VENTA_2023', 'VENTA_2024']
-                    if all(col in df.columns for col in sales_columns):
-                        years = ['2021', '2022', '2023', '2024']
-                        customer_sales = customer_data[sales_columns].values[0]
-
-                        fig_sales = px.line(x=years, y=customer_sales, markers=True, title=f'Sales Over the Years for Customer {customer_code}')
-                        fig_sales.update_layout(xaxis_title="Year", yaxis_title="Sales")
-                        st.plotly_chart(fig_sales)
+                    sales_columns = ['VENTA_2021', 'VENTA_2022', 'VENTA_2023']
+                    if all(col in ventas_clientes.columns for col in sales_columns):
+                        years = ['2021', '2022', '2023']
+                        customer_sales_data = ventas_clientes[ventas_clientes['codigo_cliente'] == customer_code]
+                        
+                        if not customer_sales_data.empty:
+                            customer_sales = customer_sales_data[sales_columns].values[0]
+                            
+                            fig_sales = px.line(x=years, y=customer_sales, markers=True, title=f'Sales Over the Years for Customer {customer_code}')
+                            fig_sales.update_layout(xaxis_title="Year", yaxis_title="Sales")
+                            st.plotly_chart(fig_sales)
+                        else:
+                            st.warning(f"No historical sales data found for customer {customer_code}")
                     else:
-                        st.warning("Sales data for 2021-2024 not available.")
+                        st.warning("Sales data for 2021-2023 not available in the dataset.")
                 else:
                     st.warning(f"No data found for customer {customer_code}. Please check the code.")
         else: