[main] more cleanup

app.py

@@ -4,24 +4,20 @@ import gradio as gr
 from gradio_client import Client
 import os
 import threading 
-#import json
-#import random
-#from PIL import Image
 import asyncio
-#import glob
-
 import falcon
 from falcon import try_falcon
 from falcon import continue_falcon
-
 import deepfloydif
 from deepfloydif import deepfloydif_stage_1
 from deepfloydif import deepfloydif_stage_2
 from deepfloydif import deepfloydif_stage_2_react_check
-#-------------------------------------------------------------------------------------------------------------------------------------
-MY_GUILD = discord.Object(id=1077674588122648679)  # HF = 879548962464493619, test = 1077674588122648679
+
+# HF GUILD SETTINGS
+MY_GUILD_ID = 1077674588122648679 if os.getenv("TEST_ENV", False) else 879548962464493619
+MY_GUILD = discord.Object(id=MY_GUILD_ID)
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
-#-------------------------------------------------------------------------------------------------------------------------------------
+
 class MyClient(discord.Client):
     """This structure allows slash commands to work instantly (instead of needing to sync global commands for up to an hour)"""
     def __init__(self, *, intents: discord.Intents):
@@ -34,12 +30,12 @@ class MyClient(discord.Client):
         await self.tree.sync(guild=MY_GUILD)
 
 client = MyClient(intents=discord.Intents.all())
-#-------------------------------------------------------------------------------------------------------------------------------------
+
 @client.event
 async def on_ready():
     print(f'Logged in as {client.user} (ID: {client.user.id})')
     print('------')
-#------------------------------------------------------------------------------------------------------------------------------------- 
+
 @client.tree.command()
 @app_commands.describe(prompt='Enter some text to chat with the bot! Like this: /falcon Hello, how are you?')
 async def falcon(interaction: discord.Interaction, prompt: str):
@@ -48,7 +44,7 @@ async def falcon(interaction: discord.Interaction, prompt: str):
         await try_falcon(interaction, prompt)
     except Exception as e:
         print(f"Error: {e}")
-#-------------------------------------------------------------------------------------------------------------------------------------
+
 @client.event
 async def on_message(message):
     """Checks channel and continues Falcon conversation if it's the right Discord Thread"""
@@ -57,7 +53,7 @@ async def on_message(message):
         await message.remove_reaction('🔁', client.user) # test=🔁  hf=🔁  
     except Exception as e:
         print(f"Error: {e}")
-#------------------------------------------------------------------------------------------------------------------------------------- 
+
 @client.tree.command()
 @app_commands.describe(prompt='Enter a prompt to generate an image! Can generate realistic text, too!')
 async def deepfloydif(interaction: discord.Interaction, prompt: str):
@@ -67,7 +63,7 @@ async def deepfloydif(interaction: discord.Interaction, prompt: str):
 
     except Exception as e:
         print(f"Error: {e}")
-#-------------------------------------------------------------------------------------------------------------------------------------   
+
 @client.event
 async def on_reaction_add(reaction, user): 
     """Checks for a reaction in order to call dfif2"""
@@ -76,9 +72,8 @@ async def on_reaction_add(reaction, user):
 
     except Exception as e:
         print(f"Error: {e} (known error, does not cause issues, low priority)")        
-#-------------------------------------------------------------------------------------------------------------------------------------        
+        
 """This allows us to run the Discord bot in a Python thread"""
-DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 def run_bot():
     client.run(DISCORD_TOKEN)
 threading.Thread(target=run_bot).start()