Update petroll.py

petroll.py

@@ -34,12 +34,31 @@ async def perform_roll(interaction: discord.Interaction):
     user_id = interaction.user.id
     luck_multiplier = luck_multipliers.get(user_id, 1)
     
+    # Sort pets by difficulty (rarity)
     sorted_pets = sorted(pets, key=lambda x: x['configData']['difficulty'])
     
-    max_index = len(sorted_pets) - 1
-    index = int(max_index * (luck_multiplier - 1) / 9)
+    # Divide pets into tiers
+    tier_size = len(sorted_pets) // 5
+    tiers = [sorted_pets[i:i+tier_size] for i in range(0, len(sorted_pets), tier_size)]
     
-    rolled_pet = random.choice(sorted_pets[:index+1])
+    # Adjust probabilities based on luck
+    tier_probabilities = [0.5, 0.25, 0.15, 0.07, 0.03]
+    luck_boost = (luck_multiplier - 1) * 0.1  # 10% boost per luck level
+    for i in range(len(tier_probabilities)):
+        if i < len(tier_probabilities) - 1:
+            tier_probabilities[i] -= luck_boost
+            tier_probabilities[i+1] += luck_boost
+    
+    # Ensure probabilities are non-negative and sum to 1
+    tier_probabilities = [max(0, p) for p in tier_probabilities]
+    total = sum(tier_probabilities)
+    tier_probabilities = [p / total for p in tier_probabilities]
+    
+    # Select tier based on probabilities
+    selected_tier = random.choices(tiers, weights=tier_probabilities)[0]
+    
+    # Select pet from the chosen tier
+    rolled_pet = random.choice(selected_tier)
 
     pet_rap = next((pet for pet in rap_data['data'] if pet['configData']['id'] == rolled_pet['configName']), None)
 
@@ -70,7 +89,7 @@ async def perform_roll(interaction: discord.Interaction):
     if user_id in luck_expiration:
         remaining_time = int(luck_expiration[user_id] - time.time())
         if remaining_time > 0:
-            luck_percentage = (luck_multiplier - 1) * 100
+            luck_percentage = (luck_multiplier - 1) * 10
             luck_text = f"\nYou have {remaining_time // 60} minutes and {remaining_time % 60} seconds of luck left! ({luck_percentage}% luck)"
         else:
             del luck_multipliers[user_id]
@@ -113,17 +132,17 @@ async def perform_roll(interaction: discord.Interaction):
     view.add_item(sell_button)
 
     if user_id not in first_luck_claimed:
-        first_luck_button = discord.ui.Button(style=discord.ButtonStyle.success, label="Claim 100% Luck Forever", custom_id="first_luck")
+        first_luck_button = discord.ui.Button(style=discord.ButtonStyle.success, label="Claim 10% Luck Forever", custom_id="first_luck")
         
         async def first_luck_callback(interaction: discord.Interaction):
             if interaction.user.id != user_id:
                 await interaction.response.send_message("You cannot use this button.", ephemeral=True)
                 return
             
-            luck_multipliers[user_id] = 10  # 100% luck
+            luck_multipliers[user_id] = 2  # 10% luck
             first_luck_claimed.add(user_id)
             
-            await interaction.response.send_message("You've claimed 100% luck forever!", ephemeral=True)
+            await interaction.response.send_message("You've claimed 10% luck forever!", ephemeral=True)
             
             for item in view.children:
                 if item.custom_id == "first_luck":
@@ -148,7 +167,7 @@ async def perform_roll(interaction: discord.Interaction):
                 return
 
             current_luck = luck_multipliers.get(user_id, 1)
-            new_luck = min(current_luck + 1, 10)
+            new_luck = min(current_luck + 0.1, 2)  # Max 20% luck
             luck_multipliers[user_id] = new_luck
             luck_expiration[user_id] = time.time() + 1800
             
@@ -157,7 +176,7 @@ async def perform_roll(interaction: discord.Interaction):
             used_luck_opportunities[user_id].add(luck_opportunities[user_id])
             
             luck_percentage = (new_luck - 1) * 100
-            await interaction.response.send_message(f"Your luck has been increased to {luck_percentage}% for 30 minutes!", ephemeral=True)
+            await interaction.response.send_message(f"Your luck has been increased to {luck_percentage:.1f}% for 30 minutes!", ephemeral=True)
             
             for item in view.children:
                 if item.custom_id == interaction.custom_id: