Update app.py

app.py

@@ -167,8 +167,17 @@ or
 
 @client.event
 async def on_message(message):
-    if message.content.startswith("--setup "):
-        args = message.content.split()[1:]
+    msgchannel = message.channel
+    try:
+        msgchannel_name = msgchannel.name
+        guild = message.guild
+        guild_name = guild.name
+    except:
+        guild_name = "Direct"
+        msgchannel_name = str(message.author)
+    if message.content.startswith("--setup"):
+        args = message.content.split()
+        del args[0]
         if len(args) == 0:
             embed = discord.Embed(title="Error", description="No args provided. Use `--help` for command help.")
             await message.reply(embed=embed)
@@ -186,6 +195,8 @@ async def on_message(message):
                 return 0
             c = client.get_channel(cid)
             if c in message.guild.text_channels:
+                with open(f"{guild_name}.guild", "a") as f:
+                    f.write(f"{c.id}\n")
                 embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{cid}>")
                 await message.reply(embed=embed)
                 return 0
@@ -203,6 +214,8 @@ async def on_message(message):
             try:
                 c = discord.utils.get(client.get_all_channels(), guild__id=message.guild.id, name=cname)
                 if c != None:
+                    with open(f"{guild_name}.guild", "a") as f:
+                        f.write(f"{c.id}\n")
                     embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{c.id}>")
                     message.reply(embed=embed)
                     return 0
@@ -255,14 +268,6 @@ async def on_message(message):
             message.content = message.content.replace(f"<@{user.id}>",
                                                   f"<@{str(user)}>")
     try:
-        msgchannel = message.channel
-        try:
-            msgchannel_name = msgchannel.name
-            guild = message.guild
-            guild_name = guild.name
-        except:
-            guild_name = "Direct"
-            msgchannel_name = str(message.author)
         s = f":green[{message.author}]: :violet[{message.content}] :blue[{msgchannel_name}] :orange[{guild_name}]"
         if message.author == client.user:
             return
@@ -312,7 +317,9 @@ async def on_message(message):
                     )
         with open(f"data/{guild_name}/{msgchannel_name}", "r") as f:
             context = f.read()
-        if ("Feeway" in guild_name) or guild_name == "Direct":
+        with open(f"{guild_name}.guild", "r") as f:
+            o = f.read()
+        if str(message.channel.id) in o.split("\n") or guild_name == "Direct":
             async with msgchannel.typing():
                 context += f"GPT4 Correct Assistant:"
                 load = random.choice(