Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -115,7 +115,7 @@ class TaskManager:
 
 task_manager = TaskManager()
 
-async def progress(current, total, message, start, type_of_ps, file_name=None):
+async def progress(current, total, message, start, type_of_ps, file_name=None, reply_markup=None):
     """Progress Bar For Showing Progress While Uploading / Downloading File - Normal"""
     now = time.time()
     diff = now - start
@@ -145,9 +145,11 @@ async def progress(current, total, message, start, type_of_ps, file_name=None):
                 pass
         else:
             try:
-                await message.edit(
-                    f"{type_of_ps}\n{tmp}", parse_mode=enums.ParseMode.MARKDOWN
-                    )
+                await message.edit_text(
+                    f"{type_of_ps}\n{tmp}",
+                    parse_mode=enums.ParseMode.MARKDOWN,
+                    reply_markup=reply_markup
+                )
             except FloodWait as e:
                 await asyncio.sleep(e.x)
             except MessageNotModified:
@@ -210,7 +212,7 @@ async def handle_photo(client, message):
             )
             seconds_time = time.time()
             file_path = await message.download(
-                progress=progress,
+                progress=progress(reply_markup=InlineKeyboardMarkup(buttons),
                 progress_args=(ai_reply, seconds_time, "Uploading...")
             )
             caption = message.caption or "What's this?"
@@ -266,7 +268,7 @@ async def handle_video(client, message, model_):
             seconds_time = time.time()
             video_file_name = await message.download(
                 file_name="newvideo.mp4",
-                progress=progress,
+                progress=progress(reply_markup=InlineKeyboardMarkup(buttons),
                 progress_args=(ai_reply, seconds_time, "Uploading...")
             )
             caption = message.caption or "What's this?"