Update app/akasa_logic.py

app/akasa_logic.py

@@ -10,12 +10,19 @@ class AkasaApp:
 
     async def start(self):
         self.playwright = await async_playwright().start()
+        # Launch với các flags tối ưu nhất cho HuggingFace (Low RAM)
         self.browser = await self.playwright.chromium.launch(
             headless=True,
-            args=['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage']
+            args=[
+                '--no-sandbox', 
+                '--disable-setuid-sandbox', 
+                '--disable-dev-shm-usage',
+                '--disable-blink-features=AutomationControlled', # Tránh bị check bot
+                '--disable-extensions'
+            ]
         )
         asyncio.create_task(self._cleanup_cache())
-        print("🚀 [System] Akasha Logic (Browser) started.")
+        print("🚀 [Akasha] Browser Ready")
 
     async def _cleanup_cache(self):
         while True:
@@ -25,11 +32,10 @@ class AkasaApp:
             for uid in uids_to_remove:
                 item = self.cache.pop(uid)
                 await item["context"].close()
-                print(f"🧹 [Akasha] Closed expired tab for UID: {uid}")
 
     async def get_page(self, uid: str):
         if uid in self.cache:
-            self.cache[uid]["expires"] = time.time() + 180 
+            self.cache[uid]["expires"] = time.time() + 300
             return self.cache[uid]["page"], False
 
         context = await self.browser.new_context(
@@ -37,52 +43,46 @@ class AkasaApp:
             user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
         )
         page = await context.new_page()
-        self.cache[uid] = {"page": page, "context": context, "expires": time.time() + 180}
+        # Chặn bớt ảnh và css thừa của Akasha để tải cực nhanh
+        await page.route("**/*.{png,jpg,jpeg,gif,webp,css}", lambda route: route.abort())
+        
+        self.cache[uid] = {"page": page, "context": context, "expires": time.time() + 300}
         return page, True
 
-        async def refresh_akasha(self, uid: str):
+    async def refresh_akasha(self, uid: str):
         start_time = time.time()
         page, is_new = await self.get_page(uid)
-        url = f"https://akasha.cv/profile/{uid}"
         
         try:
-            # Tăng timeout lên 45s cho lần load đầu
-            await page.goto(url, wait_until="networkidle", timeout=45000)
-            
-            # Selector dựa trên thuộc tính title bạn đã gửi
-            refresh_btn_selector = 'div[title="Refresh builds"]'
+            # 1. Đi tới profile
+            await page.goto(f"https://akasha.cv/profile/{uid}", wait_until="commit", timeout=20000)
             
-            # Đợi nút xuất hiện
-            await page.wait_for_selector(refresh_btn_selector, state="visible", timeout=15000)
+            # 2. Đợi nút Refresh (title="Refresh builds") xuất hiện trong DOM
+            # Dùng selector chính xác từ mẫu HTML bạn đưa
+            refresh_btn = page.locator('div[title="Refresh builds"]')
             
-            # Cuộn đến nút để chắc chắn nó click được
-            await page.locator(refresh_btn_selector).scroll_into_view_if_needed()
+            # Chờ nút hiện ra (timeout 10s)
+            await refresh_btn.wait_for(state="attached", timeout=10000)
             
-            # Click vào nút
-            await page.click(refresh_btn_selector)
+            # 3. Click cưỡng ép (force=True) vì đôi khi nó bị che bởi overlay
+            await refresh_btn.click(force=True)
             
-            # Kiểm tra xem sau khi click có hiện thông báo "Last profile update" đang thay đổi không
-            # Đây là bước phụ để xác nhận lệnh đã chạy
-            await asyncio.sleep(2) 
+            # 4. Đợi 1 giây để trigger gửi request lên server Akasha
+            await asyncio.sleep(1.5)
 
             return {
                 "status": "success",
-                "message": "Refresh command executed",
                 "uid": uid,
-                "execution_time": f"{round(time.time() - start_time, 2)}s"
+                "msg": "Lệnh Refresh đã được gửi thành công",
+                "time": f"{round(time.time() - start_time, 2)}s"
             }
         except Exception as e:
-            # Chụp ảnh màn hình lỗi để debug (tùy chọn, lưu trên HuggingFace)
-            # await page.screenshot(path="error_akasha.png")
             return {
-                "status": "error", 
-                "message": "Nút Refresh không tìm thấy hoặc trang load quá chậm.",
-                "detail": str(e),
-                "uid": uid
+                "status": "error",
+                "msg": "Không tìm thấy nút Refresh. Có thể UID sai hoặc Akasha đang lag.",
+                "error": str(e)
             }
 
     async def stop(self):
-        for item in self.cache.values():
-            await item["context"].close()
         if self.browser: await self.browser.close()
         if self.playwright: await self.playwright.stop()