Update main.py

main.py

@@ -1,6 +1,7 @@
 import asyncio
 import requests
 import time
+import re
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
 
@@ -8,7 +9,6 @@ app = FastAPI()
 
 IMGBB_API_KEY = "093dee93ec19418d26cc76f40e053725"
 
-# --- HELPER: KHỞI TẠO TRÌNH DUYỆT VỚI TÙY CHỌN TỐI ƯU ---
 async def get_page(uid: str, fast_mode=False):
     p = await async_playwright().start()
     browser = await p.chromium.launch(headless=True, args=['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'])
@@ -18,20 +18,17 @@ async def get_page(uid: str, fast_mode=False):
     )
     page = await context.new_page()
 
-    # NẾU LÀ FAST MODE (CHỈ LẤY TEXT), CHẶN TẢI ẢNH VÀ CSS ĐỂ TĂNG TỐC
     if fast_mode:
+        # Chỉ cho phép tải tài nguyên cần thiết cho Text, chặn ảnh/css nặng
         await page.route("**/*", lambda route: route.abort() if route.request.resource_type in ["image", "media", "font", "stylesheet"] else route.continue_())
 
     url = f"https://enka.network/u/{uid}/"
     await page.goto(url, wait_until="domcontentloaded", timeout=60000)
     return browser, page
 
-# --- ENDPOINTS ---
-
 @app.get("/info")
 async def info(uid: str):
     start_time = time.time()
-    # Sử dụng Fast Mode để lấy thông tin text cực nhanh
     browser, page = await get_page(uid, fast_mode=True)
     try:
         data = await page.evaluate("""
@@ -50,10 +47,24 @@ async def info(uid: str):
                     wl = wlMatch ? wlMatch[1] : null;
                 }
 
-                const stats = Array.from(document.querySelectorAll('tr.stat')).map(tr => ({
-                    value: tr.querySelector('td:first-child')?.innerText.trim(),
-                    label: tr.querySelector('td:last-child')?.innerText.trim()
-                }));
+                // FETCH STATS CÙNG ICON URL
+                const stats = Array.from(document.querySelectorAll('tr.stat')).map(tr => {
+                    const val = tr.querySelector('td:first-child')?.innerText.trim();
+                    const label = tr.querySelector('td:last-child')?.innerText.trim();
+                    const iconTd = tr.querySelector('td.icon');
+                    let iconUrl = null;
+                    
+                    if (iconTd) {
+                        const style = iconTd.getAttribute('style');
+                        // Regex bóc tách link từ --icon: url(...)
+                        const match = style.match(/url\\(["']?(.*?)["']?\\)/);
+                        if (match && match[1]) {
+                            iconUrl = "https://enka.network" + match[1].replace(/\\\\/g, "");
+                        }
+                    }
+                    
+                    return { value: val, label: label, icon: iconUrl };
+                });
 
                 const characters = Array.from(document.querySelectorAll('.CharacterList .avatar')).map(el => {
                     const figure = el.querySelector('figure.chara');
@@ -72,25 +83,27 @@ async def info(uid: str):
             }
         """)
         await browser.close()
-        execution_time = f"{round(time.time() - start_time, 2)}s"
-        return {"execution_time": execution_time, "uid": uid, **data}
+        return {
+            "execution_time": f"{round(time.time() - start_time, 2)}s",
+            "uid": uid,
+            **data
+        }
     except Exception as e:
         await browser.close()
-        return {"execution_time": f"{round(time.time() - start_time, 2)}s", "error": str(e)}
+        return {"error": str(e)}
 
 @app.get("/gen")
 async def generate(uid: str, char: str = None):
     start_time = time.time()
-    # Không dùng fast_mode vì cần tải CSS/Ảnh để render Card
     browser, page = await get_page(uid, fast_mode=False)
     try:
         if char:
-            formatted_char = char.strip().capitalize()
+            formatted_char = char.strip().capitalize();
             char_selector = f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{formatted_char}')]"
             try:
                 await page.wait_for_selector(char_selector, timeout=5000)
                 await page.click(char_selector)
-                await asyncio.sleep(1) 
+                await asyncio.sleep(1)
             except: pass
 
         await page.click('button[data-icon="image"]')
@@ -110,12 +123,15 @@ async def generate(uid: str, char: str = None):
         """)
         await browser.close()
         res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_API_KEY, "image": img_base64})
-        execution_time = f"{round(time.time() - start_time, 2)}s"
-        return {"execution_time": execution_time, "uid": uid, "char": char or "default", "card_url": res.json().get('data', {}).get('url')}
+        return {
+            "execution_time": f"{round(time.time() - start_time, 2)}s",
+            "uid": uid,
+            "card_url": res.json().get('data', {}).get('url')
+        }
     except Exception as e:
         await browser.close()
-        return {"execution_time": f"{round(time.time() - start_time, 2)}s", "error": str(e)}
+        return {"error": str(e)}
 
 @app.get("/")
 def home():
-    return {"status": "Enka API Optimized"}
+    return {"status": "Enka API Pro is Live"}