Update Akeno/plugins/imageai.py

Akeno/plugins/imageai.py

@@ -47,19 +47,20 @@ async def schellwithflux(args):
     & ~filters.forwarded
 )
 async def imgfluxai_(client: Client, message: Message):
+    ok = await message.reply_text("Processing...")
     question = message.text.split(" ", 1)[1] if len(message.command) > 1 else None
     if not question:
-        return await message.reply_text("Please provide a question for Flux.")
+        return await ok.edit_text("Please provide a question for Flux.")
     try:
         if not HUGGING_TOKEN:
-            return await message.reply_text("`HUGGING_TOKEN` is required to use this feature.")
+            return await ok.edit_text("`HUGGING_TOKEN` is required to use this feature.")
         image_bytes = await schellwithflux(question)
         if image_bytes is None:
-            return await message.reply_text("Failed to generate an image.")
-        pro = await message.reply_text("Generating image, please wait...")
+            return await ok.edit_text("Failed to generate an image.")
+        # 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 pro.edit_text("Uploading image...")
+        # 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()
     except Exception as e: