diff --git a/scripts/discordbot.py b/scripts/discordbot.py
index 0628162..3f6475e 100644
--- a/scripts/discordbot.py
+++ b/scripts/discordbot.py
@@ -64,8 +64,13 @@ async def on_message(message):
     if message.author == bot.user:
         return
 
-    if bot.user in message.mentions:
-        prompt = message.content.replace(f'<@{bot.user.id}>', '').strip()
+    # Respond to DMs or when mentioned in a server
+    if isinstance(message.channel, discord.DMChannel) or bot.user in message.mentions:
+        # For mentions, remove the bot mention from the message
+        if bot.user in message.mentions:
+            prompt = message.content.replace(f'<@{bot.user.id}>', '').strip()
+        else:
+            prompt = message.content.strip()
         
         if not prompt:
             await message.channel.send("Hello! How can I help you?")