Update main.py

main.py

@@ -1,7 +1,6 @@
 import asyncio
 import requests
 import time
-import re
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
 
@@ -19,13 +18,14 @@ async def get_page(uid: str, fast_mode=False):
     page = await context.new_page()
 
     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()
@@ -47,22 +47,16 @@ async def info(uid: str):
                     wl = wlMatch ? wlMatch[1] : null;
                 }
 
-                // 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, "");
-                        }
+                        if (match && match[1]) iconUrl = "https://enka.network" + match[1].replace(/\\\\/g, "");
                     }
-                    
                     return { value: val, label: label, icon: iconUrl };
                 });
 
@@ -76,17 +70,46 @@ async def info(uid: str):
                     return { name: charName, level, icon: "https://enka.network" + icon_path };
                 }).filter(i => i !== null);
 
-                return {
-                    player: { name, ar, wl, signature, avatar: avatar_img ? "https://enka.network" + avatar_img : null, stats },
-                    characters: characters
-                };
+                return { player: { name, ar, wl, signature, avatar: avatar_img ? "https://enka.network" + avatar_img : null, stats }, characters };
             }
         """)
+        await browser.close()
+        return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, **data}
+    except Exception as e:
+        await browser.close()
+        return {"error": str(e)}
+
+@app.get("/namecard")
+async def namecard(uid: str):
+    """Lấy link ảnh Danh Thiếp hiện tại"""
+    start_time = time.time()
+    # Fast mode = False vì cần click và chờ hiện element
+    browser, page = await get_page(uid, fast_mode=False)
+    try:
+        # Click nút Danh Thiếp
+        btn_selector = 'button[data-icon="cards"]'
+        await page.wait_for_selector(btn_selector, timeout=15000)
+        await page.click(btn_selector)
+        
+        # Đợi modal Danh Thiếp hiện ra và lấy background-image
+        img_selector = '.Loda .bg'
+        await page.wait_for_selector(img_selector, timeout=15000)
+        
+        namecard_url = await page.evaluate("""
+            () => {
+                const el = document.querySelector('.Loda .bg');
+                if (!el) return null;
+                const style = el.getAttribute('style');
+                const match = style.match(/url\\(["']?(.*?)["']?\\)/);
+                return match ? "https://enka.network" + match[1] : null;
+            }
+        """)
+        
         await browser.close()
         return {
             "execution_time": f"{round(time.time() - start_time, 2)}s",
             "uid": uid,
-            **data
+            "namecard_url": namecard_url
         }
     except Exception as e:
         await browser.close()
@@ -98,12 +121,12 @@ async def generate(uid: str, char: str = None):
     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.5)
             except: pass
 
         await page.click('button[data-icon="image"]')
@@ -123,15 +146,11 @@ 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})
-        return {
-            "execution_time": f"{round(time.time() - start_time, 2)}s",
-            "uid": uid,
-            "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 {"error": str(e)}
 
 @app.get("/")
 def home():
-    return {"status": "Enka API Pro is Live"}
+    return {"status": "Enka API Ultimate Live", "endpoints": ["/info", "/gen", "/namecard"]}