Update app.py

app.py

@@ -88,7 +88,20 @@ async def perform_roll(interaction: discord.Interaction):
     if not pets:
         return None
 
-    rolled_pet = random.choice(pets)
+    # Calculate total difficulty
+    total_difficulty = sum(1 / pet['configData']['difficulty'] for pet in pets)
+
+    # Generate a random number between 0 and total_difficulty
+    random_value = random.uniform(0, total_difficulty)
+
+    # Select a pet based on the random value
+    cumulative_probability = 0
+    for pet in pets:
+        cumulative_probability += 1 / pet['configData']['difficulty']
+        if random_value <= cumulative_probability:
+            rolled_pet = pet
+            break
+
     pet_rap = next((pet for pet in rap_data['data'] if pet['configData']['id'] == rolled_pet['configName']), None)
 
     if not pet_rap: