[app.py] black + ruff

app.py

@@ -1,25 +1,24 @@
 import discord
 from discord import app_commands
 import gradio as gr
-from gradio_client import Client
 import os
-import threading 
-import asyncio
-import falcon
+import threading
 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
 
 # HF GUILD SETTINGS
-MY_GUILD_ID = 1077674588122648679 if os.getenv("TEST_ENV", False) else 879548962464493619
+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):
         super().__init__(intents=intents)
         self.tree = app_commands.CommandTree(self)
@@ -29,15 +28,20 @@ class MyClient(discord.Client):
         self.tree.copy_global_to(guild=MY_GUILD)
         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('------')
+    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?')
+@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):
     """Command that begins a new conversation with Falcon"""
     try:
@@ -45,43 +49,53 @@ async def falcon(interaction: discord.Interaction, prompt: str):
     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"""
     try:
         await continue_falcon(message)
-        #await message.remove_reaction('🔁', client.user) # test=🔁  hf=🔁  
+        # 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!')
+@app_commands.describe(
+    prompt="Enter a prompt to generate an image! Can generate realistic text, too!"
+)
 async def deepfloydif(interaction: discord.Interaction, prompt: str):
     """DeepfloydIF stage 1 generation"""
     try:
         await deepfloydif_stage_1(interaction, prompt, client)
-
     except Exception as e:
         print(f"Error: {e}")
 
+
 @client.event
-async def on_reaction_add(reaction, user): 
+async def on_reaction_add(reaction, user):
     """Checks for a reaction in order to call dfif2"""
     try:
         await deepfloydif_stage_2_react_check(reaction, user)
-
     except Exception as e:
-        print(f"Error: {e} (known error, does not cause issues, low priority)")        
-        
+        print(f"Error: {e} (known error, does not cause issues, low priority)")
+
+
 """This allows us to run the Discord bot in a Python thread"""
+
+
 def run_bot():
     client.run(DISCORD_TOKEN)
+
+
 threading.Thread(target=run_bot).start()
 with gr.Blocks() as demo:
-    gr.Markdown("""
+    gr.Markdown(
+        """
     # Huggingbots Server
     This space hosts the huggingbots discord bot.
     Currently supported models are Falcon and DeepfloydIF
-    """)
+    """
+    )
 demo.queue(concurrency_count=20)
-demo.launch()
+demo.launch()