Update petroll.py

petroll.py

@@ -96,7 +96,7 @@ async def perform_roll(interaction: discord.Interaction):
     view = discord.ui.View()
     view.add_item(roll_again_button)
 
-    # Check if the user has auto-sell enabled
+    # Only add the sell button if auto-sell is not active for this user
     if user_id not in auto_sell_users:
         sell_button = discord.ui.Button(style=discord.ButtonStyle.success, label=f"Sell Pet for ${rap_value}", custom_id="sell_pet")
 
@@ -220,27 +220,30 @@ async def perform_roll(interaction: discord.Interaction):
             auto_sell_button.style = discord.ButtonStyle.primary
             auto_sell_button.label = "Auto Pet Sell"
             await interaction.response.send_message("Auto pet sell stopped.", ephemeral=True)
+
+            # Re-add the normal sell button when stopping auto-sell
+            if sell_button not in view.children:
+                view.add_item(sell_button)
         else:
             auto_sell_users.add(user_id)
             auto_sell_button.style = discord.ButtonStyle.danger
             auto_sell_button.label = "Stop Auto Pet Sell"
             await interaction.response.send_message("Auto pet sell started.", ephemeral=True)
 
+            # Automatically add cash for the rolled pet when auto-sell is enabled
+            user_cash[user_id] = user_cash.get(user_id, 0) + rap_value
+            embed.add_field(name="Auto Sell", value=f"Pet automatically sold for ${rap_value}. New balance: ${user_cash[user_id]}", inline=False)
+
         await interaction.message.edit(view=view)
 
     auto_sell_button.callback = auto_sell_callback
     view.add_item(auto_sell_button)
 
-    # Automatically sell the pet and update the balance when auto-sell is enabled.
-    if user_id in auto_sell_users:
-        user_cash[user_id] += rap_value
-        embed.add_field(name="Auto Sell", value=f"Pet automatically sold for ${rap_value}. New balance: ${user_cash[user_id]}", inline=False)
-
     return embed, view
 
 async def schedule_next_luck_opportunity(interaction: discord.Interaction, user_id: int):
     await asyncio.sleep(1800)  # Wait for 30 minutes
-    luck_opportunities[user_id] += 1
+    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]}")
     
     async def increase_luck_callback(interaction: discord.Interaction):
@@ -255,73 +258,65 @@ async def schedule_next_luck_opportunity(interaction: discord.Interaction, user_
         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
+        luck_expiration[user_id] = time.time() + 1800
         
-        used_luck_opportunities.setdefault(user_id, set()).add(luck_opportunities[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)
+        
+        view = interaction.message.components[0]
+        for item in view.children:
+            if item.custom_id == interaction.custom_id:
+                view.remove_item(item)
+                break
+        await interaction.message.edit(view=view)
 
-        view.children.remove(next(item for item in view.children if item.custom.id == increase.luck.button.custom.id))
+        # Schedule the next luck opportunity
+        asyncio.create_task(schedule_next_luck_opportunity(interaction, user_id))
+    
+    increase_luck_button.callback = increase_luck_callback
+    
+    view = interaction.message.components[0]
+    view.add_item(increase_luck_button)
+    await interaction.message.edit(view=view)
+
+async def auto_roll(interaction: discord.Interaction):
+    user_id = interaction.user.id
+    start_time = time.time()
+    while user_id in auto_roll_users:
+        if time.time() - start_time >= 180:  # 3 minutes
+            auto_roll_users.remove(user_id)
+            await interaction.followup.send("auto roll stopped, to turn it on again roll again and turn it on.", ephemeral=True)
+            break
         
-async def schedule_next.lucK.opportunity(interaction: discord.Interaction, user.id: int):
-await asyncio.sleep(1800) # Wait for thirty minutes.
-luck.opportunities[user.id] += one.
-increase.lucK.button equals to discord.ui.button(style equals to discord.buttonstyle.success,label equals to "Increase Luck",custom.id equals to f"increase.lucK.{luck.opportunities[user.id]}")
-
-async def increase.lucK.callback(interaction:discord.interaction):
-if (interaction.user.id!=user.id): 
-await (interaction.response.send.message)("You cannot use this button.",ephemeral equals to true.)
-return.
-if ((user.id)in(used.lucK.opportunities))and(len(used.lucK.opportunities[user.id])>=four): 
-await (interaction.response.send.message)("You have already used all your lucK opportunities.",ephemeral equals to true.)
-return.
-current.lucK equals to lucK.multipliers.get(user.id,one.)
-new.lucK equals to min(current.lucK plus one.,ten.)
-lucK.multipliers[user.id]=new.lucK.
-lucK.expiration[user.id]+=time.time() plus eighteen.hundred.
-used.lucK.opportunities.setdefault(user.id,set()).add(lucK.opportunities[user.id])
-lucK.percentage equals to(new.lucK minus one.)times one.hundred.
-await (interaction.response.send.message)(f"Your lucK has been increased to{lucK.percentage}%for thirty minutes!",ephemeral equals to true.)
-view.children.remove(next(item for item in view.children.if.item.custom.id==increase.lucK.button.custom.id))
-
-view equals to (interaction.message.components[zero])
-view.add.item(increase.lucK.button).
-await (interaction.message.edit)(view equals to view).
-
-async def (auto.roll)(interaction:discord.interaction): 
-user id equals to (interaction.user id).
-start time equals to (time.time())
-while ((user id)in(auto.roll.users)): 
-if((time.time())minus start time)>=one.eighty):#three minutes.
-auto.roll.users.remove(user id).
-await (interaction.followup send)("Auto roll has been automatically stopped after three minutes. You can start it again. If you wish.",ephemeral equals to true.)
-break.
-result equals to(await perform.roll)(interaction).
-if(result): 
-await (interaction.followup send)(embed equals to result[zero],view equals to result[one])
-else: 
-await (interaction.followup send)("An error occurred.")
-await asyncio.sleep(five)#Wait for five seconds between rolls.
-
-@app.commands.command(name equals to "petroll",description equals to "Roll for a random pet")
-async def(petroll)(interaction:discord.interaction): 
-await (interaction response defer)()
-result equals to(await perform.roll)(interaction).
-if(result): 
-await (interaction followup send)(embed equals to result[zero],view.equals.to.result[one])
-else: 
-await (interaction followup send)("An error occurred.")
-
-@app.commands.command(name.equals.to."balance",description.equals.to."Check your current balance")
-async def(balance)(interaction:discord.interaction): 
-user id.equals.to.(interaction.user id).
-current balance.equals.to.(user cash get)(user id.zero.)
-await (interaction response send message)(f"Your current balance is${current.balance}.",ephemeral.equals.to.true.)
-
-@app.commands.command(name.equals.to."dice",description.equals.to."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 i
+        result = await perform_roll(interaction)
+        if result:
+            await interaction.followup.send(embed=result[0], view=result[1])
+        else:
+            await interaction.followup.send("An error occurred.")
+        await asyncio.sleep(5)  # Wait for 5 seconds between rolls
+
+@app_commands.command(name="petroll", description="Roll for a random pet")
+async def petroll(interaction: discord.Interaction):
+    await interaction.response.defer()
+    result = await perform_roll(interaction)
+    if result:
+        await interaction.followup.send(embed=result[0], view=result[1])
+    else:
+        await interaction.followup.send("An error occurred.")
+
+@app_commands.command(name="balance", description="Check your current balance")
+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)
+
+@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_i