Update chatbot/plugins/chat.py

chatbot/plugins/chat.py

@@ -17,13 +17,24 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+import asyncio
+import importlib
+import math
+import os
+import re
+import shlex
+import subprocess
+import sys
+import time
+import traceback
+from io import BytesIO
+from types import ModuleType
+from typing import Dict, List, Optional, Tuple, Union
+
 import requests
 import time
 import json
-import asyncio
 import io
-import os
-import re
 
 from PIL import Image
 from pyrogram import *
@@ -104,6 +115,71 @@ class TaskManager:
 
 task_manager = TaskManager()
 
+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
+    if round(diff % 10.00) == 0 or current == total:
+        percentage = current * 100 / total
+        speed = current / diff
+        elapsed_time = round(diff) * 1000
+        if elapsed_time == 0:
+            return
+        time_to_completion = round((total - current) / speed) * 1000
+        estimated_total_time = elapsed_time + time_to_completion
+        progress_str = f"{''.join(['▰' for i in range(math.floor(percentage / 10))])}"
+        progress_str += (
+            f"{''.join(['▱' for i in range(10 - math.floor(percentage / 10))])}"
+            )
+        progress_str += f"{round(percentage, 2)}%\n"
+        tmp = f"{progress_str}{humanbytes(current)} of {humanbytes(total)}\n"
+        tmp += f"ETA: {time_formatter(estimated_total_time)}"
+        if file_name:
+            try:
+                await message.edit(
+                    f"{type_of_ps}\n**File Name:** `{file_name}`\n{tmp}"
+                    )
+            except FloodWait as e:
+                await asyncio.sleep(e.x)
+            except MessageNotModified:
+                pass
+        else:
+            try:
+                await message.edit(
+                    f"{type_of_ps}\n{tmp}", parse_mode=enums.ParseMode.MARKDOWN
+                    )
+            except FloodWait as e:
+                await asyncio.sleep(e.x)
+            except MessageNotModified:
+                pass
+
+def humanbytes(size):
+    """Convert Bytes To Bytes So That Human Can Read It"""
+    if not size:
+        return ""
+    power = 2**10
+    raised_to_pow = 0
+    dict_power_n = {0: "", 1: "Ki", 2: "Mi", 3: "Gi", 4: "Ti"}
+    while size > power:
+        size /= power
+        raised_to_pow += 1
+    return str(round(size, 2)) + " " + dict_power_n[raised_to_pow] + "B"
+
+ def time_formatter(milliseconds: int) -> str:
+    """Time Formatter"""
+    seconds, milliseconds = divmod(int(milliseconds), 1000)
+    minutes, seconds = divmod(seconds, 60)
+    hours, minutes = divmod(minutes, 60)
+    days, hours = divmod(hours, 24)
+    tmp = (
+        ((str(days) + " day(s), ") if days else "")
+        + ((str(hours) + " hour(s), ") if hours else "")
+        + ((str(minutes) + " minute(s), ") if minutes else "")
+        + ((str(seconds) + " second(s), ") if seconds else "")
+        + ((str(milliseconds) + " millisecond(s), ") if milliseconds else "")
+    )
+    return tmp[:-2]
+
 @Client.on_callback_query(filters.regex("^cancels"))
 async def cancel_(client, callback_query):
     chat_id = callback_query.message.chat.id
@@ -126,15 +202,21 @@ async def handle_photo(client, message):
         ]
         try:
             spam_chats.append(chat_id)
-            file_path = await message.download()
-            caption = message.caption or "What's this?"
-            x = GeminiLatest(api_keys=GOOGLE_API_KEY)
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
             ai_reply = await message.reply_text(
                 "Uploading file..",
                 reply_markup=InlineKeyboardMarkup(buttons)
             )
+            seconds_time = time.time()
+            file_path = await message.download(
+                progress=progress,
+                progress_args=(ai_reply, seconds_time, "Uploading...")
+            )
+            caption = message.caption or "What's this?"
+            x = GeminiLatest(api_keys=GOOGLE_API_KEY)
+            await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
+            await asyncio.sleep(1.5)
             backup_chat = await db._get_chatbot_chat_from_db(message.from_user.id)
             backup_chat.append({"role": "user", "parts": [{"text": caption}]})
             response_reads = x.get_response_image(caption, file_path)
@@ -175,15 +257,20 @@ async def handle_video(client, message, model_):
         ]
         try:
             spam_chats.append(chat_id)
-            video_file_name = await message.download(file_name="newvideo.mp4")
-            caption = message.caption or "What's this?"
-            model = genai.GenerativeModel(model_name=model_)
             await client.send_chat_action(message.chat.id, enums.ChatAction.UPLOAD_PHOTO)
             await asyncio.sleep(1.5)
             ai_reply = await message.reply_text(
                 "Uploading file..",
                 reply_markup=InlineKeyboardMarkup(buttons)
             )
+            seconds_time = time.time()
+            video_file_name = await message.download(
+                file_name="newvideo.mp4",
+                progress=progress,
+                progress_args=(ai_reply, seconds_time, "Uploading...")
+            )
+            caption = message.caption or "What's this?"
+            model = genai.GenerativeModel(model_name=model_)
             video_file = genai.upload_file(path=video_file_name)
             while video_file.state.name == "PROCESSING":
                 await asyncio.sleep(10)