intrgrating suggestions after betta 1

transportmanager/client/pages/clients_page.py

@@ -265,6 +265,18 @@ class ClientsPage:
         except Exception as e:
             print("Error loading subscription:", e)
 
+    def get_client_access(self):
+        token = self.page.client_storage.get("token")
+        headers = {"Authorization": f"Bearer {token}"}
+        response = requests.get(f"{API_BASE_URL}/profile/", headers=headers, timeout=10)
+        user = response.json()
+        if user['user_role'] == 'user':
+            return True
+        else:
+            id = self.page.session.get("user_id")
+            response = requests.get(f"{API_BASE_URL}/company_user/access/{id}", headers=headers)
+            return True if response.json()['clients'] == 1 else False
+
     def build(self):
         self.load_clients()
         self.subscription = self.get_current_subscription_plan()
@@ -291,4 +303,30 @@ class ClientsPage:
                 expand=True
             ),
             expand=True,
+        ) if self.get_client_access() else ft.Container(
+            content=ft.Column(
+                [
+                    ft.Row(
+                        [
+                            ft.Text("Clients", size=24, weight=ft.FontWeight.BOLD),
+                        ],
+                        alignment=ft.MainAxisAlignment.SPACE_BETWEEN
+                    ),
+                    ft.Row(
+                        [
+                            ft.Text(
+                                "You do not have access to this page content", 
+                                size=24, 
+                                weight=ft.FontWeight.BOLD,
+                                color=ft.Colors.RED
+                            )
+                        ],
+                        alignment=ft.MainAxisAlignment.CENTER
+                    ),
+                    ft.Text("")
+                ],
+                alignment=ft.MainAxisAlignment.SPACE_BETWEEN,
+                expand=True
+            ),
+            expand=True
         )