Update app.py

app.py

@@ -5,7 +5,7 @@ import requests
 import asyncio
 import re
 import urllib3
-from flask import Flask, jsonify, make_response, request
+from flask import Flask, jsonify, make_response, request, Response
 from supabase import create_client
 from pyrogram import Client, filters, enums, idle
 from pyrogram.errors import SessionPasswordNeeded, PhoneCodeInvalid, PhoneCodeExpired, UserDeactivated, SessionRevoked, AuthKeyUnregistered, FloodWait
@@ -19,9 +19,12 @@ API_ID = int(os.environ.get("API_ID", 0))
 API_HASH = os.environ.get("API_HASH")     
 SUPABASE_URL = os.environ.get("SUPABASE_URL")
 SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
-BYSE_API_KEY = os.environ.get("BYSE_API_KEY")
 
 PREMIUM_CHANNEL_ID = -1002825744390
+STORAGE_CHANNEL_ID = -1002825744390  # যে চ্যানেলে ভিডিও স্টোর হবে
+
+# আপনার Hugging Face স্পেসের ডিরেক্ট ইউআরএল (টোকেন জেনারেট করার জন্য)
+BACKEND_URL = os.environ.get("BACKEND_URL", "https://pmrony-forwardbot.hf.space")
 
 # WebApp URL (index.html)
 WEB_APP_URL = "https://rony90790.github.io/Forward-bot/index.html" 
@@ -47,6 +50,84 @@ 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)
 
+# ==================== 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:
+                return
+            
+            # chunk_size অনুযায়ী ডাটা রিয়েল-টাইমে রিড করে পাঠানো হবে
+            async for chunk in bot.stream_media(msg.video):
+                yield chunk
+        except Exception as e:
+            print(f"Error in stream_generator: {e}")
+
+    # Async জেনারেটরকে Flask-এর সাথে সিঙ্ক করার জন্য ইভেন্ট লুপ ফিক্স
+    loop = asyncio.new_event_loop()
+    def run():
+        asyncio.set_event_loop(loop)
+        gen = stream_generator()
+        while True:
+            try:
+                chunk = loop.run_until_complete(gen.__anext__())
+                yield chunk
+            except StopAsyncIteration:
+                break
+            except Exception as e:
+                print(f"Sync wrapper error: {e}")
+                break
+        loop.close()
+    
+    return run()
+
+@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
+            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.headers['Content-Length'] = file_size
+        response.headers['Content-Type'] = mime_type or "video/mp4"
+        response.headers['Accept-Ranges'] = 'bytes'
+        response.headers['Content-Disposition'] = f'inline; filename="{file_name or "video.mp4"}"'
+        return response
+    except Exception as e:
+        return f"Error: {e}", 500
+
+@app.route('/download/<int:message_id>')
+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
+            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.headers['Content-Length'] = file_size
+        response.headers['Content-Disposition'] = f'attachment; filename="{file_name or "video.mp4"}"'
+        return response
+    except Exception as e:
+        return f"Error: {e}", 500
+# =======================================================================
+
 # ================= FLASK API ROUTES =================
 @app.route('/')
 def index():
@@ -312,12 +393,10 @@ async def clone_videos_background(client, source_id, dest_id, status_msg):
         else:
             await status_msg.edit_text(f"⏳ <b>Cloning started!</b>\nFetching video list from <code>{source_id}</code>...\n<i>This might take a few minutes if the group has many videos.</i>", parse_mode=enums.ParseMode.HTML)
         
-        # === Auto Reconnect & Retry loop for Video Scanning ===
         video_ids = []
         retries = 5
         while retries > 0:
             try:
-                # যদি বট ক্লায়েন্ট কোনো কারণে ডিসকানেক্ট হয়, রিকানেক্ট করবে
                 if not client.is_connected:
                     try: await client.connect()
                     except: pass
@@ -328,18 +407,16 @@ async def clone_videos_background(client, source_id, dest_id, status_msg):
                     video_ids.append(msg.id)
                     if len(video_ids) % 200 == 0:
                         await asyncio.sleep(0.1)
-                break # সফল হলে লুপ থেকে বের হবে
+                break 
             except Exception as e:
                 err_msg = str(e).lower()
-                # যদি ডিসকানেক্ট বা কানেকশন জনিত সমস্যা হয়
                 if "disconnect" in err_msg or "connection" in err_msg or "timeout" in err_msg or "reset" in err_msg:
                     retries -= 1
-                    video_ids = [] # লিস্ট রিসেট করা হলো যাতে কোনো ভিডিও মিস না যায়
+                    video_ids = [] 
                     await status_msg.edit_text(f"⚠️ <b>Network issue detected!</b>\nRetrying in 10 seconds... (Attempts left: {retries})\nError: <code>{e}</code>", parse_mode=enums.ParseMode.HTML)
                     await asyncio.sleep(10)
                 else:
-                    raise e # অন্য কোনো পারমিশন বা বড় এরর হলে কাজ থামিয়ে দেবে
-        # =======================================================
+                    raise e 
         
         if not video_ids:
             if last_copied_id:
@@ -347,7 +424,7 @@ async def clone_videos_background(client, source_id, dest_id, status_msg):
             else:
                 return await status_msg.edit_text("❌ <b>No videos found in the source group!</b>\n(Make sure the bot is an admin with read history permission in that group).", parse_mode=enums.ParseMode.HTML)
             
-        video_ids.reverse() # Oldest to newest
+        video_ids.reverse() 
         total = len(video_ids)
         
         if last_copied_id:
@@ -362,7 +439,6 @@ async def clone_videos_background(client, source_id, dest_id, status_msg):
             copy_success = False
             copy_retries = 3
             
-            # === Video Sending Auto Reconnect & Retry ===
             while copy_retries > 0:
                 try:
                     if not client.is_connected:
@@ -373,21 +449,19 @@ async def clone_videos_background(client, source_id, dest_id, status_msg):
                     success += 1
                     await save_progress(source_id, dest_id, msg_id)
                     copy_success = True
-                    break # কপি সফল হলে রিট্রাই লুপ থেকে বের হবে
+                    break 
                 except FloodWait as e:
-                    # লিমিট এররে রিট্রাই কমানো হবে না, শুধু ওয়েট করবে
                     await asyncio.sleep(e.value + 2)
                 except Exception as e:
                     err_msg = str(e).lower()
                     if "disconnect" in err_msg or "connection" in err_msg or "timeout" in err_msg or "reset" in err_msg:
                         copy_retries -= 1
-                        await asyncio.sleep(5) # ৫ সেকেন্ড পর আবার চেষ্টা করবে
+                        await asyncio.sleep(5) 
                     else:
-                        break # অন্য কোনো এরর (যেমন: ভিডিও ডিলিট করা) হলে লুপ স্কিপ করবে
+                        break 
             
             if not copy_success:
                 failed += 1
-            # ============================================
             
             if index % 20 == 0 or index == total:
                 try:
@@ -546,7 +620,7 @@ async def handle_media_upload(client, message):
         original_file = await message.download(progress=download_progress)
         clean_upload_file = original_file 
         
-        # ১. ওয়াটারমার্ক (Watermark) করা - Balanced (Superfast + CRF 23)
+        # ১. ওয়াটারমার্ক (Watermark) করা - HD 
         if media_type == "video" and not is_large_video:
             await status_msg.edit_text("⏳ Watermarking video... (HD + Superfast Processing)")
             watermarked_file = f"{original_file}_wm.mp4"
@@ -561,35 +635,31 @@ 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 / Clean ভিডিও আপলোড করা
+        # ২. byse.sx এর বদলে আপনার নিজের টেলিগ্রাম স্টোরেজ চ্যানেলে HD ভিডিও আপলোড
         if media_type == "video":
-            await status_msg.edit_text("⏳ Uploading Clean HD video to byse.sx server...")
-            api_endpoint = "https://api.byse.sx/upload/server"
-            loop = asyncio.get_event_loop()
-            response = await loop.run_in_executor(None, lambda: requests.get(api_endpoint, params={'key': BYSE_API_KEY}, timeout=30))
-            result = response.json()
+            await status_msg.edit_text("⏳ Uploading Clean HD video to your storage channel...")
             
-            if result.get('status') == 200:
-                upload_res = await loop.run_in_executor(None, upload_file_sync, result.get('result'), clean_upload_file, BYSE_API_KEY)
-                if upload_res.get('status') == 200 and 'files' in upload_res and len(upload_res['files']) > 0:
-                    file_status = upload_res['files'][0].get('status', '')
-                    if "not allowed" in str(file_status).lower():
-                        await status_msg.edit_text(f"❌ byse.sx rejected the file: <code>{file_status}</code>", parse_mode=enums.ParseMode.HTML)
-                        return
-                    file_code = upload_res['files'][0].get('filecode')
-                    if file_code: embed_link = f"https://bysesayeveum.com/e/{file_code}"
+            # আপনার নিজের চ্যানেলে ব্যাকআপ হিসেবে 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
             
-            if not embed_link:
-                await status_msg.edit_text("❌ Uploaded to byse.sx but Embed Link not found.")
-                return
+            # আপনার নিজস্ব ডোমেইন বা সার্ভারের স্ট্রিমিং ও ডিরেক্ট ডাউনলোড লিংক জেনারেট
+            stream_link = f"{BACKEND_URL}/stream/{storage_msg_id}"
+            download_link = f"{BACKEND_URL}/download/{storage_msg_id}"
+            embed_link = stream_link # এমবেড লিংক হিসেবে এখন নিজস্ব স্ট্রিমিং লিংক ব্যবহার হবে
 
         if is_large_video:
-            admin_cap = f"✅ <b>Success! (Large Video)</b>\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>\n\n📌 <i>Broadcast skipped due to large file size.</i>"
+            # যদি লার্জ ভিডিও হয়, তবে সেটিরও স্ট্রিমিং লিংক জেনারেট হয়ে অ্যাডমিন চ্যাটে শো করবে
+            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()
             return  
 
-        # ৩. টেলিগ্রামে পাঠানোর জন্য ভিডিও ব্লার করা
+        # ৩. টেলিগ্রামের জন্য ভিডিও ব্লার করা
         telegram_file = clean_upload_file
         if is_blur and not is_large_video:
             await status_msg.edit_text(f"⏳ Applying {blur_percent}% blur for Telegram broadcast...")
@@ -624,7 +694,13 @@ async def handle_media_upload(client, message):
             caption_text = f"{clean_caption}\n\n👇 <i>Click the button below to open Bot!</i>" if clean_caption else f"🔥 <b>New Premium Viral Content!</b> 🔞\n\n🎬 <b>Watch HD Video Here:</b>\n👉 <b><a href='{bot_link}'>▶️ Click Here to Watch</a></b>\n\n👇 <i>Click the button below to open Bot!</i>"
         
         group_markup = InlineKeyboardMarkup([[InlineKeyboardButton("🎬 Watch Full Video Here 🔞", url=bot_link)]])
-        admin_cap = f"✅ <b>Success!</b> Media is broadcasting...\n\n🔗 <b>Embed Link (Clean HD):</b>\n<code>{embed_link or 'N/A'}</code>"
+        
+        # অ্যাডমিন চ্যাটে সাকস���স মেসেজ ও লিংক পাঠানো হচ্ছে
+        admin_cap = (
+            f"✅ <b>Upload and Processing Complete!</b>\n\n"
+            f"🎬 <b>Stream/Watch Online Link:</b>\n<code>{stream_link}</code>\n\n"
+            f"📥 <b>Direct Download Link:</b>\n<code>{download_link}</code>"
+        )
         
         thumb_path = None
         if media_type == "video":