try again

app.py

@@ -3,35 +3,43 @@ import glob
 import os
 import pathlib
 import random
-import os
-import random
 import threading
-from threading import Event
-from typing import Optional
-from discord import Permissions
-from discord.ext import commands
-from discord.utils import oauth_url
-
 
-import discord
 import gradio as gr
-
 import discord
 from gradio_client import Client
 from PIL import Image
 
 from discord.ui import Button, View
 
-
-#---------------------------------------------------------------------------------------------------------------------
-HF_TOKEN = os.getenv("HF_TOKEN")
 deepfloydif_client = Client("huggingface-projects/IF", HF_TOKEN)
-#---------------------------------------------------------------------------------------------------------------------
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
+
+#---------------------------------------------------------------------------------------------------------------------
 intents = discord.Intents.default()
 intents.message_content = True
 bot = commands.Bot(command_prefix="/", intents=intents)
 #---------------------------------------------------------------------------------------------------------------------
+@bot.event
+async def on_ready():
+    print(f"Logged in as {bot.user} (ID: {bot.user.id})")
+    synced = await bot.tree.sync()
+    print(f"Synced commands: {', '.join([s.name for s in synced])}.")
+    event.set()
+    print("------")
+#---------------------------------------------------------------------------------------------------------------------
+@bot.hybrid_command(
+    name="deepfloydif",
+    description="Enter a prompt to generate an image! Can generate realistic text, too!",
+)
+async def deepfloydif(ctx, prompt: str):
+    """DeepfloydIF stage 1 generation"""
+    try:
+        await deepfloydif_generate64(ctx, prompt)
+    except Exception as e:
+        print(f"Error: {e}")
+
+
 
 def deepfloydif_generate64_inference(prompt):
     """Generates four images based on a prompt"""
@@ -130,23 +138,12 @@ def combine_images(png_files, stage_1_images, partial_path):
     return combined_image_path
 
 
-@bot.hybrid_command(
-    name="deepfloydif",
-    description="Enter a prompt to generate an image! Can generate realistic text, too!",
-)
-async def deepfloydif(ctx, prompt: str):
-    """DeepfloydIF stage 1 generation"""
-    try:
-        await deepfloydif_generate64(ctx, prompt, bot)
-    except Exception as e:
-        print(f"Error: {e}")
 
 
-async def deepfloydif_generate64(ctx, prompt, bot):
+
+async def deepfloydif_generate64(ctx, prompt):
     """DeepfloydIF command (generate images with realistic text using slash commands)"""
     try:
-        if os.environ.get("TEST_ENV") == "True":
-            print("Safety checks passed for deepfloydif_generate64")
         channel = bot.get_channel(DEEPFLOYDIF_CHANNEL_ID)
         # interaction.response message can't be used to create a thread, so we create another message
         message = await ctx.send(f"**{prompt}** - {ctx.author.mention} <a:loading:1114111677990981692>")
@@ -275,20 +272,17 @@ async def deepfloydif_upscale1024(index: int, path_for_upscale256_upscaling, pro
 
 #---------------------------------------------------------------------------------------------------------------------
 def run_bot():
-    if not DISCORD_TOKEN:
-        print("DISCORD_TOKEN NOT SET")
-    else:
-        bot.run(DISCORD_TOKEN)
+    client.run(DISCORD_TOKEN)
 
 
 threading.Thread(target=run_bot).start()
-
+"""This allows us to run the Discord bot in a Python thread"""
 with gr.Blocks() as demo:
-    gr.Markdown(
-    """
-    # Discord bot of https://huggingface.co/spaces/facebook/MusicGen
-    https://discord.com/api/oauth2/authorize?client_id=1151888750662664222&permissions=309237696512&scope=bot
-    """
-    )
-
+    gr.Markdown("""
+    # Huggingbots Server
+    This space hosts the huggingbots discord bot.
+    Currently supported models are Falcon and DeepfloydIF
+    """)
+demo.queue(concurrency_count=100)
+demo.queue(max_size=100)
 demo.launch()