bugfix: added import lgbm

app.py

@@ -276,12 +276,22 @@ elif page == "Customer Analysis":
 
                 # Load X_predict for that cluster
                 X_predict_cluster = pd.read_csv(f'predicts/X_predict_cluster_{cluster}.csv')
+                
+                # Convert cliente_id to string
+                X_predict_cluster['cliente_id'] = X_predict_cluster['cliente_id'].astype(str)
+                
                 st.write("### X_predict_cluster DataFrame:")
                 st.write(X_predict_cluster.head())
                 st.write(f"Shape: {X_predict_cluster.shape}")
 
                 # Filter for the specific customer
-                X_cliente = X_predict_cluster[X_predict_cluster['cliente_id'] == customer_code]
+                customer_code_str = str(customer_code)
+                X_cliente = X_predict_cluster[X_predict_cluster['cliente_id'] == customer_code_str]
+                
+                # Add debug statements
+                st.write(f"Unique customer IDs in X_predict_cluster: {X_predict_cluster['cliente_id'].unique()}")
+                st.write(f"Customer code we're looking for: {customer_code_str}")
+
                 st.write("### X_cliente DataFrame:")
                 st.write(X_cliente.head())
                 st.write(f"Shape: {X_cliente.shape}")