Update main.py

main.py

@@ -1,5 +1,4 @@
 import asyncio
-import os
 import requests
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
@@ -8,92 +7,117 @@ app = FastAPI()
 
 IMGBB_API_KEY = "093dee93ec19418d26cc76f40e053725"
 
-# Proxy ScrapingAnt
-PROXY_CONFIG = {
-    "server": "http://proxy.scrapingant.com:8080",
-    "username": "scrapingant&proxy_type=residential&proxy_country=VN&browser=false",
-    "password": "20e836f16c914c2f816d9d4ab53f83b0"
-}
-
-async def run_logic(uid: str):
+async def get_enka_data(uid: str, char_name: str = None):
     async with async_playwright() as p:
-        browser = await p.chromium.launch(headless=True, proxy=PROXY_CONFIG)
+        # Khởi tạo trình duyệt
+        browser = await p.chromium.launch(headless=True, args=['--no-sandbox'])
         context = await browser.new_context(
-            accept_downloads=True,
-            ignore_https_errors=True,
-            viewport={'width': 1280, 'height': 1500},
+            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()
-
-        # CHẶN QUẢNG CÁO & TRACKER ĐỂ TIẾT KIỆM PROXY
-        async def block_aggressively(route):
-            if route.request.resource_type in ["image", "media", "font"] and "enka.network" not in route.request.url:
-                await route.abort()
-            elif any(x in route.request.url for x in ["google-analytics", "doubleclick", "playwire", "ads"]):
-                await route.abort()
-            else:
-                await route.continue_()
-        
-        await page.route("**/*", block_aggressively)
         
         try:
-            # Truy cập với Timeout cực lớn 120s
-            await page.goto(f"https://akasha.cv/profile/{uid}", wait_until="commit", timeout=120000)
-            
-            # Đợi hàng nhân vật xuất hiện
-            char_row = "table.custom-table tbody tr.pointer"
-            await page.wait_for_selector(char_row, timeout=60000)
-            
-            # Click và đợi bảng render
-            await page.click(char_row)
-            
-            # Đợi nút Download - Akasha render card khá lâu nên phải kiên nhẫn
-            dl_btn = 'button.dl-button'
-            await page.wait_for_selector(dl_btn, timeout=60000)
-            await asyncio.sleep(8) # Chờ card vẽ xong hoàn toàn
+            url = f"https://enka.network/u/{uid}/"
+            print(f"📡 Truy cập Enka: {url}")
+            await page.goto(url, wait_until="networkidle", timeout=60000)
 
-            # Bắt đầu tải file
-            async with page.expect_download(timeout=90000) as download_info:
-                await page.click(dl_btn)
-            
-            download = await download_info.value
-            temp_path = f"card_{uid}.png"
-            await download.save_as(temp_path)
+            # --- LOGIC CHỌN NHÂN VẬT ---
+            if char_name:
+                # Tự động viết hoa chữ cái đầu để khớp với URL ảnh của Enka (vd: citlali -> Citlali)
+                formatted_name = char_name.capitalize()
+                # Tìm avatar dựa trên thuộc tính style chứa link ảnh nhân vật
+                char_selector = f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{formatted_name}')]"
+                
+                try:
+                    await page.wait_for_selector(char_selector, timeout=10000)
+                    await page.click(char_selector)
+                    print(f"✅ Đã chọn nhân vật: {formatted_name}")
+                    # Đợi 2s để Enka render lại thông số nhân vật mới
+                    await asyncio.sleep(2)
+                except:
+                    print(f"⚠️ Không tìm thấy nhân vật '{formatted_name}', dùng mặc định.")
 
-            # Upload ImgBB
-            with open(temp_path, "rb") as file:
-                res = requests.post(
-                    "https://api.imgbb.com/1/upload",
-                    data={"key": IMGBB_API_KEY},
-                    files={"image": file.read()},
-                    timeout=60
-                )
+            # --- LOGIC CHỤP CARD ---
+            btn_selector = 'button[data-icon="image"]'
+            await page.wait_for_selector(btn_selector, timeout=20000)
+            await page.click(btn_selector)
+            
+            # Chờ ảnh Blob (Card đã render xong)
+            img_selector = '.Loda img[src^="blob:"]'
+            await page.wait_for_selector(img_selector, timeout=30000)
+            
+            # Chuyển ảnh từ Blob sang Base64
+            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()
-            if os.path.exists(temp_path): os.remove(temp_path)
+
+            # --- UPLOAD IMGBB ---
+            res = requests.post(
+                "https://api.imgbb.com/1/upload",
+                data={"key": IMGBB_API_KEY, "image": img_base64}
+            )
             
-            result = res.json()
-            if result.get('success'):
-                return result['data']['url']
-            raise Exception(f"ImgBB Fail: {result.get('error')}")
+            return res.json().get('data', {}).get('url', "Lỗi Upload")
 
         except Exception as e:
-            if 'browser' in locals(): await browser.close()
-            if os.path.exists(f"card_{uid}.png"): os.remove(f"card_{uid}.png")
-            raise e
+            if 'browser' in locals():
+                await browser.close()
+            return f"Error: {str(e)}"
 
-@app.get("/akasha")
-async def generate(uid: str):
-    # Thử lại tối đa 2 lần
-    for i in range(2):
-        try:
-            url = await run_logic(uid)
-            return {"uid": uid, "card_url": url}
-        except Exception as e:
-            if i == 1: return {"uid": uid, "card_url": f"Error: {str(e)}"}
-            await asyncio.sleep(5)
+# --- ENDPOINTS ---
+
+@app.get("/gen")
+async def generate(uid: str, char: str = None):
+    """
+    Endpoint lấy Card nhân vật.
+    Sử dụng: /gen?uid=890096220&char=citlali
+    """
+    url = await get_enka_data(uid, char)
+    return {
+        "uid": uid, 
+        "char_requested": char or "default", 
+        "card_url": url
+    }
+
+@app.get("/list")
+async def list_chars(uid: str):
+    """Lấy danh sách tên nhân vật đang hiển thị để Bot dễ gọi"""
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True, args=['--no-sandbox'])
+        page = await browser.new_page()
+        await page.goto(f"https://enka.network/u/{uid}/", wait_until="networkidle")
+        
+        chars = await page.evaluate("""
+            () => {
+                const items = document.querySelectorAll('.CharacterList .avatar figure.chara');
+                return Array.from(items).map(el => {
+                    const style = el.getAttribute('style');
+                    const match = style.match(/UI_AvatarIcon_Side_(.*?)\\./);
+                    return match ? match[1] : null;
+                }).filter(n => n !== null);
+            }
+        """)
+        await browser.close()
+        return {"uid": uid, "available_characters": chars}
 
 @app.get("/")
 def home():
-    return {"status": "Aggressive Blocking API is Live"}
+    return {
+        "status": "Enka API Ready",
+        "usage": {
+            "get_card": "/gen?uid=890096220&char=citlali",
+            "list_chars": "/list?uid=890096220"
+        }
+    }