Update app.py

app.py

@@ -5,6 +5,7 @@ import asyncio
 from fastapi import FastAPI
 import uvicorn
 import random
+import time
 
 app = FastAPI()
 intents = discord.Intents.default()
@@ -12,6 +13,9 @@ intents.message_content = True
 bot = discord.Client(intents=intents)
 tree = app_commands.CommandTree(bot)
 
+luck_multipliers = {}
+luck_expiration = {}
+
 @app.get("/")
 async def read_root():
     return {"Hello": "World"}
@@ -88,8 +92,11 @@ async def perform_roll(interaction: discord.Interaction):
     if not pets:
         return None
 
+    user_id = interaction.user.id
+    luck_multiplier = luck_multipliers.get(user_id, 1)
+    
     # Calculate total difficulty
-    total_difficulty = sum(1 / pet['configData']['difficulty'] for pet in pets)
+    total_difficulty = sum(1 / (pet['configData']['difficulty'] / luck_multiplier) for pet in pets)
 
     # Generate a random number between 0 and total_difficulty
     random_value = random.uniform(0, total_difficulty)
@@ -97,7 +104,7 @@ async def perform_roll(interaction: discord.Interaction):
     # Select a pet based on the random value
     cumulative_probability = 0
     for pet in pets:
-        cumulative_probability += 1 / pet['configData']['difficulty']
+        cumulative_probability += 1 / (pet['configData']['difficulty'] / luck_multiplier)
         if random_value <= cumulative_probability:
             rolled_pet = pet
             break
@@ -121,15 +128,25 @@ async def perform_roll(interaction: discord.Interaction):
         else:
             return f"{difficulty} ({difficulty:,})"
 
-    embed = discord.Embed(title=f"You rolled: {rolled_pet['configData']['name']}", color=0x787878)
+    embed = discord.Embed(title=f"{interaction.user.name} rolled: {rolled_pet['configData']['name']}", color=0x787878)
     embed.add_field(name="value", value=f"{rap_value:,} diamonds", inline=True)
     embed.add_field(name="difficulty", value=format_difficulty(rolled_pet['configData']['difficulty']), inline=True)
     embed.add_field(name="category", value=rolled_pet['category'], inline=True)
     embed.set_thumbnail(url=thumbnail_url)
-    embed.set_footer(text="Click 'Roll Again' to roll again!")
+    
+    luck_text = ""
+    if user_id in luck_expiration:
+        remaining_time = int(luck_expiration[user_id] - time.time())
+        if remaining_time > 0:
+            luck_text = f"\nYou have {remaining_time // 60} minutes and {remaining_time % 60} seconds of luck left!"
+        else:
+            del luck_multipliers[user_id]
+            del luck_expiration[user_id]
+    
+    embed.set_footer(text=f"Click 'Roll Again' to roll again!{luck_text}")
 
     roll_again_button = discord.ui.Button(style=discord.ButtonStyle.primary, label="Roll Again", custom_id="roll_again")
-
+    
     async def roll_again_callback(interaction: discord.Interaction):
         await interaction.response.defer()
         result = await perform_roll(interaction)
@@ -143,6 +160,21 @@ async def perform_roll(interaction: discord.Interaction):
     view = discord.ui.View()
     view.add_item(roll_again_button)
 
+    if random.random() < 0.2:  
+        increase_luck_button = discord.ui.Button(style=discord.ButtonStyle.success, label="Increase Luck", custom_id="increase_luck")
+        
+        async def increase_luck_callback(interaction: discord.Interaction):
+            user_id = interaction.user.id
+            if user_id not in luck_multipliers:
+                luck_multipliers[user_id] = 2
+                luck_expiration[user_id] = time.time() + 1800  # 30 minutes
+            else:
+                luck_multipliers[user_id] *= 10
+            await interaction.response.send_message(f"luck increased to: {luck_multipliers[user_id]}x", ephemeral=True)
+        
+        increase_luck_button.callback = increase_luck_callback
+        view.add_item(increase_luck_button)
+
     return embed, view
 
 @tree.command(name="petroll", description="Roll for a random pet")