Groundspout commited on
Commit
9d3ded5
·
1 Parent(s): 55e02db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -28,14 +28,9 @@ pipe = pipe.to("cpu")
28
 
29
  semaphore = threading.Semaphore(value=1)
30
 
31
- from collections import deque
32
- from typing import Deque, Tuple
33
 
34
- queue: Deque[Tuple[int, discord.Message]] = deque()
35
 
36
- async def update_queue_messages():
37
- for i, (_, message) in enumerate(queue):
38
- await message.edit(content=f"Generating image... (position in queue: {i + 1})")
39
 
40
  def generate_image(prompt, id):
41
  semaphore.acquire()
@@ -62,10 +57,10 @@ async def generate(interaction:discord.Interaction, prompt: str=None):
62
  if not prompt or len(prompt)>100:
63
  await interaction.response.send_message("Please write a description of the image up to 100 characters.")
64
  position = len(queue) + 1
65
- await interaction.response.send_message(f"Generating image... (position in queue: {position})")
66
  response = await interaction.original_response()
67
- queue.append((interaction.user.id, response))
68
- asyncio.run_coroutine_threadsafe(update_queue_messages(), bot.loop)
69
  value = await bot.loop.run_in_executor(executor, generate_image, prompt, interaction.user.id)
70
  image = value[0]
71
  if image:
 
28
 
29
  semaphore = threading.Semaphore(value=1)
30
 
 
 
31
 
 
32
 
33
+
 
 
34
 
35
  def generate_image(prompt, id):
36
  semaphore.acquire()
 
57
  if not prompt or len(prompt)>100:
58
  await interaction.response.send_message("Please write a description of the image up to 100 characters.")
59
  position = len(queue) + 1
60
+ await interaction.response.send_message(f"Generating image... may take some time")
61
  response = await interaction.original_response()
62
+
63
+
64
  value = await bot.loop.run_in_executor(executor, generate_image, prompt, interaction.user.id)
65
  image = value[0]
66
  if image: