Update petroll.py

petroll.py

@@ -10,6 +10,7 @@ luck_multipliers = {}
 luck_expiration = {}
 luck_opportunities = {}
 used_luck_opportunities = set()
+first_luck_claimed = set()
 
 async def perform_roll(interaction: discord.Interaction):
     async def fetch_data(url):
@@ -111,7 +112,29 @@ async def perform_roll(interaction: discord.Interaction):
     sell_button.callback = sell_pet_callback
     view.add_item(sell_button)
 
-    if random.random() < 0.6 and luck_opportunities.get(user_id, 0) < 3:
+    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")
+        
+        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
+            first_luck_claimed.add(user_id)
+            
+            await interaction.response.send_message("You've claimed 100% luck forever!", ephemeral=True)
+            
+            for item in view.children:
+                if item.custom_id == "first_luck":
+                    view.remove_item(item)
+                    break
+            await interaction.message.edit(view=view)
+        
+        first_luck_button.callback = first_luck_callback
+        view.add_item(first_luck_button)
+    
+    elif random.random() < 0.6 and luck_opportunities.get(user_id, 0) < 4:
         luck_opportunities[user_id] = luck_opportunities.get(user_id, 0) + 1
         increase_luck_button = discord.ui.Button(style=discord.ButtonStyle.success, label="Increase Luck", custom_id=f"increase_luck_{luck_opportunities[user_id]}")
         
@@ -120,15 +143,18 @@ async def perform_roll(interaction: discord.Interaction):
                 await interaction.response.send_message("You cannot use this button.", ephemeral=True)
                 return
             
-            if user_id in used_luck_opportunities:
-                await interaction.response.send_message("You have already used your luck opportunity.", ephemeral=True)
+            if user_id in used_luck_opportunities and len(used_luck_opportunities[user_id]) >= 4:
+                await interaction.response.send_message("You have already used all your luck opportunities.", ephemeral=True)
                 return
 
             current_luck = luck_multipliers.get(user_id, 1)
             new_luck = min(current_luck + 1, 10)
             luck_multipliers[user_id] = new_luck
             luck_expiration[user_id] = time.time() + 1800
-            used_luck_opportunities.add(user_id)
+            
+            if user_id not in used_luck_opportunities:
+                used_luck_opportunities[user_id] = set()
+            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)
@@ -157,4 +183,68 @@ async def petroll(interaction: discord.Interaction):
 async def balance(interaction: discord.Interaction):
     user_id = interaction.user.id
     current_balance = user_cash.get(user_id, 0)
-    await interaction.response.send_message(f"Your current balance is ${current_balance}.", ephemeral=True)
+    await interaction.response.send_message(f"Your current balance is ${current_balance}.", ephemeral=True)
+
+@app_commands.command(name="dice", description="Roll the dice and bet")
+async def dice(interaction: discord.Interaction, bet: int):
+    await roll_dice(interaction, bet)
+
+async def roll_dice(interaction: discord.Interaction, bet: int):
+    user_id = interaction.user.id
+    balance = user_cash.get(user_id, 0)
+    
+    if bet <= 0:
+        await interaction.response.send_message("Bet Higher than 0 Idiot.")
+        return
+    
+    if bet > balance:
+        await interaction.response.send_message(f"You don't have enough cash. Your current balance is ${balance:.2f}")
+        return
+    
+    embed = discord.Embed(title="Dice Roll", description=f"{interaction.user.name} is betting ${bet:.2f}", color=0x787878)
+    embed.add_field(name="Current Balance", value=f"${balance:.2f}", inline=False)
+    
+    roll_button = discord.ui.Button(style=discord.ButtonStyle.primary, label="Roll the Dice", custom_id="roll_dice")
+    
+    async def roll_dice_callback(interaction: discord.Interaction):
+        nonlocal balance
+        result = random.choice(["win", "lose"])
+        
+        if result == "win":
+            winnings = bet
+            balance += winnings
+            result_text = f"You won ${winnings:.2f}!"
+        else:
+            balance -= bet
+            result_text = f"You lost ${bet:.2f}."
+        
+        user_cash[user_id] = balance
+        
+        embed.clear_fields()
+        embed.add_field(name="Result", value=result_text, inline=False)
+        embed.add_field(name="New Balance", value=f"${balance:.2f}", inline=False)
+        
+        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):
+            if interaction.user.id == user_id:
+                await roll_dice(interaction, bet)
+            else:
+                await interaction.response.send_message("you cant roll this", ephemeral=True)
+        
+        roll_again_button.callback = roll_again_callback
+        
+        new_view = discord.ui.View()
+        new_view.add_item(roll_again_button)
+        
+        await interaction.response.edit_message(embed=embed, view=new_view)
+    
+    roll_button.callback = roll_dice_callback
+    
+    view = discord.ui.View()
+    view.add_item(roll_button)
+    
+    if interaction.response.is_done():
+        await interaction.followup.send(embed=embed, view=view)
+    else:
+        await interaction.response.send_message(embed=embed, view=view)