Update Akeno/plugins/imageai.py

Akeno/plugins/imageai.py

@@ -60,9 +60,10 @@ async def imgfluxai_(client: Client, message: Message):
         # await ok.edit_text("Generating image, please wait...")
         with Image.open(io.BytesIO(image_bytes)) as img:
             img.save("testing.jpg", format="JPEG")
-        # ok = await ok.edit_text("Uploading image...")
-        await message.reply_photo("testing.jpg", progress=progress, progress_args=(ok, time.time(), "Uploading image..."))
-        await ok.delete()
+        seconds_uploading = await ok.edit_text("Uploading image...")
+        seconds_time = time.time()
+        await message.reply_photo("testing.jpg", progress=progress, progress_args=(seconds_uploading, seconds_time, "Uploading image..."))
+        await seconds_image.delete()
     except Exception as e:
         LOGS.error(str(e))
-        await ok.edit_text(f"An error occurred: {str(e)}")
+        await message.edit_text(f"An error occurred: {str(e)}")