Update main.py

main.py

@@ -89,6 +89,25 @@ def upload_image(data, public_id=None):
     except Exception as e:
         raise Exception(f"Cloudinary upload error: {str(e)}")
 
+# Helper functions to create genshin and starrail cards/profiles
+async def genshin_card(id, designtype, character_id=None, character_art_url=None):
+    character_art = {str(character_id): character_art_url} if character_id and character_art_url else None
+    async with enkacard.ENC(uid=str(id), character_art=character_art) as encard:
+        return await encard.creat(akasha=True, template=(2 if str(designtype) == "2" else 1))
+
+async def starrail_card(id, designtype, character_id=None, character_art_url=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) as card:
+        return await card.create(id, style=(2 if str(designtype) == "2" else 1))
+
+async def genshinprofile(id):
+    async with enkacard.ENC(uid=id) as encard:
+        return await encard.profile(card=True)
+
+async def starrail_profile(id):
+    async with starrailcard.Card(remove_logo=True, seeleland=True) as card:
+        return await card.create_profile(id, style=2)
+
 # Utility to process images in card generation endpoints
 def process_images(result, player_id):
     characters = []
@@ -141,7 +160,7 @@ async def decrypt_and_upload(
     except Exception as e:
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
-# Genshin Impact and Star Rail card/profile creation endpoints (Example for Genshin)
+# Genshin Impact and Star Rail card/profile creation endpoints
 @app.get("/genshin/{id}")
 async def genshin_characters(id: int, design: str = "1", character_id: int = None, character_art_url: str = None):
     try:
@@ -160,6 +179,24 @@ async def genshin_profile_route(id: int):
     except Exception as e:
         return JSONResponse(content={'error': str(e)}, status_code=500)
 
+@app.get("/starrail/{id}")
+async def starrail_characters(id: int, design: str = "1", character_id: int = None, character_art_url: str = None):
+    try:
+        result = await starrail_card(id, design, character_id, character_art_url)
+        characters = process_images(result, id)
+        return JSONResponse(content={'response': characters})
+    except Exception as e:
+        return JSONResponse(content={'error': str(e)}, status_code=500)
+
+@app.get("/starrail/profile/{id}")
+async def starrail_profile_route(id: int):
+    try:
+        result = await starrail_profile(id)
+        profile_data = process_profile(result)
+        return JSONResponse(content={'response': profile_data})
+    except Exception as e:
+        return JSONResponse(content={'error': str(e)}, status_code=500)
+
 # Character AI message endpoint
 @app.get("/send_message")
 async def send_message(