Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -115,10 +115,18 @@ class TaskManager:
 
 task_manager = TaskManager()
 
-async def progress(current, total, message, start, type_of_ps, file_name=None, reply_markup=None):
+async def progress(current, total, message, start, type_of_ps, file_name=None):
     """Progress Bar For Showing Progress While Uploading / Downloading File - Normal"""
     now = time.time()
     diff = now - start
+    buttons = [
+        [
+            InlineKeyboardButton(
+                text="Cancel",
+                callback_data="cancels"
+            )
+        ],
+    ]
     if round(diff % 10.00) == 0 or current == total:
         percentage = current * 100 / total
         speed = current / diff
@@ -148,7 +156,7 @@ async def progress(current, total, message, start, type_of_ps, file_name=None, r
                 await message.edit_text(
                     f"{type_of_ps}\n{tmp}",
                     parse_mode=enums.ParseMode.MARKDOWN,
-                    reply_markup=reply_markup
+                    reply_markup=InlineKeyboardMarkup(buttons)
                 )
             except FloodWait as e:
                 await asyncio.sleep(e.x)
@@ -212,9 +220,7 @@ async def handle_photo(client, message):
             )
             seconds_time = time.time()
             file_path = await message.download(
-                progress=progress(
-                    reply_markup=InlineKeyboardMarkup(buttons)
-                ),
+                progress=progress,
                 progress_args=(ai_reply, seconds_time, "Uploading..."),
             )
             caption = message.caption or "What's this?"
@@ -270,9 +276,7 @@ async def handle_video(client, message, model_):
             seconds_time = time.time()
             video_file_name = await message.download(
                 file_name="newvideo.mp4",
-                progress=progress(
-                    reply_markup=InlineKeyboardMarkup(buttons)
-                ),
+                progress=progress,
                 progress_args=(ai_reply, seconds_time, "Uploading...")
             )
             caption = message.caption or "What's this?"