Add access to users that are part of Enterprise orgs

app.py

@@ -8,14 +8,25 @@ FAL_KEY = os.getenv("FAL_KEY", "")
 fal_client.api_key = FAL_KEY
 
 def verify_pro_status(token: Optional[gr.OAuthToken]) -> bool:
-    """Verifies if the user is a Hugging Face PRO user using their token."""
+    """Verifies if the user is a Hugging Face PRO user or part of an enterprise org."""
     if not token:
         return False
     try:
         user_info = whoami(token=token.token)
-        return user_info.get("isPro", False)
+
+        # Case 1: User is PRO
+        if user_info.get("isPro", False):
+            return True
+
+        # Case 2: User is in any enterprise org
+        orgs = user_info.get("orgs", [])
+        if any(org.get("isEnterprise", False) for org in orgs):
+            return True
+
+        return False
+
     except Exception as e:
-        print(f"Could not verify user's PRO status: {e}")
+        print(f"Could not verify user's PRO/Enterprise status: {e}")
         return False
 
 # --- Backend Generation Functions ---