Update main.py

main.py

@@ -260,14 +260,16 @@ async def handle_media(chat_id: int, reply_to_id: int,
     # ── Progress callback for Pyrogram download ──────────────
     async def on_progress(current: int, total: int):
         now = time.time()
-        if now - last_edit[0] < 2.5:          # throttle edits to every 2.5s
+        if now - last_edit[0] < 2.5:
             return
         last_edit[0] = now
         elapsed = now - dl_start
         speed   = current / elapsed if elapsed > 0 else 0
+        # Pyrogram often passes total=0 — use file_size from webhook as fallback
+        actual_total = total if total and total > 0 else file_size
         try:
             await status_msg.edit(
-                build_progress_text("download", current, total,
+                build_progress_text("download", current, actual_total,
                                     speed, elapsed, short_name)
             )
         except Exception: