Update petroll.py

petroll.py

@@ -4,9 +4,12 @@ import aiohttp
 import random
 import time
 
+from cash import user_cash
+
 luck_multipliers = {}
 luck_expiration = {}
 luck_opportunities = {}
+used_luck_opportunities = set()
 
 async def perform_roll(interaction: discord.Interaction):
     async def fetch_data(url):
@@ -57,9 +60,9 @@ async def perform_roll(interaction: discord.Interaction):
             return f"{difficulty} ({difficulty:,})"
 
     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.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)
     
     luck_text = ""
@@ -82,29 +85,53 @@ async def perform_roll(interaction: discord.Interaction):
         if result:
             await interaction.followup.send(embed=result[0], view=result[1])
         else:
-            await interaction.followup.send("errer.")
+            await interaction.followup.send("An error occurred.")
 
     roll_again_button.callback = roll_again_callback
 
     view = discord.ui.View()
     view.add_item(roll_again_button)
 
-    if random.random() < 0.2:
+    sell_button = discord.ui.Button(style=discord.ButtonStyle.success, label=f"Sell Pet for ${rap_value // 100}", custom_id="sell_pet")
+
+    async def sell_pet_callback(interaction: discord.Interaction):
+        if interaction.user.id != user_id:
+            await interaction.response.send_message("You cannot sell this pet.", ephemeral=True)
+            return
+
+        sell_value = rap_value // 100
+        user_cash[user_id] = user_cash.get(user_id, 0) + sell_value
+        await interaction.response.send_message(f"You sold the pet for ${sell_value}. Your new balance is ${user_cash[user_id]}.", ephemeral=True)
+        for item in view.children:
+            if item.custom_id == "sell_pet":
+                view.remove_item(item)
+                break
+        await interaction.message.edit(view=view)
+
+    sell_button.callback = sell_pet_callback
+    view.add_item(sell_button)
+
+    if random.random() < 0.2 and user_id not in used_luck_opportunities:
         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):
             if interaction.user.id != user_id:
-                await interaction.response.send_message("cannot use", ephemeral=True)
+                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", 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)
             
             luck_percentage = (new_luck - 1) * 100
-            await interaction.response.send_message(f"luck increased to {luck_percentage}% for 30 minutes!", ephemeral=True)
+            await interaction.response.send_message(f"Your luck has been increased to {luck_percentage}% for 30 minutes!", ephemeral=True)
             
             for item in view.children:
                 if item.custom_id == interaction.custom_id:
@@ -124,4 +151,10 @@ async def petroll(interaction: discord.Interaction):
     if result:
         await interaction.followup.send(embed=result[0], view=result[1])
     else:
-        await interaction.followup.send("errer")
+        await interaction.followup.send("errer")
+
+@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)