diff --git a/bot.py b/bot.py index 60c4164..5ac5128 100644 --- a/bot.py +++ b/bot.py @@ -298,6 +298,10 @@ async def queue_info(ev): waiting_usrs = conn.execute("SELECT count(*) FROM prompt WHERE is_done IS NULL").fetchone() avg_speed = conn.execute("SELECT avg(aa) FROM (SELECT (inference_steps*number)/abs(completed_at-started_at) AS aa FROM prompt WHERE started_at IS NOT NULL AND completed_at IS NOT NULL ORDER BY id DESC LIMIT 10)").fetchone() + avg_comp_time = avg_comp_time if avg_comp_time[0] is not None else [0] + min_max_wait = min_max_wait if min_max_wait[0] is not None else [0, 0] + avg_speed = avg_speed if avg_speed[0] is not None else [0] + await ev.respond("\n".join([ f"👯‍♂️ {waiting_usrs[0]} people are waiting in the queue", @@ -377,6 +381,7 @@ async def queue(ev, new_message=False): raise events.StopPropagation avg_comp_time = conn.execute('SELECT avg(aa) FROM (SELECT abs(queued_at-started_at) AS aa FROM prompt WHERE queued_at IS NOT NULL AND completed_at IS NOT NULL ORDER BY id DESC LIMIT 10)').fetchone() + avg_comp_time = avg_comp_time if avg_comp_time[0] is not None else [0] behind_you = conn.execute('SELECT count(*) FROM prompt WHERE is_done IS NULL AND id > ?', (prompt['id'],)).fetchone()[0] front_you = conn.execute('SELECT count(*) FROM prompt WHERE is_done IS NULL AND id < ?', (prompt['id'],)).fetchone()[0]