Update main.py

main.py

@@ -160,7 +160,7 @@ async def update():
             content={'error': f'UNKNOWN ERR: {str(e)}', 'details': error_details},
             status_code=500
         )
-        
+
 # Helper function to upload the image to Cloudinary
 def upload_image(data, character_id, player_id):
     try:
@@ -210,6 +210,24 @@ def process_profile(profile_card):
             "url": image_url,
         }
 
+from io import BytesIO
+
+def process_profilee(profile_card):
+    # Ensure profile_card contains an image
+    if not profile_card.cards or not isinstance(profile_card.cards, list):
+        raise ValueError("Invalid profile_card: 'cards' is missing or not a list.")
+
+    profile_image = profile_card.cards[0].card  # Extract the PIL image from the first card
+
+    with BytesIO() as byte_io:
+        profile_image.save(byte_io, "PNG")  # Save the image as PNG
+        byte_io.seek(0)
+        image_url = upload_imagee(byte_io)  # Upload the image
+        return {
+            "url": image_url,
+        }
+
+
 # Process all the images returned
 def process_images(result, player_id):
     characters = []
@@ -222,16 +240,26 @@ def process_images(result, player_id):
                 print(f"Error processing image: {e}")
     return characters
 
+def process_imagess(result, player_id):
+    characters = []
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        futures = [executor.submit(process_image, dt, player_id) for dt in result.cards]
+        for future in concurrent.futures.as_completed(futures):
+            try:
+                characters.append(future.result())
+            except Exception as e:
+                print(f"Error processing image: {e}")
+    return characters
+
 # ZZZ Card and Profile creation functions
 async def zenless_card(uid, character_id=None, character_art=None):
     from zenka import zenka, ZZZError, CacheConfig, Lang
-    
+
     config = zenka.Config(
-        save=True,  # Changed to match your example
         asset_save=True,
         hide_uid=False,
     )
-    
+
     try:
         # Simplified to match your example exactly
         async with zenka.Client(
@@ -252,13 +280,12 @@ async def zenless_card(uid, character_id=None, character_art=None):
 
 async def zenless_profile(uid):
     from zenka import zenka, ZZZError, CacheConfig, Lang
-    
+
     config = zenka.Config(
-        save=True,  # Changed to match your example
         asset_save=True,
         hide_uid=False,
     )
-    
+
     try:
         # Simplified to match your example exactly
         async with zenka.Client(
@@ -284,16 +311,16 @@ async def zenless_characters(uid: int, character_id: str = None, character_art_u
         char_id = None
         if character_id:
             char_id = [int(id) if id.isdigit() else id for id in character_id.split(',')]
-        
+
         # Process character_art from URL to dict if provided
         char_art = None
         if character_id and character_art_url:
             # Create a dictionary mapping character IDs to art URLs
             char_ids = char_id if char_id else []
             char_art = {str(char_ids[0]): character_art_url} if char_ids else None
-        
+
         result = await zenless_card(uid, char_id, char_art)
-        characters = process_images(result, uid)
+        characters = process_imagess(result, uid)
         return JSONResponse(content={'response': characters})
 
     except Exception as e:
@@ -303,11 +330,11 @@ async def zenless_characters(uid: int, character_id: str = None, character_art_u
 async def zenless_profile_route(uid: int):
     try:
         result = await zenless_profile(uid)
-        profile_data = process_profile(result)
+        profile_data = process_profilee(result)
         return JSONResponse(content={'response': profile_data})
 
     except Exception as e:
         return JSONResponse(content={'error': 'UNKNOWN ERR: ' + str(e)}, status_code=500)
-        
+
 if __name__ == "__main__":
     uvicorn.run("main:app", host="0.0.0.0", port=7860, workers=8, timeout_keep_alive=60000)