Update bot.js

bot.js

@@ -1,197 +1,180 @@
-// ============================================================
-// 🪄 HUGGING FACE KEEP-ALIVE TRICK (Explicit 0.0.0.0 bind)
-// ============================================================
-const http = require('http');
-const server = http.createServer((req, res) => {
-    res.writeHead(200, { 'Content-Type': 'text/plain' });
-    res.end('Lecture Downloader Bot is alive and running on Hugging Face!');
-});
-// Explicitly binding to 0.0.0.0 is critical for Hugging Face network routing
-server.listen(7860, '0.0.0.0', () => console.log('🌐 Web server running on 0.0.0.0:7860'));
-
-// ============================================================
-// 🤖 BOT LIBRARIES
-// ============================================================
-const TelegramBot = require('node-telegram-bot-api');
-const axios = require('axios');
-const fs = require('fs');
-const path = require('path');
-const { exec } = require('child_process');
-const FormData = require('form-data');
-
-// ============================================================
-// ⚙️ CONFIGURATION
-// ============================================================
-const TOKEN = '8476068831:AAG9mEiMYhlIOEBYSy2P11V88NePwrJh06w'; 
-const OWNER_USERNAME = 'silent000666'; 
-const BANNER_URL = 'https://img.freepik.com/free-vector/laptop-with-program-code-isometric-icon-software-development-programming-applications-dark-neon_39422-971.jpg';
-const userState = {}; 
-
-// ============================================================
-// 📝 LOGGER
-// ============================================================
-const LOG_FILE = path.join(__dirname, 'debug_log.txt');
-fs.writeFileSync(LOG_FILE, `--- BOT STARTED ---\n`);
-
-function log(message) {
-    const timestamp = new Date().toISOString().split('T')[1].split('.')[0];
-    console.log(`[${timestamp}] ${message}`);
-    fs.appendFileSync(LOG_FILE, `[${timestamp}] ${message}\n`);
-}
-
-// ============================================================
-// 🚀 DELAYED BOOT SEQUENCE (The Network Fix)
-// ============================================================
-console.log("⏳ Waiting 5 seconds for Hugging Face DNS to stabilize...");
-
-setTimeout(() => {
-    console.log("🚀 Network Ready! Booting Telegram Bot...");
+import os
+import time
+import re
+import asyncio
+import requests
+import threading
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from pyrogram import Client, filters, idle
+from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery
+
+# ==========================================
+# 🪄 HUGGING FACE KEEP-ALIVE TRICK
+# ==========================================
+class DummyHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        self.send_response(200)
+        self.end_headers()
+        self.wfile.write(b"Lecture Bot is alive and running on Hugging Face!")
+
+def keep_alive():
+    server = HTTPServer(('0.0.0.0', 7860), DummyHandler)
+    server.serve_forever()
+
+threading.Thread(target=keep_alive, daemon=True).start()
+
+# ==========================================
+# ⚙️ CONFIGURATION
+# ==========================================
+TOKEN = "8476068831:AAG9mEiMYhlIOEBYSy2P11V88NePwrJh06w"
+API_ID = 2040 
+API_HASH = "b18441a1ff607e10a989891a5462e627" 
+
+BANNER_URL = "https://img.freepik.com/free-vector/laptop-with-program-code-isometric-icon-software-development-programming-applications-dark-neon_39422-971.jpg"
+
+app = Client("LectureBot", api_id=API_ID, api_hash=API_HASH, bot_token=TOKEN)
+user_states = {}
+
+print("🚀 LECTURE BOT STARTED...")
+
+# ==========================================
+# 📤 GOFILE UPLOAD HELPER
+# ==========================================
+def upload_to_gofile(filepath):
+    print("Initiating Gofile Upload...")
+    with open(filepath, 'rb') as f:
+        files = {'file': f}
+        # Gofile creates a guest account automatically when no token is provided
+        res = requests.post('https://upload.gofile.io/uploadfile', files=files)
+        data = res.json()
+        if data.get('status') == 'ok':
+            return data['data']['downloadPage']
+        else:
+            raise Exception(f"Gofile API Error: {data.get('status')}")
+
+# ==========================================
+# 🎮 BOT HANDLERS
+# ==========================================
+@app.on_message(filters.command("start"))
+async def start_cmd(client, message):
+    user_states[message.chat.id] = None
     
-    const bot = new TelegramBot(TOKEN, { polling: true });
-
-    // Stop errors from crashing the bot
-    bot.on("polling_error", (err) => log(`[Polling Error] ${err.message}`));
-
-    // ============================================================
-    // 🎮 HANDLERS
-    // ============================================================
-    bot.onText(/\/start/, async (msg) => {
-        userState[msg.chat.id] = null;
-        await bot.sendPhoto(msg.chat.id, BANNER_URL, {
-            caption: `👋 *Welcome, ${msg.from.first_name}*\n\nI am your **Lecture Downloader**.\n\n🔹 **Modes:**\n• < 50MB: Direct Telegram File\n• > 50MB: High-Speed Gofile Link\n\n_Click below to start:_`,
-            parse_mode: 'Markdown',
-            reply_markup: {
-                inline_keyboard: [
-                    [{ text: "🚀 Start Extraction", callback_data: "btn_extract" }]
-                ]
-            }
-        });
-    });
-
-    bot.on('callback_query', async (q) => {
-        const chatId = q.message.chat.id;
-        bot.answerCallbackQuery(q.id);
-
-        if (q.data === 'btn_extract') {
-            userState[chatId] = 'WAITING_URL';
-            await bot.sendMessage(chatId, "🔗 *Send the Lecture Link now.*", {
-                reply_markup: { inline_keyboard: [[{ text: "❌ Cancel", callback_data: "btn_cancel" }]] }
-            });
-        }
-        else if (q.data === 'btn_cancel') {
-            userState[chatId] = null;
-            bot.deleteMessage(chatId, q.message.message_id);
-        }
-    });
-
-    bot.on('message', async (msg) => {
-        const chatId = msg.chat.id;
-        const text = msg.text;
-
-        if (!text || text.startsWith('/') || userState[chatId] !== 'WAITING_URL') return;
-
-        userState[chatId] = null;
-        if (!text.startsWith('http')) return bot.sendMessage(chatId, "⚠️ Invalid Link.");
-
-        fs.writeFileSync(LOG_FILE, `--- NEW PROCESS STARTED ---\n`);
-        log(`User sent link: ${text}`);
-
-        const processMsg = await bot.sendMessage(chatId, "🕵️‍♂️ *Starting Process...*\n_Analyzing..._", { parse_mode: 'Markdown' });
-
-        try {
-            log("STEP 1: Analyzing URL...");
-            let streamUrl = text;
-            let cookieString = "";
-            let userAgent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)';
-
-            try {
-                const res = await axios.get(text, { headers: { 'User-Agent': userAgent }, timeout: 15000 });
-                if (res.headers['set-cookie']) {
-                    cookieString = res.headers['set-cookie'].map(c => c.split(';')[0]).join('; ');
-                }
-                const match = res.data.match(/src:\s*'(\/Stream\/Index\/[^']+)'/);
-                if (match) {
-                    streamUrl = `https://vss-v2-app.techsol360.com${match[1]}`;
-                    log(`✅ Found HIDDEN Stream URL`);
-                } 
-            } catch (e) {
-                log(`⚠️ Scrape Error: ${e.message} (Using original link)`);
-            }
-
-            log("STEP 2: Starting FFmpeg Download...");
-            await bot.editMessageText(`⬇️ *Downloading...*\n_Please wait..._`, { chat_id: chatId, message_id: processMsg.message_id, parse_mode: 'Markdown' });
-
-            const timestamp = Date.now();
-            const filePath = path.join(__dirname, `Lecture_${timestamp}.mp4`);
+    keyboard = InlineKeyboardMarkup([
+        [InlineKeyboardButton("🚀 Start Extraction", callback_data="btn_extract")]
+    ])
+    
+    await message.reply_photo(
+        photo=BANNER_URL,
+        caption=f"👋 **Welcome, {message.from_user.first_name}**\n\nI am your **Lecture Downloader**.\n\n🔹 **Modes:**\n• < 50MB: Direct Telegram File\n• > 50MB: High-Speed Gofile Link\n\n_Click below to start:_".replace(".", "\\."),
+        reply_markup=keyboard
+    )
+
+@app.on_callback_query()
+async def callback_handler(client, query: CallbackQuery):
+    chat_id = query.message.chat.id
+    
+    if query.data == "btn_extract":
+        user_states[chat_id] = "WAITING_URL"
+        keyboard = InlineKeyboardMarkup([[InlineKeyboardButton("❌ Cancel", callback_data="btn_cancel")]])
+        await query.message.reply_text("🔗 **Send the Lecture Link now.**", reply_markup=keyboard)
+        await query.answer()
+        
+    elif query.data == "btn_cancel":
+        user_states[chat_id] = None
+        await query.message.delete()
+        await query.answer("Cancelled.")
+
+@app.on_message(filters.text & ~filters.command("start"))
+async def message_handler(client, message):
+    chat_id = message.chat.id
+    text = message.text
+
+    if user_states.get(chat_id) != "WAITING_URL":
+        return
+
+    user_states[chat_id] = None # Reset state
+
+    if not text.startswith("http"):
+        return await message.reply_text("⚠️ Invalid Link.")
+
+    process_msg = await message.reply_text("🕵️‍♂️ **Starting Process...**\n_Analyzing..._")
+
+    try:
+        # --- STEP 1: ANALYZE URL ---
+        print(f"STEP 1: Analyzing URL: {text}")
+        stream_url = text
+        cookie_string = ""
+        user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64)"
+
+        try:
+            res = requests.get(text, headers={'User-Agent': user_agent}, timeout=15)
             
-            const cmd = `ffmpeg -user_agent "${userAgent}" -headers "Cookie: ${cookieString}" -i "${streamUrl}" -c copy -bsf:a aac_adtstoasc "${filePath}" -y`;
+            # Extract cookies if present
+            if 'set-cookie' in res.headers:
+                raw_cookies = res.headers['set-cookie'].split(',')
+                cookie_string = "; ".join([c.split(';')[0] for c in raw_cookies])
+
+            # Regex check for hidden stream URL
+            match = re.search(r"src:\s*'(/Stream/Index/[^']+)'", res.text)
+            if match:
+                stream_url = f"https://vss-v2-app.techsol360.com{match.group(1)}"
+                print("✅ Found HIDDEN Stream URL")
+        except Exception as e:
+            print(f"⚠️ Scrape Error: {str(e)} (Using original link)")
+
+        # --- STEP 2: DOWNLOAD ---
+        print("STEP 2: Starting FFmpeg Download...")
+        await process_msg.edit_text("⬇️ **Downloading...**\n_Please wait..._")
+        
+        timestamp = int(time.time())
+        filepath = f"Lecture_{timestamp}.mp4"
+
+        cmd = f'ffmpeg -user_agent "{user_agent}" -headers "Cookie: {cookie_string}" -i "{stream_url}" -c copy -bsf:a aac_adtstoasc "{filepath}" -y'
+        
+        # Run FFmpeg asynchronously so it doesn't block the bot
+        process = await asyncio.create_subprocess_shell(
+            cmd,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE
+        )
+        await process.communicate()
+
+        if not os.path.exists(filepath):
+            raise Exception("File missing after FFmpeg download.")
+
+        # Check size
+        size_bytes = os.path.getsize(filepath)
+        size_mb = size_bytes / (1024 * 1024)
+        print(f"✅ Download Success. Size: {size_mb:.2f} MB")
+
+        # --- STEP 3: UPLOAD ---
+        if size_mb < 49.0:
+            print("STEP 3: Uploading to Telegram...")
+            await process_msg.edit_text("📤 **Uploading to Telegram...**")
+            await message.reply_document(
+                document=filepath, 
+                caption=f"📦 {size_mb:.2f} MB"
+            )
+        else:
+            print("STEP 3: Uploading to Gofile...")
+            await process_msg.edit_text("🚀 **Uploading to Gofile Cloud...**")
+            link = await asyncio.to_thread(upload_to_gofile, filepath)
+            print(f"✅ Gofile Success: {link}")
             
-            exec(cmd, { maxBuffer: 1024 * 1024 * 50 }, async (err, stdout, stderr) => {
-                if (err) {
-                    log(`❌ FFmpeg Error: ${err.message}`);
-                    await sendErrorLog(chatId, processMsg.message_id, bot);
-                    return;
-                }
-                if (!fs.existsSync(filePath)) {
-                    log("❌ File missing after download.");
-                    await sendErrorLog(chatId, processMsg.message_id, bot);
-                    return;
-                }
-
-                const stats = fs.statSync(filePath);
-                const sizeMB = stats.size / (1024 * 1024);
-                log(`✅ Download Success. Size: ${sizeMB.toFixed(2)} MB`);
-
-                if (sizeMB < 49) {
-                    log("STEP 3: Uploading to Telegram...");
-                    await bot.editMessageText(`📤 *Uploading to Telegram...*`, { chat_id: chatId, message_id: processMsg.message_id, parse_mode: 'Markdown' });
-                    try {
-                        await bot.sendDocument(chatId, filePath, { caption: `📦 ${sizeMB.toFixed(2)} MB` });
-                        log("Telegram Upload Success.");
-                    } catch (tgErr) { log(`Telegram Error: ${tgErr.message}`); }
-                } else {
-                    log("STEP 3: Uploading to Gofile...");
-                    await bot.editMessageText(`🚀 *Uploading to Gofile Cloud...*`, { chat_id: chatId, message_id: processMsg.message_id, parse_mode: 'Markdown' });
-                    try {
-                        const link = await uploadToGofile(filePath);
-                        log(`✅ Gofile Success: ${link}`);
-                        await bot.sendMessage(chatId, `✅ *Done!*\n\n🔗 [Download Link](${link})\n📦 Size: ${sizeMB.toFixed(2)} MB`, { parse_mode: 'Markdown', disable_web_page_preview: true });
-                    } catch (cloudErr) {
-                        log(`❌ Gofile Failed: ${cloudErr.message}`);
-                        await sendErrorLog(chatId, processMsg.message_id, bot);
-                    }
-                }
-                if (fs.existsSync(filePath)) fs.unlinkSync(filePath);
-            });
-        } catch (error) {
-            log(`❌ CRITICAL: ${error.message}`);
-            await sendErrorLog(chatId, processMsg.message_id, bot);
-        }
-    });
-
-}, 5000); // <-- This is the 5-second Magic Delay!
-
-// ============================================================
-// 📤 GOFILE UPLOAD
-// ============================================================
-async function uploadToGofile(filePath) {
-    log("Initiating Gofile Upload...");
-    const form = new FormData();
-    form.append('file', fs.createReadStream(filePath));
-    const res = await axios.post('https://upload.gofile.io/uploadfile', form, {
-        headers: { ...form.getHeaders(), 'User-Agent': 'Mozilla/5.0' },
-        maxContentLength: Infinity, maxBodyLength: Infinity
-    });
-    if (res.data.status === 'ok') return res.data.data.downloadPage;
-    else throw new Error(`Gofile API Status: ${res.data.status}`);
-}
-
-// ============================================================
-// 🚨 ERROR REPORTER
-// ============================================================
-async function sendErrorLog(chatId, msgId, botInstance) {
-    await botInstance.editMessageText("❌ *Process Failed.* Sending logs...", { chat_id: chatId, message_id: msgId, parse_mode: 'Markdown' });
-    try {
-        await botInstance.sendDocument(chatId, LOG_FILE, { caption: "📜 *Error Log*\nOpen this file to see details." });
-    } catch (e) { console.error(e); }
-}
+            await message.reply_text(
+                f"✅ **Done!**\n\n🔗 [Download Link]({link})\n📦 Size: {size_mb:.2f} MB", 
+                disable_web_page_preview=True
+            )
+
+        # Cleanup
+        if os.path.exists(filepath):
+            os.remove(filepath)
+
+    except Exception as e:
+        print(f"❌ CRITICAL ERROR: {str(e)}")
+        await process_msg.edit_text(f"❌ **Process Failed.**\n`{str(e)}`")
+        if 'filepath' in locals() and os.path.exists(filepath):
+            os.remove(filepath)
+
+if __name__ == "__main__":
+    app.run()