Update app.py

app.py

@@ -84,75 +84,46 @@ class TokenManager:
     def __init__(self):
         self.token = None
         self.last_refresh = 0
-        self.expires_in = 3600  # Default expiry
-
-    def _make_auth_request(self, payload: Dict[str, Any]) -> Optional[Dict[str, Any]]:
-        """Make authentication request with multiple content-type attempts"""
-        content_types = [
-            ("application/json", lambda d: {"json": d}),
-            ("application/x-www-form-urlencoded", lambda d: {"data": d}),
-        ]
-
-        for content_type, payload_processor in content_types:
-            try:
-                headers = {
-                    "Content-Type": content_type,
-                    "Accept": "application/json"
-                }
-                res = requests.post(
-                    f"{BACKEND_URL}/auth/login",
-                    headers=headers,
-                    **payload_processor(payload),
-                    timeout=10
-                )
-                if res.status_code == 200:
-                    return res.json()
-                logger.debug(f"Attempt with {content_type} failed: {res.status_code}")
-            except Exception as e:
-                logger.debug(f"Request with {content_type} failed: {str(e)}")
-        
-        return None
-
-    def refresh_token(self) -> str:
-        """Obtain a new token with retry logic"""
-        payload = {
-            "username": ADMIN_CREDENTIALS["email"],
-            "password": ADMIN_CREDENTIALS["password"],
-            "device_token": "admin-console"
+        self.expires_in = 3600
+        self.backend_url = "https://rocketfarmstudios-cps-api.hf.space"
+        self.credentials = {
+            "username": "yakdhanali97@gmail.com",
+            "password": "123456",
+            "device_token": "admin-device-token"
         }
 
-        for attempt in range(1, MAX_TOKEN_RETRIES + 1):
-            try:
-                logger.info(f"Attempting token refresh (attempt {attempt})")
-                response = self._make_auth_request(payload)
-                
-                if response and "access_token" in response:
-                    self.token = response["access_token"]
-                    self.expires_in = response.get("expires_in", 3600)
-                    self.last_refresh = time.time()
-                    logger.info("Successfully refreshed admin token")
-                    return self.token
-                
-                logger.warning(f"Token refresh failed: {response}")
-                
-            except Exception as e:
-                logger.error(f"Token refresh error: {str(e)}")
-            
-            if attempt < MAX_TOKEN_RETRIES:
-                delay = TOKEN_RETRY_DELAY * (2 ** (attempt - 1))  # Exponential backoff
-                time.sleep(delay)
-
-        raise HTTPException(
-            status_code=500,
-            detail="Failed to obtain admin token after multiple attempts"
-        )
-
-    def get_token(self) -> str:
-        """Get current valid token, refreshing if needed"""
+    async def _try_login(self, session):
+        try:
+            async with session.post(
+                f"{self.backend_url}/auth/login",
+                json=self.credentials,
+                timeout=10
+            ) as response:
+                if response.status == 200:
+                    data = await response.json()
+                    return data.get("access_token")
+                else:
+                    error = await response.text()
+                    logger.error(f"Login failed: {response.status} - {error}")
+                    return None
+        except Exception as e:
+            logger.error(f"Login error: {str(e)}")
+            return None
+
+    async def refresh_token(self):
+        async with aiohttp.ClientSession() as session:
+            for attempt in range(3):
+                token = await self._try_login(session)
+                if token:
+                    return token
+                await asyncio.sleep(2 ** attempt)  # Exponential backoff
+        raise Exception("Failed to obtain admin token after multiple attempts")
+
+    async def get_token(self):
         if not self.token or (time.time() - self.last_refresh) > (self.expires_in - 60):
-            return self.refresh_token()
+            self.token = await self.refresh_token()
+            self.last_refresh = time.time()
         return self.token
-
 token_manager = TokenManager()
 
 # ======================