Create main.py

main.py

@@ -0,0 +1,109 @@
+import logging
+import os
+import yt_dlp
+from telegram import Update
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackContext
+
+# Enable logging
+logging.basicConfig(
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+# Bot token from environment variable
+TOKEN = "7449252023:AAGl88ZXaJD6Gu1M49zmnZNUvddsgIZZ0C0"
+if not TOKEN:
+    raise ValueError("No TELEGRAM_BOT_TOKEN found in environment variables")
+
+bot_username = "ethioytdownloaderbot"
+
+async def start(update: Update, context: CallbackContext) -> None:
+    await update.message.reply_text('Send me a YouTube link or a list of links separated by spaces, and I will download the videos for you!')
+
+def progress_hook(d, status_message, context):
+    if d['status'] == 'downloading':
+        total_bytes = d.get('total_bytes', 0)
+        downloaded_bytes = d.get('downloaded_bytes', 0)
+        percentage = (downloaded_bytes / total_bytes) * 100 if total_bytes > 0 else 0
+        message = f"Downloading video... {percentage:.2f}%"
+        context.bot.edit_message_text(chat_id=status_message.chat_id, message_id=status_message.message_id, text=message)
+
+async def download_youtube_video(update: Update, context: CallbackContext) -> None:
+    urls = update.message.text.split()
+    valid_urls = [url for url in urls if "youtube.com" in url or "youtu.be" in url]
+
+    if not valid_urls:
+        await update.message.reply_text('Please send valid YouTube links.')
+        return
+
+    status_message = await context.bot.send_message(chat_id=update.effective_chat.id, text="Processing your request...")
+
+    for url in valid_urls:
+        try:
+            ydl_opts = {
+                'format': 'best',  # Download the best available format without merging
+                'progress_hooks': [lambda d: progress_hook(d, status_message, context)],
+                'outtmpl': 'downloads/%(title)s.%(ext)s',
+            }
+
+            # Ensure 'downloads' directory exists
+            os.makedirs('downloads', exist_ok=True)
+
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info_dict = ydl.extract_info(url, download=True)
+                file_path = ydl.prepare_filename(info_dict)
+                thumbnail_url = info_dict.get('thumbnail', '')
+                description = info_dict.get('description', 'No description available')
+                title = info_dict.get('title', 'Unknown Title')
+                artist = info_dict.get('uploader', 'Unknown Artist')
+
+            file_size = os.path.getsize(file_path)
+            file_size_mb = file_size / (1024 * 1024)
+            await context.bot.edit_message_text(chat_id=update.effective_chat.id, message_id=status_message.message_id, text=f"Sending the video... (Size: {file_size_mb:.2f} MB)")
+
+            caption = f"Downloaded by @{bot_username}\nTitle: {title}\nArtist: {artist}\n\nDescription: {description}"
+            if len(caption) > 1024:
+                caption = caption[:1020] + "..."
+
+            await context.bot.send_video(
+                chat_id=update.effective_chat.id,
+                video=open(file_path, 'rb'),
+                caption=caption,
+                reply_to_message_id=update.message.message_id
+            )
+
+            # Send the thumbnail if it exists
+            if thumbnail_url:
+                await context.bot.send_photo(
+                    chat_id=update.effective_chat.id,
+                    photo=thumbnail_url,
+                    caption=f"Thumbnail for {title}"
+                )
+
+            os.remove(file_path)  # Delete the file after sending
+
+        except Exception as e:
+            logger.error(f"Error: {e}")
+            await context.bot.edit_message_text(chat_id=update.effective_chat.id, message_id=status_message.message_id, text=f'Failed to download the YouTube video: {url}. Please make sure the link is correct.')
+
+    await context.bot.delete_message(chat_id=update.effective_chat.id, message_id=status_message.message_id)
+
+async def error_handler(update: Update, context: CallbackContext) -> None:
+    logger.error(msg="Exception while handling an update:", exc_info=context.error)
+    if update and update.message:
+        await update.message.reply_text('An error occurred. Please try again later.')
+
+def main() -> None:
+    # Create the Application and pass it your bot's token
+    application = Application.builder().token(TOKEN).build()
+
+    # Add handlers
+    application.add_handler(CommandHandler("start", start))
+    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, download_youtube_video))
+    application.add_error_handler(error_handler)
+
+    # Start polling
+    application.run_polling()
+
+if __name__ == '__main__':
+    main()