Update main.py

main.py

@@ -31,13 +31,22 @@ async def genshin_card(id, designtype,character_id=None, character_art_url=None)
     async with encbanner.ENC(uid=str(id), character_art=character_art) as encard:
         return await encard.creat(akasha=True, template=(2 if str(designtype) == "2" else 1))
 
-# Star Rail card creation with optional character ID
-async def starrail_card(id, designtype, character_id=None, character_art_url=None):
-    # Use the provided character ID and character art URL from user input, if available
+# Star Rail card creation with optional character ID and cookies
+async def starrail_card(id, designtype, character_id=None, character_art_url=None, ltmid_v2=None, ltoken_v2=None, ltuid_v2=None):
     character_art = {str(character_id): character_art_url} if character_id and character_art_url else None
-    
-    async with starrailcard.Card(seeleland=True, remove_logo=True, character_art=character_art,enka=True) as card:
-        return await card.create(id, style=(2 if str(designtype) == "2" else 1))
+    # Use cookies if provided
+    if ltmid_v2 and ltoken_v2 and ltuid_v2:
+        cookie = {
+            "ltmid_v2": ltmid_v2,
+            "ltoken_v2": ltoken_v2,
+            "ltuid_v2": ltuid_v2
+        }
+        async with starrailcard.HoYoCard(cookie=cookie) as card:
+            return await card.create(id, style=(2 if str(designtype) == "2" else 1))
+    else:
+        # Fallback to the existing process
+        async with starrailcard.Card(seeleland=True, remove_logo=True, character_art=character_art, enka=True) as card:
+            return await card.create(id, style=(2 if str(designtype) == "2" else 1))
 
 
 # Star Rail profile creation
@@ -69,17 +78,24 @@ async def genshin_characters(id: int, design: str = "1", character_id: int = Non
 
 # Route for Star Rail with optional character ID
 @app.get("/starrail/{id}")
-async def starrail_characters(id: int, design: str = "1", character_id: int = None, character_art_url: str = None):
+async def starrail_characters(
+    id: int,
+    design: str = "1",
+    character_id: int = None,
+    character_art_url: str = None,
+    ltmid_v2: str = None,
+    ltoken_v2: str = None,
+    ltuid_v2: str = None
+):
     try:
-        # Call starrail_card function with optional character ID and art URL
-        result = await starrail_card(id, design, character_id, character_art_url)
+        # Call starrail_card with cookies if provided
+        result = await starrail_card(id, design, character_id, character_art_url, ltmid_v2, ltoken_v2, ltuid_v2)
         characters = process_images(result, id)
         return JSONResponse(content={'response': characters})
 
     except Exception as e:
         return JSONResponse(content={'error': 'UNKNOWN ERR: ' + str(e)}, status_code=500)
 
-
 # Route for Star Rail profile
 @app.get("/starrail/profile/{id}")
 async def starrail_profile_route(id: int):