Update app.py

app.py

@@ -7,14 +7,13 @@ import re
 import urllib3
 from flask import Flask, jsonify, make_response, request, Response
 from supabase import create_client
-from pyrogram import Client, filters, enums, idle, utils  # utils ইম্পোর্ট করা হলো
+from pyrogram import Client, filters, enums, idle, utils  
 from pyrogram.errors import SessionPasswordNeeded, PhoneCodeInvalid, PhoneCodeExpired, UserDeactivated, SessionRevoked, AuthKeyUnregistered, FloodWait
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
 # ==================== PYROGRAM NEW ID RANGE FIX (MONKEY PATCH) ====================
-# টেলিগ্রামের নতুন আইডিগুলোর জন্য Pyrogram-এর Peer ID Invalid বাগ সমাধান করা হলো
 def get_peer_type_new(peer_id: int) -> str:
     peer_id_str = str(peer_id)
     if not peer_id_str.startswith("-"):
@@ -37,7 +36,7 @@ SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
 PREMIUM_CHANNEL_ID = -1002825744390
 STORAGE_CHANNEL_ID = -1002825744390  # যে চ্যানেলে ভিডিও স্টোর হবে
 
-# আপনার Hugging Face স্পেসের ডিরেক্ট ইউআরএল (টোকেন জেনারেট করার জন্য)
+# আপনার Hugging Face স্পেসের ডিরেক্ট ইউআরএল
 BACKEND_URL = os.environ.get("BACKEND_URL", "https://pmrony-forwardbot.hf.space")
 
 # WebApp URL (index.html)
@@ -64,17 +63,45 @@ bot = Client("file_unlocker_bot", api_id=API_ID, api_hash=API_HASH, bot_token=BO
 async def db_query(func):
     return await asyncio.to_thread(func)
 
+# ==================== UNIVERSAL MEDIA HELPER ====================
+def get_media_obj(msg):
+    """মেসেজ থেকে যেকোনো ধরনের মিডিয়া অবজেক্ট (ভিডিও, জিআইএফ, ডকুমেন্ট) খুঁজে বের করার ফাংশন"""
+    if not msg:
+        return None
+    if msg.video:
+        return msg.video
+    if msg.animation:
+        return msg.animation
+    if msg.document:
+        return msg.document
+    if msg.audio:
+        return msg.audio
+    return None
+
+def get_msg_file_id(msg):
+    """মেসেজ থেকে ক্র্যাশ-ফ্রি উপায়ে file_id বের করার ফাংশন"""
+    if not msg:
+        return None
+    if msg.photo:
+        return msg.photo.file_id
+    media = get_media_obj(msg)
+    if media:
+        return media.file_id
+    return None
+# ================================================================
+
 # ==================== CUSTOM VIDEO STREAMING ENGINE ====================
 def get_file_stream(message_id):
     """টেলিগ্রামের স্টোরেজ চ্যানেল থেকে রিয়েল-টাইমে ডাটা স্ট্রিম করার জেনারেটর"""
     async def stream_generator():
         try:
             msg = await bot.get_messages(STORAGE_CHANNEL_ID, message_id)
-            if not msg or not msg.video:
+            media = get_media_obj(msg)
+            if not media:
                 return
             
             # chunk_size অনুযায়ী ডাটা রিয়েল-টাইমে রিড করে পাঠানো হবে
-            async for chunk in bot.stream_media(msg.video):
+            async for chunk in bot.stream_media(media):
                 yield chunk
         except Exception as e:
             print(f"Error in stream_generator: {e}")
@@ -99,21 +126,23 @@ def get_file_stream(message_id):
 
 @app.route('/stream/<int:message_id>')
 def stream_video(message_id):
-    """অনলাইনে প্লেয়ারে ভিডিও দেখার লিংক"""
+    """অনলাইনে প্লেয়ারে ভিডিও/জিআইএফ দেখার লিংক"""
     try:
         async def get_media_info():
             msg = await bot.get_messages(STORAGE_CHANNEL_ID, message_id)
-            if msg and msg.video:
-                return msg.video.file_size, msg.video.file_name, msg.video.mime_type
+            media = get_media_obj(msg)
+            if media:
+                mime = getattr(media, 'mime_type', 'video/mp4') or 'video/mp4'
+                return media.file_size, getattr(media, 'file_name', 'video.mp4'), mime
             return None, None, None
 
         file_size, file_name, mime_type = run_async(get_media_info())
         if not file_size:
             return "File not found or invalid message", 404
 
-        response = make_response(Response(get_file_stream(message_id), mimetype=mime_type or "video/mp4"))
+        response = make_response(Response(get_file_stream(message_id), mimetype=mime_type))
         response.headers['Content-Length'] = file_size
-        response.headers['Content-Type'] = mime_type or "video/mp4"
+        response.headers['Content-Type'] = mime_type
         response.headers['Accept-Ranges'] = 'bytes'
         response.headers['Content-Disposition'] = f'inline; filename="{file_name or "video.mp4"}"'
         return response
@@ -126,15 +155,17 @@ def download_video(message_id):
     try:
         async def get_media_info():
             msg = await bot.get_messages(STORAGE_CHANNEL_ID, message_id)
-            if msg and msg.video:
-                return msg.video.file_size, msg.video.file_name, msg.video.mime_type
+            media = get_media_obj(msg)
+            if media:
+                mime = getattr(media, 'mime_type', 'application/octet-stream') or 'application/octet-stream'
+                return media.file_size, getattr(media, 'file_name', 'video.mp4'), mime
             return None, None, None
 
         file_size, file_name, mime_type = run_async(get_media_info())
         if not file_size:
             return "File not found or invalid message", 404
 
-        response = make_response(Response(get_file_stream(message_id), mimetype=mime_type or "application/octet-stream"))
+        response = make_response(Response(get_file_stream(message_id), mimetype=mime_type))
         response.headers['Content-Length'] = file_size
         response.headers['Content-Disposition'] = f'attachment; filename="{file_name or "video.mp4"}"'
         return response
@@ -649,25 +680,26 @@ async def handle_media_upload(client, message):
             if process.returncode == 0 and os.path.exists(watermarked_file): 
                 clean_upload_file = watermarked_file
 
-        # ২. byse.sx এর বদলে আপনার নিজের টেলিগ্রাম স্টোরেজ চ্যানেলে HD ভিডিও আপলোড
-        if media_type == "video":
-            await status_msg.edit_text("⏳ Uploading Clean HD video to your storage channel...")
+        # ২. আপনার নিজের টেলিগ্রাম স্টোরেজ চ্যানেলে ফাইল আপলোড (ভিডিও, অ্যানিমেশন বা ফটো)
+        storage_msg_id = None
+        if media_type in ["video", "animation", "photo"]:
+            await status_msg.edit_text(f"⏳ Uploading {media_type} to your storage channel...")
+            
+            if media_type == "photo":
+                sent_to_channel = await client.send_photo(chat_id=STORAGE_CHANNEL_ID, photo=clean_upload_file, caption="Backup of photo uploaded by Admin.")
+            elif media_type == "animation":
+                sent_to_channel = await client.send_animation(chat_id=STORAGE_CHANNEL_ID, animation=clean_upload_file, caption="Backup of GIF uploaded by Admin.")
+            else:
+                sent_to_channel = await client.send_video(chat_id=STORAGE_CHANNEL_ID, video=clean_upload_file, caption=f"Backup of video uploaded by Admin. File: {os.path.basename(clean_upload_file)}")
             
-            # আপনার নিজের চ্যানেলে ব্যাকআপ হিসেবে HD ওয়াটারমার্কড ভিডিওটি পাঠানো হচ্ছে
-            sent_to_channel = await client.send_video(
-                chat_id=STORAGE_CHANNEL_ID,
-                video=clean_upload_file,
-                caption=f"Backup of video uploaded by Admin. File: {os.path.basename(clean_upload_file)}"
-            )
             storage_msg_id = sent_to_channel.id
             
             # আপনার নিজস্ব ডোমেইন বা সার্ভারের স্ট্রিমিং ও ডিরেক্ট ডাউনলোড লিংক জেনারেট
             stream_link = f"{BACKEND_URL}/stream/{storage_msg_id}"
             download_link = f"{BACKEND_URL}/download/{storage_msg_id}"
-            embed_link = stream_link # এমবেড লিংক হিসেবে এখন নিজস্ব স্ট্রিমিং লিংক ব্যবহার হবে
+            embed_link = stream_link
 
         if is_large_video:
-            # যদি লার্জ ভিডিও হয়, তবে সেটিরও স্ট্রিমিং লিংক জেনারেট হয়ে অ্যাডমিন চ্যাটে শো করবে
             admin_cap = f"✅ <b>Success! (Large Video)</b>\n\n🔗 <b>Embed Link (Clean HD):</b>\n<code>{embed_link or 'N/A'}</code>\n\n📌 <i>Broadcast skipped due to large file size.</i>"
             await client.send_video(message.chat.id, message.video.file_id, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
             await status_msg.delete()
@@ -725,10 +757,8 @@ async def handle_media_upload(client, message):
 
         if media_type == "photo":
             sent_to_admin = await client.send_photo(message.chat.id, telegram_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
-            tg_file_id = sent_to_admin.photo.file_id
         elif media_type == "animation":
             sent_to_admin = await client.send_animation(message.chat.id, telegram_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)
-            tg_file_id = sent_to_admin.animation.file_id
         else:
             vid_duration = message.video.duration if message.video and message.video.duration else 0
             vid_width = message.video.width if message.video and message.video.width else 0
@@ -744,7 +774,11 @@ async def handle_media_upload(client, message):
                 height=vid_height, 
                 thumb=thumb_path
             )
-            tg_file_id = sent_to_admin.video.file_id
+            
+        # ক্র্যাশ-ফ্রি উপায়ে file_id রিড করার জন্য ইউনিভার্সাল এক্সট্রাক্টর
+        tg_file_id = get_msg_file_id(sent_to_admin)
+        if not tg_file_id:
+            tg_file_id = get_msg_file_id(message) # Fallback to original
             
         await status_msg.delete()