Update app.py

app.py

@@ -9,7 +9,6 @@ import gradio as gr
 import discord
 from gradio_client import Client
 from PIL import Image
-from discord import app_commands
 from discord.ext import commands
 
 from discord.ui import Button, View
@@ -17,18 +16,18 @@ 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.all()
 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])}.")
     print("------")
-#---------------------------------------------------------------------------------------------------------------------
+
+
 @bot.hybrid_command(
     name="deepfloydif",
     description="Enter a prompt to generate an image! Can generate realistic text, too!",
@@ -41,7 +40,6 @@ async def deepfloydif(ctx, prompt: str):
         print(f"Error: {e}")
 
 
-
 def deepfloydif_generate64_inference(prompt):
     """Generates four images based on a prompt"""
     negative_prompt = ""
@@ -139,9 +137,6 @@ def combine_images(png_files, stage_1_images, partial_path):
     return combined_image_path
 
 
-
-
-
 async def deepfloydif_generate64(ctx, prompt):
     """DeepfloydIF command (generate images with realistic text using slash commands)"""
     try:
@@ -188,9 +183,7 @@ async def deepfloydif_generate64(ctx, prompt):
 
                     # create and use upscale 1024 button
                     with open(result_path, "rb") as f:
-                        upscale1024 = Button(
-                            label="High-quality upscale (x4)", custom_id=str(index)
-                        )  # "0", "1" etc
+                        upscale1024 = Button(label="High-quality upscale (x4)", custom_id=str(index))
                         upscale1024.callback = upscale1024_callback
                         view = View(timeout=None)
                         view.add_item(upscale1024)
@@ -233,8 +226,7 @@ async def deepfloydif_generate64(ctx, prompt):
 
                 # could store this message as combined_image_dfif in case it's useful for future testing
                 await channel.send(
-                    f"**{prompt}** - {ctx.author.mention} Click a button to upscale! (make larger + enhance"
-                    " quality)",
+                    f"**{prompt}** - {ctx.author.mention} Click a button to upscale! (make larger + enhance quality)",
                     file=discord.File(f, f"{partial_path}.png"),
                     view=view,
                 )
@@ -271,7 +263,6 @@ async def deepfloydif_upscale1024(index: int, path_for_upscale256_upscaling, pro
         print(f"Error: {e}")
 
 
-#---------------------------------------------------------------------------------------------------------------------
 def run_bot():
     bot.run(DISCORD_TOKEN)
 
@@ -286,4 +277,4 @@ with gr.Blocks() as demo:
     """)
 demo.queue(concurrency_count=100)
 demo.queue(max_size=100)
-demo.launch()
+demo.launch()