dfif test

app.py

@@ -4,7 +4,7 @@ from gradio_client import Client
 import gradio as gr
 import os
 import threading
-......
+
 #for deepfloydif
 import requests
 import json
@@ -22,17 +22,10 @@ import matplotlib.image as mpimg
 # Get Gradio client
 jojogan = gradio_client.Client("akhaliq/JoJoGAN")
 
-#new
-'''
-deepfloydif = gradio_client.Client("DeepFloyd/IF")
-client = Client("DeepFloyd/IF")
-client.view_api()
-
-'''
-
-
-df = Client("DeepFloyd/IF") 
 
+#deepfloydif
+#df = Client("DeepFloyd/IF") #not reliable at the moment
+df = Client("huggingface-projects/IF", hf_token=DFIF_TOKEN)
 
 
 # Set up discord bot 
@@ -40,8 +33,15 @@ class MyClient(discord.Client):
     async def on_ready(self):
         print('Logged on as', self.user)
 
+    #----------------------------------------------------------------------------------------------------
     async def on_message(self, message):
 
+        #tldr, bot should run if
+        #1) it does not have @offline role
+        #2) user has @verified role
+        #3) bot is in #bot-test channel
+        
+
         # if the bot has this role, it won't run
         OFFLINE_ROLE_ID = 1103676632667017266  # 1103676632667017266 = @offline-bot
         guild = message.guild
@@ -63,6 +63,9 @@ class MyClient(discord.Client):
         if message.channel.id not in ALLOWED_CHANNEL_IDS:
             return
 
+        #----------------------------------------------------------------------------------------------------
+        #jojogan
+        
         if message.content.find("!help") != -1:
             await message.reply("Use !jojo !disney !spidey or !sketch. Have fun!", mention_author=True)
 
@@ -90,7 +93,7 @@ class MyClient(discord.Client):
             else:
                 await message.channel.send("No attachments to be found...Can't animify dat! Try sending me an image 😉")
 
-        #new
+        #----------------------------------------------------------------------------------------------------
         #deepfloydif
         '''
         if message.content.startswith('!deepfloydif'):
@@ -108,6 +111,30 @@ class MyClient(discord.Client):
         #deepfloydif
         # start with preset values for now
         if message.content.startswith('!deepfloydif'):
+
+            #predict
+            stage_1_results, stage_1_param_path, stage_1_result_path = df.predict("gradio written on a wall", "blur", 1,4,7.0, 'smart100',50, api_name="/generate64")
+            
+            # Assuming stage_1_results contains the path to the directory
+            png_files = [f for f in os.listdir(stage_1_results) if f.endswith('.png')]
+
+            # Assuming png_files contains the list of all PNG files in the directory
+            if png_files:
+                first_png = png_files[0]
+                first_png_path = os.path.join(stage_1_results, first_png)
+
+            # Send the image file as a Discord attachment
+            with open(first_png_path, 'rb') as f:
+                await message.reply(f'Here is the first generated image', file=discord.File(f, 'first_png.png'))   
+
+
+
+
+
+
+
+
+            '''
             # stage 1
             prompt = "llama"
             negative_prompt = "blue"
@@ -134,7 +161,14 @@ class MyClient(discord.Client):
             
             # Send the image file as a Discord attachment
             with open(img_path, 'rb') as f:
-                await message.reply(f'Here is the generated image', file=discord.File(f, 'generated_image.png'))           
+                await message.reply(f'Here is the generated image', file=discord.File(f, 'generated_image.png'))                
+            
+            
+            
+            
+            
+            '''
+       
             
 
 DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)