Update main.py

main.py

@@ -7,91 +7,114 @@ app = FastAPI()
 
 IMGBB_API_KEY = "093dee93ec19418d26cc76f40e053725"
 
-# Từ điển Việt hóa
-TRANSLATIONS = {
-    "Total Achievements": "Tổng Số Thành Tựu",
-    "Spiral Abyss": "La Hoàn Thâm Cảnh",
-    "Stygian Onslaught": "Ảo Cảnh Hiểm Ác"
-}
+# --- HELPER: KHỞI TẠO TRÌNH DUYỆT ---
+async def get_page(uid: str):
+    browser = await async_playwright().start()
+    browser_instance = await browser.chromium.launch(headless=True, args=['--no-sandbox'])
+    context = await browser_instance.new_context(
+        viewport={'width': 1920, 'height': 1080},
+        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()
+    url = f"https://enka.network/u/{uid}/"
+    await page.goto(url, wait_until="networkidle", timeout=60000)
+    return browser_instance, page
 
-async def get_enka_profile(uid: str):
-    async with async_playwright() as p:
-        browser = await p.chromium.launch(headless=True, args=['--no-sandbox'])
-        context = await browser.new_context(
-            viewport={'width': 1920, 'height': 1080},
-            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()
-        
-        try:
-            url = f"https://enka.network/u/{uid}/"
-            await page.goto(url, wait_until="networkidle", timeout=60000)
-
-            player_data = await page.evaluate("""
-                () => {
-                    const name = document.querySelector('.details h1')?.innerText;
-                    const arMatch = document.querySelector('.ar')?.innerText.match(/AR(\\d+)/);
-                    const wlMatch = document.querySelector('.ar')?.innerText.match(/WL(\\d+)/);
-                    const signature = document.querySelector('.signature')?.innerText;
-                    const avatar_img = document.querySelector('.avatar-icon img')?.getAttribute('src');
-                    
-                    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()
-                    }));
-
-                    return { 
-                        name, 
-                        ar: arMatch ? arMatch[1] : null, 
-                        wl: wlMatch ? wlMatch[1] : null, 
-                        signature, 
-                        avatar: avatar_img ? "https://enka.network" + avatar_img : null,
-                        stats 
-                    };
-                }
-            """)
+# --- ENDPOINTS ---
 
-            # --- BẮT ĐẦU VIỆT HÓA ---
-            if player_data.get("stats"):
-                for item in player_data["stats"]:
-                    original_label = item["label"]
-                    # Nếu nhãn có trong từ điển thì thay thế, không thì giữ nguyên
-                    item["label"] = TRANSLATIONS.get(original_label, original_label)
-            # ------------------------
-
-            # Lấy danh sách nhân vật
-            characters = await page.evaluate("""
-                () => {
-                    const items = document.querySelectorAll('.CharacterList .avatar');
-                    return Array.from(items).map(el => {
-                        const figure = el.querySelector('figure.chara');
-                        if (!figure) return null;
-                        const style = figure.getAttribute('style');
-                        const icon_path = style.match(/url\\((.*)\\)/)[1].replace(/["']/g, "");
-                        const level = el.querySelector('.level')?.innerText || "0";
-                        const name = icon_path.split('_').pop().split('.')[0];
-                        return { name, level, icon: "https://enka.network" + icon_path };
-                    }).filter(i => i !== null);
+@app.get("/info")
+async def info(uid: str):
+    """Lấy thông tin Profile + Danh sách nhân vật (Đã fix lỗi NULL)"""
+    browser, page = await get_page(uid)
+    try:
+        data = await page.evaluate("""
+            () => {
+                const name = document.querySelector('.details h1')?.innerText;
+                const signature = document.querySelector('.signature')?.innerText;
+                const avatar_img = document.querySelector('.avatar-icon img')?.getAttribute('src');
+                
+                // Fix AR & WL: Lấy text tổng rồi dùng Regex linh hoạt
+                const arContainer = document.querySelector('.ar');
+                let ar = null, wl = null;
+                if (arContainer) {
+                    const text = arContainer.innerText;
+                    const arMatch = text.match(/AR\\s*(\\d+)/i);
+                    const wlMatch = text.match(/WL\\s*(\\d+)/i);
+                    ar = arMatch ? arMatch[1] : null;
+                    wl = wlMatch ? wlMatch[1] : null;
                 }
-            """)
 
-            await browser.close()
-            return {"uid": uid, "player": player_data, "characters": characters}
+                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()
+                }));
 
-        except Exception as e:
-            if 'browser' in locals(): await browser.close()
-            return {"error": str(e)}
+                const characters = Array.from(document.querySelectorAll('.CharacterList .avatar')).map(el => {
+                    const figure = el.querySelector('figure.chara');
+                    if (!figure) return null;
+                    const style = figure.getAttribute('style');
+                    const icon_path = style.match(/url\\((.*)\\)/)[1].replace(/["']/g, "");
+                    const level = el.querySelector('.level')?.innerText || "0";
+                    const charName = icon_path.split('_').pop().split('.')[0];
+                    return { 
+                        name: charName, 
+                        level: level, 
+                        icon: "https://enka.network" + icon_path 
+                    };
+                }).filter(i => i !== null);
 
-# --- GIỮ NGUYÊN CÁC ENDPOINT KHÁC ---
-@app.get("/info")
-async def info(uid: str):
-    return await get_enka_profile(uid)
+                return {
+                    player: { name, ar, wl, signature, avatar: avatar_img ? "https://enka.network" + avatar_img : null, stats },
+                    characters: characters
+                };
+            }
+        """)
+        await browser.close()
+        return {"uid": uid, **data}
+    except Exception as e:
+        await browser.close()
+        return {"error": str(e)}
 
 @app.get("/gen")
 async def generate(uid: str, char: str = None):
-    # (Giữ nguyên hàm get_enka_card như cũ)
-    pass
+    """Tự động chọn nhân vật và chụp ảnh Card"""
+    browser, page = await get_page(uid)
+    try:
+        if char:
+            # Chuẩn hóa tên (viết hoa chữ đầu) để tìm trong style của Enka
+            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.5) # Đợi render
+            except: pass
+
+        # Nhấn nút download ảnh
+        await page.click('button[data-icon="image"]')
+        await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=20000)
+        
+        img_base64 = await page.evaluate("""
+            async () => {
+                const img = document.querySelector('.Loda img[src^="blob:"]');
+                const res = await fetch(img.src);
+                const blob = await res.blob();
+                return new Promise(r => {
+                    const reader = new FileReader();
+                    reader.onloadend = () => r(reader.result.split(',')[1]);
+                    reader.readAsDataURL(blob);
+                });
+            }
+        """)
+        await browser.close()
+
+        # Upload ImgBB
+        res = requests.post("https://api.imgbb.com/1/upload", data={"key": IMGBB_API_KEY, "image": img_base64})
+        return {"uid": uid, "char": char or "default", "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 Vietnamese API is Live"}
+    return {"status": "Enka API Final is Running", "endpoints": ["/info?uid=", "/gen?uid=&char="]}