Merge pull request #28 from Nexus-AI-Bot/calculate-fix

Fixed issue

cool_functions.py

@@ -58,14 +58,19 @@ def password(msg):
   except:
     return 'Error, not a number'
 
-def math(msg):
-    math = msg
-    try:
-        tree = ast.parse(math, mode='eval')
-        solution = ast.literal_eval(tree.body)
-        return solution
-    except Exception:
-        return "You wrote that wrong:("
+def math(num1, num2, op):
+    if op == '+':
+        return num1 + num2
+    elif op == '-':
+        return num1 - num2
+    elif op == 'x':
+        return num1 * num2
+    elif op == '/':
+        if num2 == 0:
+            return "Division by zero is not allowed."
+        return num1 / num2
+    else:
+        return "Invalid operator."
 
 def math_ran():
   random_number = random.randint(1, 4)

main.py

@@ -419,15 +419,15 @@ async def self(interaction: discord.Interaction):
 
 @logger.log
 @tree.command(name="calculate", description="Calculate a math problem")
-async def self(interaction: discord.Interaction, math_problem: str):
+async def self(interaction: discord.Interaction, num1: int, num2: int, operator: str):
   id = interaction.guild.id
   json = {
     'guild_id':id,
     'command_name': interaction.command.name
   }
   request = requests.post('https://nexus-ai.xyz/bot/command/query', json=json)
   if request.text == 'True':
-    await interaction.response.send_message(f.math(math_problem))
+    await interaction.response.send_message(f.math(num1, num2, operator))
   else:
     await interaction.response.send_message('This command has been disabled')
 
@@ -929,7 +929,7 @@ async def drink_autocompletion(interaction: discord.Interaction, current: str) -
 @logger.log
 @tree.command(name="sell", description="Sell your pet")
 @app_commands.autocomplete(pet=drink_autocompletion)
-async def sell(interaction: discord.Interaction, pet: str):
+async def self(interaction: discord.Interaction, pet: str):
 
   id = interaction.guild.id
   json = {