import os
from telegram import Update
from telegram.ext import ApplicationBuilder, MessageHandler, filters, ContextTypes

with open("/run/secrets/bot_token", 'r') as f:
    BOT_TOKEN = f.read().strip()

SAVE_FOLDER = "/app/media_folder"

async def media_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
    if update.message.photo:
        file = await update.message.photo[-1].get_file()
        filename = os.path.join(SAVE_FOLDER, f"{file.file_unique_id}.jpg")
        await file.download_to_drive(filename)
    elif update.message.video:
        file = await update.message.video.get_file()
        filename = os.path.join(SAVE_FOLDER, f"{file.file_unique_id}.mp4")
        await file.download_to_drive(filename)

if __name__ == "__main__":
    app = ApplicationBuilder().token(BOT_TOKEN).build()
    app.add_handler(MessageHandler(filters.PHOTO | filters.VIDEO, media_handler))
    app.run_polling()