[deepfloydif.py] creation

deepfloydif.py

@@ -15,6 +15,9 @@ MY_GUILD = discord.Object(id=MY_GUILD_ID)
 HF_TOKEN = os.getenv('HF_TOKEN')
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", Nonedeepfloyd_client = Client("huggingface-projects/IF", HF_TOKEN)
 
+BOT_USER_ID = 1086256910572986469
+DEEPFLOYD_CHANNEL_ID = 1121834257959092234
+
 #-------------------------------------------------------------------------------------------------------------------------------------  
 # deepfloydif stage 1 generation
 def inference(prompt):
@@ -31,11 +34,7 @@ def inference(prompt):
     
     return [stage_1_results, stage_1_param_path, stage_1_result_path]        
 #------------------------------------------------------------------------------------------------------------------------------------- 
-# deepfloydIF command (generate images with realistic text using slash commands)
-@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):
+async def try_deepfloydif(interaction, prompt):
     thread = None
     try:
         global BOT_USER_ID
@@ -75,23 +74,6 @@ async def deepfloydif(interaction: discord.Interaction, prompt: str):
                     img3 = images[2]
                     img4 = images[3]
 
-                    
-                    '''
-                    first_png = png_files[0]
-                    second_png = png_files[1]
-                    third_png = png_files[2]
-                    fourth_png = png_files[3]
-                
-                    first_png_path = os.path.join(stage_1_results, first_png)
-                    second_png_path = os.path.join(stage_1_results, second_png)
-                    third_png_path = os.path.join(stage_1_results, third_png)
-                    fourth_png_path = os.path.join(stage_1_results, fourth_png)
-                
-                    img1 = Image.open(first_png_path)
-                    img2 = Image.open(second_png_path)
-                    img3 = Image.open(third_png_path)
-                    img4 = Image.open(fourth_png_path)
-                    '''
                     combined_image = Image.new('RGB', (img1.width * 2, img1.height * 2))
                 
                     combined_image.paste(img1, (0, 0))
@@ -114,9 +96,6 @@ async def deepfloydif(interaction: discord.Interaction, prompt: str):
 
     except Exception as e:
         print(f"Error: {e}")
-        await message.remove_reaction('🔁', client.user)
-        await message.add_reaction('❌')
-        #await thread.send(f"Error: {e} <@811235357663297546> (continue_falcon error)")
 #-------------------------------------------------------------------------------------------------------------------------------------   
 def load_image(png_files, stage_1_results, png_file_index):
     for file in png_files: