Update app.py

app.py

@@ -338,7 +338,6 @@ async def do_download_job(chat_id, enc_path, orig_name, job_id, cancel_event, st
 
 # --------------- DELETE FUNCTIONS ---------------
 async def delete_user_file(chat_id, user_id, enc_path, token, query):
-    """Delete a single file from HF and DB, then refresh the list."""
     try:
         hf.delete_file(
             path_in_repo=enc_path,
@@ -353,13 +352,11 @@ async def delete_user_file(chat_id, user_id, enc_path, token, query):
         await query.answer(f"Error: {e}", show_alert=True)
 
 async def delete_all_user_files(chat_id, user_id, query):
-    """Delete all files of a user from HF and DB, then refresh."""
     user_files = db_get_user_files(user_id)
     if not user_files:
         await query.answer("No files to delete.", show_alert=True)
         return
 
-    # Delete from HF
     for enc_path, _, _ in user_files:
         try:
             hf.delete_file(
@@ -368,12 +365,10 @@ async def delete_all_user_files(chat_id, user_id, query):
                 repo_type="dataset",
             )
         except Exception:
-            pass  # skip if not found
+            pass
 
-    # Clear DB
     db_remove_all_user_files(user_id)
 
-    # Clear callback tokens for this user
     tokens_to_delete = [tok for tok, path in file_callback_map.items() if path.startswith(get_encrypted_folder(user_id))]
     for tok in tokens_to_delete:
         file_callback_map.pop(tok, None)
@@ -382,7 +377,6 @@ async def delete_all_user_files(chat_id, user_id, query):
     await refresh_file_list(chat_id, user_id, query.message)
 
 async def refresh_file_list(chat_id, user_id, message):
-    """Re‑fetch user files and edit the message with updated buttons."""
     user_files = db_get_user_files(user_id)
     if not user_files:
         return await message.edit_text("You have no uploaded files.", reply_markup=None)
@@ -396,7 +390,6 @@ async def refresh_file_list(chat_id, user_id, message):
             InlineKeyboardButton("🗑 Delete", callback_data=f"delete:{token}")
         ])
 
-    # Add "Delete All" button as a separate row
     buttons.append([InlineKeyboardButton("⚠️ Delete All Files", callback_data=f"delete_all:{user_id}")])
 
     await message.edit_text(
@@ -415,7 +408,6 @@ async def process_album_group(media_group_id: str):
     messages = group_data['messages']   # list of (message_id, filename)
 
     if len(messages) == 1:
-        # single file – normal prompt
         msg_id, filename = messages[0]
         token = make_token()
         upload_prompt_map[token] = [(user_id, msg_id, filename)]
@@ -425,7 +417,6 @@ async def process_album_group(media_group_id: str):
         ])
         await app.send_message(chat_id, f"Do you want to upload `{filename}`?", reply_markup=kb)
     else:
-        # multiple files – combined prompt
         files_list = "\n".join(f"• {fname}" for _, fname in messages)
         token = make_token()
         upload_prompt_map[token] = [(user_id, msg_id, fname) for msg_id, fname in messages]
@@ -473,7 +464,7 @@ async def upload_worker():
 @app.on_message(filters.command("start"))
 async def start_cmd(_, message):
     await message.reply_text(
-        "Send me a file to upload it to the dataset.\n"
+        "Send me any file (document, video, audio, photo, animation, sticker, video note, voice) to upload.\n"
         "/files – list your uploaded files"
     )
 
@@ -482,7 +473,6 @@ async def list_files_cmd(_, message):
     user_id = message.from_user.id
     user_files = db_get_user_files(user_id)
 
-    # Fallback scan if DB empty
     if not user_files:
         enc_folder = get_encrypted_folder(user_id)
         try:
@@ -517,7 +507,6 @@ async def list_files_cmd(_, message):
             InlineKeyboardButton("🗑 Delete", callback_data=f"delete:{token}")
         ])
 
-    # Add "Delete All" button
     buttons.append([InlineKeyboardButton("⚠️ Delete All Files", callback_data=f"delete_all:{user_id}")])
 
     await message.reply_text(
@@ -641,19 +630,40 @@ async def upload_all_callback(_, query):
 async def cancel_upload_prompt(_, query):
     await query.message.delete()
 
-# ----------- RECEIVING FILES (with album grouping) -----------
-@app.on_message(filters.document | filters.video | filters.audio | filters.photo)
+# ----------- RECEIVING ALL FILE TYPES (including animations, stickers, video notes, voice) -----------
+@app.on_message(
+    filters.document | filters.video | filters.audio | filters.photo |
+    filters.animation | filters.sticker | filters.video_note | filters.voice
+)
 async def on_file(_, message):
+    # Determine filename for every possible type
     if message.document:
         filename = message.document.file_name
     elif message.video:
         filename = message.video.file_name or f"{message.video.file_unique_id}.mp4"
     elif message.audio:
         filename = message.audio.file_name or f"{message.audio.file_unique_id}.mp3"
-    else:
+    elif message.photo:
         filename = f"{message.photo.file_unique_id}.jpg"
+    elif message.animation:
+        filename = message.animation.file_name or f"{message.animation.file_unique_id}.mp4"
+    elif message.sticker:
+        # Stickers can be .webp, .tgs, .webm
+        if message.sticker.is_animated:
+            filename = f"{message.sticker.file_unique_id}.tgs"
+        elif message.sticker.is_video:
+            filename = f"{message.sticker.file_unique_id}.webm"
+        else:
+            filename = f"{message.sticker.file_unique_id}.webp"
+    elif message.video_note:
+        filename = f"{message.video_note.file_unique_id}.mp4"
+    elif message.voice:
+        filename = f"{message.voice.file_unique_id}.ogg"
+    else:
+        return  # should never happen
 
-    if message.media_group_id:
+    # Album grouping (for photos, videos, documents that belong to an album)
+    if message.media_group_id and (message.photo or message.video or message.document):
         group_id = str(message.media_group_id)
         chat_id = message.chat.id
         user_id = message.from_user.id