Update main.py

main.py

@@ -3,44 +3,55 @@ import requests
 import time
 from fastapi import FastAPI
 from playwright.async_api import async_playwright
+from contextlib import asynccontextmanager
 
-app = FastAPI()
-
-# Cấu hình API mới
+# Cấu hình API
 FREEIMAGE_API_KEY = "6d207e02198a847aa98d0a2a901485a5"
 FREEIMAGE_URL = "https://freeimage.host/api/1/upload"
 
-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', '--disable-gpu'
-    ])
-    context = await browser.new_context(
+# Biến toàn cục để giữ trình duyệt
+browser_manager = {"browser": None, "playwright": None}
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Khởi tạo trình duyệt khi Server bắt đầu
+    browser_manager["playwright"] = await async_playwright().start()
+    browser_manager["browser"] = await browser_manager["playwright"].chromium.launch(
+        headless=True, 
+        args=['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage']
+    )
+    print("🚀 Browser đã sẵn sàng!")
+    yield
+    # Đóng trình duyệt khi Server tắt
+    await browser_manager["browser"].close()
+    await browser_manager["playwright"].stop()
+
+app = FastAPI(lifespan=lifespan)
+
+async def get_fast_page(fast_mode=False):
+    # Tạo context mới (nhanh hơn tạo browser mới)
+    context = await browser_manager["browser"].new_context(
         viewport={'width': 1280, 'height': 800},
         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()
-
-    async def block_waste(route):
-        bad_types = ["media", "font", "other"]
-        if fast_mode: bad_types.extend(["image", "stylesheet"])
-        if any(x in route.request.url for x in ["google", "analytics", "doubleclick"]):
-            await route.abort()
-        else:
-            await route.continue_()
-
-    await page.route("**/*", block_waste)
-    await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=60000)
-    return browser, page
+    
+    if fast_mode:
+        await page.route("**/*", lambda route: route.abort() if route.request.resource_type in ["media", "font"] else route.continue_())
+    
+    return context, page
 
 @app.get("/info")
 async def info(uid: str):
     start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=True)
+    context, page = await get_fast_page(fast_mode=True)
     try:
+        await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=30000)
+        
+        # Click Namecard nhanh
         try:
-            await page.click('button[data-icon="cards"]', timeout=2000)
-            await page.wait_for_selector('.Loda .bg', timeout=2000)
+            await page.click('button[data-icon="cards"]', timeout=1500)
+            await page.wait_for_selector('.Loda .bg', timeout=1500)
         except: pass
         
         data = await page.evaluate("""
@@ -81,30 +92,29 @@ async def info(uid: str):
                 };
             }
         """)
-        await browser.close()
+        await context.close()
         return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, **data}
     except Exception as e:
-        if 'browser' in locals(): await browser.close()
+        await context.close()
         return {"error": str(e)}
 
 @app.get("/gen")
 async def generate(uid: str, char: str = None):
     start_time = time.time()
-    browser, page = await get_page(uid, fast_mode=False)
+    context, page = await get_fast_page(fast_mode=False)
     try:
+        await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=30000)
+        
         if char:
             target = char.strip().capitalize()
             try:
-                # Tìm chính xác avatar chứa tên nhân vật
-                await page.click(f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{target}')]", timeout=5000)
-                await asyncio.sleep(0.5) # Giảm tối đa thời gian chờ render
+                await page.click(f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{target}')]", timeout=3000)
+                await asyncio.sleep(0.4)
             except: pass
 
         await page.click('button[data-icon="image"]')
-        # Chờ ảnh blob sẵn sàng
-        await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=15000)
+        await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=10000)
         
-        # Chuyển blob sang base64 ngay trong trình duyệt
         img_base64 = await page.evaluate("""
             async () => {
                 const img = document.querySelector('.Loda img');
@@ -117,34 +127,19 @@ async def generate(uid: str, char: str = None):
                 });
             }
         """)
-        await browser.close()
+        await context.close()
 
-        # Gọi API FreeImage.host
-        res = requests.post(
-            FREEIMAGE_URL,
-            data={
-                "key": FREEIMAGE_API_KEY,
-                "action": "upload",
-                "source": img_base64,
-                "format": "json"
-            },
-            timeout=20
-        )
-        
-        # Lấy link ảnh từ cấu trúc JSON của FreeImage
-        json_res = res.json()
-        card_url = json_res.get('image', {}).get('url', "Upload Failed")
+        res = requests.post(FREEIMAGE_URL, data={"key": FREEIMAGE_API_KEY, "action": "upload", "source": img_base64, "format": "json"}, timeout=15)
         
         return {
             "execution_time": f"{round(time.time() - start_time, 2)}s",
             "uid": uid,
-            "char": char or "default",
-            "card_url": card_url
+            "card_url": res.json().get('image', {}).get('url')
         }
     except Exception as e:
-        if 'browser' in locals(): await browser.close()
+        await context.close()
         return {"error": str(e)}
 
 @app.get("/")
 def home():
-    return {"status": "Enka API Speed-Optimized with FreeImage.host"}
+    return {"status": "Warm-up Browser API Live"}